Mock suppress_stderr_during

This commit is contained in:
Teagan Glenn 2026-01-31 08:09:01 -07:00 committed by Teagan glenn
parent c911b01246
commit 205f4a1a23

View File

@ -7,7 +7,7 @@ MOCK_MODULES = [
"frigate.data_processing.real_time.custom_classification.cv2",
"frigate.data_processing.real_time.custom_classification.load_labels",
"frigate.data_processing.real_time.custom_classification.write_classification_attempt",
"frigate.data_processing.real_time.custom_classification.log",
"frigate.data_processing.real_time.custom_classification.suppress_stderr_during",
]
AUTO_SPEC_MOCK_MODULES = [
"frigate.data_processing.real_time.custom_classification.Interpreter",
@ -240,13 +240,6 @@ class TestCustomObjectClassificationIntegration(unittest.TestCase):
# If imports fail, skip these tests (they need full Docker environment)
self.skipTest(f"Requires full Frigate environment: {e}")
def tearDown(self):
for mod in MOCK_MODULES:
if mod in ORIGINAL_MODULES:
sys.modules[mod] = mod
else:
del sys.modules[mod]
def test_process_frame_with_zones_includes_zones_in_mqtt(self):
"""
Integration test: Actually call process_frame() and verify zones in MQTT.