tests.flow all pass. Now everything is GREEN ✅

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