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