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/flow/maintenance.py b/jb/flow/maintenance.py
index 744ed1d..f8ca971 100644
--- a/jb/flow/maintenance.py
+++ b/jb/flow/maintenance.py
@@ -7,7 +7,10 @@ from jb.models.definitions import HitStatus
 
 
 def check_hit_status(
-    amt_hit_id: str, amt_hit_type_id: str, reason: Optional[str] = None
+    amtm: AMTManager,
+    amt_hit_id: str,
+    amt_hit_type_id: str,
+    reason: Optional[str] = None,
 ) -> HitStatus:
     """
     (this used to be called "process_hit")
@@ -15,7 +18,7 @@ def check_hit_status(
     the local state from that response.
     """
 
-    hit_status = AMTManager.get_hit_status(amt_hit_id=amt_hit_id)
+    hit_status = amtm.get_hit_status(amt_hit_id=amt_hit_id)
 
     # We're assuming that in the db this Hit is marked as Assignable, or
     # else we wouldn't have called this function.