From f3f6a68abab8fa6366f912d37a826a77a3b1c3b4 Mon Sep 17 00:00:00 2001 From: Carlos Natalino <carlos.natalino@chalmers.se> Date: Mon, 5 Feb 2024 14:12:43 +0000 Subject: [PATCH] Replacing the custom mock by a standard one. --- src/opticalattackdetector/tests/test_unitary.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/opticalattackdetector/tests/test_unitary.py b/src/opticalattackdetector/tests/test_unitary.py index 6c7579238..ecd59b917 100644 --- a/src/opticalattackdetector/tests/test_unitary.py +++ b/src/opticalattackdetector/tests/test_unitary.py @@ -58,12 +58,12 @@ def test_detect_attack( message = dbscan.DetectionResponse() message.cluster_indices.extend([0, 1, -1, -1, -1]) - monitoring_mock = MockServicerImpl_Monitoring(queue_samples = queue.Queue()) + # monitoring_mock = MockServicerImpl_Monitoring(queue_samples = queue.Queue()) + # TODO: remove the monitoring mock with patch( "opticalattackdetector.service.OpticalAttackDetectorServiceServicerImpl.attack_mitigator_client" ) as mitigator, patch( - "opticalattackdetector.service.OpticalAttackDetectorServiceServicerImpl.monitoring_client", - monitoring_mock, + "opticalattackdetector.service.OpticalAttackDetectorServiceServicerImpl.monitoring_client.IncludeKpi", ) as monitoring, patch( "opticalattackdetector.service.OpticalAttackDetectorServiceServicerImpl.dbscanserving_client.Detect", # TODO: return dumb object with "cluster_indices" attribute @@ -77,5 +77,5 @@ def test_detect_attack( request.kpi_id.kpi_id.uuid = "1" optical_attack_detector_client.DetectAttack(request) dbscanserving.assert_called_once() - monitoring.IncludeKpi.assert_called_once() + monitoring.assert_called_once() mitigator.NotifyAttack.assert_called() -- GitLab