Skip to content

Commit 16a5357

Browse files
test: attempt to fix pending query error on ddb audit tests
1 parent f13e924 commit 16a5357

1 file changed

Lines changed: 19 additions & 22 deletions

File tree

  • tests/test_core_engine/test_backends/test_implementations/test_duckdb

tests/test_core_engine/test_backends/test_implementations/test_duckdb/test_audit_ddb.py

Lines changed: 19 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -423,25 +423,22 @@ def test_get_submission_status(ddb_audit_manager_threaded: DDBAuditingManager):
423423
SubmissionStatisticsRecord(submission_id=sub_1.submission_id, record_count=5, number_record_rejections=2, number_warnings=3),
424424
SubmissionStatisticsRecord(submission_id=sub_4.submission_id, record_count=20, number_record_rejections=0, number_warnings=1)
425425
])
426-
427-
while not aud.queue.empty():
428-
time.sleep(0.05)
429-
430-
sub_stats_1 = aud.get_submission_status(sub_1.submission_id)
431-
assert sub_stats_1.submission_result == "validation_failed"
432-
assert sub_stats_1.validation_failed
433-
assert not sub_stats_1.processing_failed
434-
assert sub_stats_1.number_of_records == 5
435-
sub_stats_2 = aud.get_submission_status(sub_2.submission_id)
436-
assert sub_stats_2.submission_result == "processing_failed"
437-
assert not sub_stats_2.validation_failed
438-
assert sub_stats_2.processing_failed
439-
sub_stats_3 = aud.get_submission_status(sub_3.submission_id)
440-
assert not sub_stats_3.validation_failed
441-
assert not sub_stats_3.processing_failed
442-
sub_stats_4 = aud.get_submission_status(sub_4.submission_id)
443-
assert sub_stats_4.submission_result == "success"
444-
assert not sub_stats_4.validation_failed
445-
assert not sub_stats_4.processing_failed
446-
assert sub_stats_4.number_of_records == 20
447-
assert not aud.get_submission_status("5")
426+
427+
sub_stats_1 = ddb_audit_manager_threaded.get_submission_status(sub_1.submission_id)
428+
assert sub_stats_1.submission_result == "validation_failed"
429+
assert sub_stats_1.validation_failed
430+
assert not sub_stats_1.processing_failed
431+
assert sub_stats_1.number_of_records == 5
432+
sub_stats_2 = ddb_audit_manager_threaded.get_submission_status(sub_2.submission_id)
433+
assert sub_stats_2.submission_result == "processing_failed"
434+
assert not sub_stats_2.validation_failed
435+
assert sub_stats_2.processing_failed
436+
sub_stats_3 = ddb_audit_manager_threaded.get_submission_status(sub_3.submission_id)
437+
assert not sub_stats_3.validation_failed
438+
assert not sub_stats_3.processing_failed
439+
sub_stats_4 = ddb_audit_manager_threaded.get_submission_status(sub_4.submission_id)
440+
assert sub_stats_4.submission_result == "success"
441+
assert not sub_stats_4.validation_failed
442+
assert not sub_stats_4.processing_failed
443+
assert sub_stats_4.number_of_records == 20
444+
assert not ddb_audit_manager_threaded.get_submission_status("5")

0 commit comments

Comments
 (0)