diff options
Diffstat (limited to 'tests/flow/test_tasks.py')
| -rw-r--r-- | tests/flow/test_tasks.py | 17 |
1 files changed, 12 insertions, 5 deletions
diff --git a/tests/flow/test_tasks.py b/tests/flow/test_tasks.py index 2aeffb9..3a71504 100644 --- a/tests/flow/test_tasks.py +++ b/tests/flow/test_tasks.py @@ -227,8 +227,8 @@ class TestProcessAssignmentSubmitted: ..., GetAssignmentResponseTypeDef ], assignment_response_no_tsid: GetAssignmentResponseTypeDef, - assignment_factory: Callable[..., Assignment], - hit_record: Hit, + assignment_record_factory: Callable[..., Assignment], + hit_record: Hit, # used in assignment_factory amt_worker_id: str, ): # An assignment is submitted. The hit and assignment stub exist in the DB. @@ -237,9 +237,16 @@ class TestProcessAssignmentSubmitted: # Going to create and submit 3 assignments w no work # (all on the same hit, which we don't do in JB for real, # but doesn't matter here) - _a1 = assignment_factory(hit_id=hit_record.id, amt_worker_id=amt_worker_id) - _a2 = assignment_factory(hit_id=hit_record.id, amt_worker_id=amt_worker_id) - _a3 = assignment_factory(hit_id=hit_record.id, amt_worker_id=amt_worker_id) + _a1 = assignment_record_factory( + hit_id=hit_record.id, amt_worker_id=amt_worker_id + ) + _a2 = assignment_record_factory( + hit_id=hit_record.id, amt_worker_id=amt_worker_id + ) + _a3 = assignment_record_factory( + hit_id=hit_record.id, amt_worker_id=amt_worker_id + ) + assert am.missing_tsid_count(amt_worker_id=amt_worker_id) == 3 # So now, we'll reject, b/c they've already gotten 3 warnings |