tests.flow all pass. Now everything is GREEN ✅

1 files changed, 18 insertions, 7 deletions

diff --git a/tests/fixtures/flow.py b/tests/fixtures/flow.py
index 5ee2026..dd2f83e 100644
--- a/tests/fixtures/flow.py
+++ b/tests/fixtures/flow.py
@@ -27,10 +27,10 @@ from generalresearchutils.models.thl.definitions import PayoutStatus
 @pytest.fixture
 def approved_assignment_stubs(
     assignment_response: GetAssignmentResponseTypeDef,
-    assignment_response_approved: Callable[[str], GetAssignmentResponseTypeDef],
+    assignment_response_factory_approved: Callable[[str], GetAssignmentResponseTypeDef],
     amt_assignment_id: str,
     amt_hit_id: str,
-    get_hit_response_reviewing: GetHITResponseTypeDef,
+    hit_response_reviewing: GetHITResponseTypeDef,
 ) -> Callable[..., list[Dict[str, Any]]]:
 
     # These are the AMT_CLIENT stubs/mocks that need to be set when running
@@ -43,8 +43,8 @@ def approved_assignment_stubs(
     ) -> list[Dict[str, Any]]:
 
         response = override_response or assignment_response
-        approve_response = override_approve_response or assignment_response_approved(
-            feedback
+        approve_response = (
+            override_approve_response or assignment_response_factory_approved(feedback)
         )
 
         return [
@@ -74,7 +74,7 @@ def approved_assignment_stubs(
             },
             {
                 "operation": "get_hit",
-                "response": get_hit_response_reviewing,
+                "response": hit_response_reviewing,
                 "expected_params": {"HITId": amt_hit_id},
             },
         ]
@@ -218,7 +218,15 @@ def mock_thl_responses(
 
                 class MockThlProfileResponse:
                     def json(self):
-                        return {"user_profile": {"user": {"blocked": user_blocked}}}
+                        return {
+                            "user_profile": {
+                                "user": {
+                                    "product_id": settings.product_id,
+                                    "product_user_id": amt_worker_id,
+                                    "blocked": user_blocked,
+                                }
+                            }
+                        }
 
                 return MockThlProfileResponse()
 
@@ -227,7 +235,10 @@ def mock_thl_responses(
                 class MockThlWalletResponse:
                     def json(self) -> Dict[str, Any]:
                         return {
-                            "wallet": {"redeemable_amount": wallet_redeemable_amount}
+                            "wallet": {
+                                "amount": wallet_redeemable_amount,
+                                "redeemable_amount": wallet_redeemable_amount,
+                            }
                         }
 
                 return MockThlWalletResponse()