Feature 10986: Autoheal switch and Autoscale switch

[osm/POL.git] / osm_policy_module / tests / unit / healing / test_healing_service.py

diff --git a/osm_policy_module/tests/unit/healing/test_healing_service.py b/osm_policy_module/tests/unit/healing/test_healing_service.py
index 7f48f83..ae5fe62 100644 (file)
--- a/osm_policy_module/tests/unit/healing/test_healing_service.py
+++ b/osm_policy_module/tests/unit/healing/test_healing_service.py
@@ -52,18 +52,37 @@ class TestHealscalingService(TestCase):
         self.loop.run_until_complete(service.update_alarm_status("test_uuid", "alarm"))
         self.assertEqual(mock_alarm.last_status, "alarm")
         mock_alarm.save.assert_called_with()
-
         service = HealingService(self.config)
         self.loop.run_until_complete(service.update_alarm_status("test_uuid", "ok"))
         self.assertEqual(mock_alarm.last_status, "ok")
         mock_alarm.save.assert_called_with()
-
-        service = HealingService(self.config)
         self.loop.run_until_complete(
             service.update_alarm_status("test_uuid", "insufficient_data")
         )
         self.assertEqual(mock_alarm.last_status, "insufficient_data")
         mock_alarm.save.assert_called_with()
+        """
+        if self.config.get("autoheal", "enabled") == "True":
+            self.loop.run_until_complete(
+                service.update_alarm_status("test_uuid", "alarm")
+            )
+            self.assertEqual(mock_alarm.last_status, "alarm")
+            mock_alarm.save.assert_called_with()
+
+        service = HealingService(self.config)
+        if self.config.get("autoheal", "enabled") == "True":
+            self.loop.run_until_complete(service.update_alarm_status("test_uuid", "ok"))
+            self.assertEqual(mock_alarm.last_status, "ok")
+            mock_alarm.save.assert_called_with()
+
+        service = HealingService(self.config)
+        if self.config.get("autoheal", "enabled") == "True":
+            self.loop.run_until_complete(
+                service.update_alarm_status("test_uuid", "insufficient_data")
+            )
+            self.assertEqual(mock_alarm.last_status, "insufficient_data")
+            mock_alarm.save.assert_called_with()
+        """
 
     @mock.patch.object(HealingActionRepository, "list")
     @mock.patch.object(HealingActionRepository, "get")
@@ -84,6 +103,20 @@ class TestHealscalingService(TestCase):
             "test_day1",
             "test_count_index",
         )
+        """
+        if self.config.get("autoheal", "enabled") == "True":
+            self.loop.run_until_complete(service.handle_alarm("test_id", "alarm"))
+            heal.assert_called_with(
+                "test_nsr_id",
+                "test_vnfinstance_id",
+                "test_vdur_name",
+                "test_vdu_id",
+                "test_vnf_member_index",
+                "test_heal_type",
+                "test_day1",
+                "test_count_index",
+            )
+        """
 
     def _build_mock_alarm(
         self,