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

1 files changed, 5 insertions, 2 deletions

diff --git a/jb/views/tasks.py b/jb/views/tasks.py
index 313295f..15857c3 100644
--- a/jb/views/tasks.py
+++ b/jb/views/tasks.py
@@ -2,7 +2,7 @@ from datetime import datetime, timezone, timedelta
 
 from fastapi import Request
 
-from jb.decorators import AM, HM
+from jb.decorators import AMTM, AM, HM
 from jb.flow.maintenance import check_hit_status
 from jb.flow.monitoring import emit_assignment_event
 from jb.models.assignment import AssignmentStub
@@ -29,7 +29,10 @@ def process_request(request: Request) -> None:
     if not hit:
         raise ValueError(f"Hit {amt_hit_id} not found in DB")
 
-    _ = check_hit_status(amt_hit_id=amt_hit_id, amt_hit_type_id=hit.amt_hit_type_id)
+    _ = check_hit_status(
+        amtm=AMTM, amt_hit_id=amt_hit_id, amt_hit_type_id=hit.amt_hit_type_id
+    )
+
     emit_assignment_event(
         status=AssignmentStatus.Accepted,
         amt_hit_type_id=hit.amt_hit_type_id,