diff options
Diffstat (limited to 'tests/fixtures')
| -rw-r--r-- | tests/fixtures/flow.py | 25 | ||||
| -rw-r--r-- | tests/fixtures/models.py | 4 |
2 files changed, 20 insertions, 9 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() diff --git a/tests/fixtures/models.py b/tests/fixtures/models.py index 03bd8a2..bdb4e12 100644 --- a/tests/fixtures/models.py +++ b/tests/fixtures/models.py @@ -242,7 +242,7 @@ def assignment_record( @pytest.fixture -def assignment_factory(hit: Hit) -> Callable[[Optional[str]], Assignment]: +def assignment_factory(hit_record: Hit) -> Callable[[Optional[str]], Assignment]: def _inner(amt_worker_id: Optional[str] = None) -> Assignment: now = datetime.now(tz=timezone.utc) @@ -251,7 +251,7 @@ def assignment_factory(hit: Hit) -> Callable[[Optional[str]], Assignment]: return Assignment( amt_assignment_id=amt_assignment_id, - amt_hit_id=hit.amt_hit_id, + amt_hit_id=hit_record.amt_hit_id, amt_worker_id=amt_worker_id, status=AssignmentStatus.Submitted, modified_at=now, |