From 44104d5942582273e1714fdb1d1e1d4cba45e30e Mon Sep 17 00:00:00 2001
From: Steven Silvester <steve.silvester@mongodb.com>
Date: Mon, 14 Oct 2024 06:32:48 -0500
Subject: [PATCH] PYTHON-4862 Fix handling of interrupt_loop in unified test
 runner

---
 test/asynchronous/unified_format.py | 8 +++++++-
 test/unified_format.py              | 8 +++++++-
 test/unified_format_shared.py       | 5 -----
 3 files changed, 14 insertions(+), 7 deletions(-)

diff --git a/test/asynchronous/unified_format.py b/test/asynchronous/unified_format.py
index 4c37422951..42bda59cb2 100644
--- a/test/asynchronous/unified_format.py
+++ b/test/asynchronous/unified_format.py
@@ -36,7 +36,6 @@
     unittest,
 )
 from test.unified_format_shared import (
-    IS_INTERRUPTED,
     KMS_TLS_OPTS,
     PLACEHOLDER_MAP,
     SKIP_CSOT_TESTS,
@@ -104,6 +103,13 @@
 
 _IS_SYNC = False
 
+IS_INTERRUPTED = False
+
+
+def interrupt_loop():
+    global IS_INTERRUPTED
+    IS_INTERRUPTED = True
+
 
 async def is_run_on_requirement_satisfied(requirement):
     topology_satisfied = True
diff --git a/test/unified_format.py b/test/unified_format.py
index 6a19082b86..13ab0af69b 100644
--- a/test/unified_format.py
+++ b/test/unified_format.py
@@ -36,7 +36,6 @@
     unittest,
 )
 from test.unified_format_shared import (
-    IS_INTERRUPTED,
     KMS_TLS_OPTS,
     PLACEHOLDER_MAP,
     SKIP_CSOT_TESTS,
@@ -104,6 +103,13 @@
 
 _IS_SYNC = True
 
+IS_INTERRUPTED = False
+
+
+def interrupt_loop():
+    global IS_INTERRUPTED
+    IS_INTERRUPTED = True
+
 
 def is_run_on_requirement_satisfied(requirement):
     topology_satisfied = True
diff --git a/test/unified_format_shared.py b/test/unified_format_shared.py
index d11624476d..f1b908a7a6 100644
--- a/test/unified_format_shared.py
+++ b/test/unified_format_shared.py
@@ -139,11 +139,6 @@
     }
 
 
-def interrupt_loop():
-    global IS_INTERRUPTED
-    IS_INTERRUPTED = True
-
-
 def with_metaclass(meta, *bases):
     """Create a base class with a metaclass.