check_hit_status to use manager param, and amt_stub_context usage in flow tests.

1 files changed, 5 insertions, 5 deletions

diff --git a/tests/fixtures/flow.py b/tests/fixtures/flow.py
index 633e05d..5ee2026 100644
--- a/tests/fixtures/flow.py
+++ b/tests/fixtures/flow.py
@@ -26,8 +26,8 @@ from generalresearchutils.models.thl.definitions import PayoutStatus
 
 @pytest.fixture
 def approved_assignment_stubs(
-    get_assignment_response: GetAssignmentResponseTypeDef,
-    get_assignment_response_approved: Callable[[str], GetAssignmentResponseTypeDef],
+    assignment_response: GetAssignmentResponseTypeDef,
+    assignment_response_approved: Callable[[str], GetAssignmentResponseTypeDef],
     amt_assignment_id: str,
     amt_hit_id: str,
     get_hit_response_reviewing: GetHITResponseTypeDef,
@@ -42,9 +42,9 @@ def approved_assignment_stubs(
         override_approve_response: Optional[str] = None,
     ) -> list[Dict[str, Any]]:
 
-        response = override_response or get_assignment_response
-        approve_response = (
-            override_approve_response or get_assignment_response_approved(feedback)
+        response = override_response or assignment_response
+        approve_response = override_approve_response or assignment_response_approved(
+            feedback
         )
 
         return [