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.

---
 jb/flow/maintenance.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

(limited to 'jb/flow')

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