From dadd9643feceac94a2b84d9a3d3dda667f17583d Mon Sep 17 00:00:00 2001
From: Max Nanis
Date: Thu, 26 Feb 2026 21:20:14 -0500
Subject: check_hit_status to use manager param, and amt_stub_context usage in
 flow tests.

---
 tests/fixtures/flow.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

(limited to 'tests/fixtures/flow.py')

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 [
-- 
cgit v1.2.3