X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=osm_policy_module%2Ftests%2Funit%2Fcore%2Ftest_policy_agent.py;h=77c285db6b534659bc805609c0e0b446e9735430;hb=4584f8e86a492d67d120bfea1195eff1475c0a65;hp=9ff6b45035b3e9fe57b1b74a7492ea95c3cfbcc5;hpb=e8ee171063d81a02033729273793f5ba2c2b71cc;p=osm%2FPOL.git

diff --git a/osm_policy_module/tests/unit/core/test_policy_agent.py b/osm_policy_module/tests/unit/core/test_policy_agent.py
index 9ff6b45..77c285d 100644
--- a/osm_policy_module/tests/unit/core/test_policy_agent.py
+++ b/osm_policy_module/tests/unit/core/test_policy_agent.py
@@ -37,23 +37,25 @@ class PolicyAgentTest(unittest.TestCase):
         self.loop = asyncio.new_event_loop()
 
     @mock.patch.object(CommonDbClient, "__init__", lambda *args, **kwargs: None)
-    @mock.patch('osm_policy_module.alarming.service.MonClient')
-    @mock.patch('osm_policy_module.alarming.service.LcmClient')
-    @mock.patch('osm_policy_module.autoscaling.service.MonClient')
-    @mock.patch('osm_policy_module.autoscaling.service.LcmClient')
-    @mock.patch.object(AutoscalingService, 'configure_scaling_groups')
-    @mock.patch.object(AlarmingService, 'configure_vnf_alarms')
-    @mock.patch.object(AutoscalingService, 'delete_orphaned_alarms')
-    @mock.patch.object(CommonDbClient, 'get_nslcmop')
-    def test_handle_instantiated(self,
-                                 get_nslcmop,
-                                 delete_orphaned_alarms,
-                                 configure_vnf_alarms,
-                                 configure_scaling_groups,
-                                 autoscaling_lcm_client,
-                                 autoscaling_mon_client,
-                                 alarming_lcm_client,
-                                 alarming_mon_client):
+    @mock.patch("osm_policy_module.alarming.service.MonClient")
+    @mock.patch("osm_policy_module.alarming.service.LcmClient")
+    @mock.patch("osm_policy_module.autoscaling.service.MonClient")
+    @mock.patch("osm_policy_module.autoscaling.service.LcmClient")
+    @mock.patch.object(AutoscalingService, "configure_scaling_groups")
+    @mock.patch.object(AlarmingService, "configure_vnf_alarms")
+    @mock.patch.object(AutoscalingService, "delete_orphaned_alarms")
+    @mock.patch.object(CommonDbClient, "get_nslcmop")
+    def test_handle_instantiated(
+        self,
+        get_nslcmop,
+        delete_orphaned_alarms,
+        configure_vnf_alarms,
+        configure_scaling_groups,
+        autoscaling_lcm_client,
+        autoscaling_mon_client,
+        alarming_lcm_client,
+        alarming_mon_client,
+    ):
         async def mock_configure_scaling_groups(nsr_id):
             pass
 
@@ -70,23 +72,20 @@ class PolicyAgentTest(unittest.TestCase):
         assert alarming_lcm_client.called
         assert alarming_mon_client.called
         content = {
-            'nslcmop_id': 'test_id',
+            "nslcmop_id": "test_id",
         }
         nslcmop_completed = {
-            'operationState': 'COMPLETED',
-            'nsInstanceId': 'test_nsr_id'
-        }
-        nslcmop_failed = {
-            'operationState': 'FAILED',
-            'nsInstanceId': 'test_nsr_id'
+            "operationState": "COMPLETED",
+            "nsInstanceId": "test_nsr_id",
         }
+        nslcmop_failed = {"operationState": "FAILED", "nsInstanceId": "test_nsr_id"}
         configure_scaling_groups.side_effect = mock_configure_scaling_groups
         configure_vnf_alarms.side_effect = mock_configure_vnf_alarms
         delete_orphaned_alarms.side_effect = mock_delete_orphaned_alarms
 
         get_nslcmop.return_value = nslcmop_completed
         self.loop.run_until_complete(agent._handle_instantiated(content))
-        configure_scaling_groups.assert_called_with('test_nsr_id')
+        configure_scaling_groups.assert_called_with("test_nsr_id")
         configure_scaling_groups.reset_mock()
 
         get_nslcmop.return_value = nslcmop_failed
@@ -94,19 +93,21 @@ class PolicyAgentTest(unittest.TestCase):
         configure_scaling_groups.assert_not_called()
 
     @mock.patch.object(CommonDbClient, "__init__", lambda *args, **kwargs: None)
-    @mock.patch('osm_policy_module.autoscaling.service.MonClient')
-    @mock.patch('osm_policy_module.autoscaling.service.LcmClient')
-    @mock.patch('osm_policy_module.alarming.service.MonClient')
-    @mock.patch('osm_policy_module.alarming.service.LcmClient')
-    @mock.patch.object(AutoscalingService, 'handle_alarm')
-    @mock.patch.object(AlarmingService, 'handle_alarm')
-    def test_handle_alarm_notification(self,
-                                       alarming_handle_alarm,
-                                       autoscaling_handle_alarm,
-                                       autoscaling_lcm_client,
-                                       autoscaling_mon_client,
-                                       alarming_lcm_client,
-                                       alarming_mon_client):
+    @mock.patch("osm_policy_module.autoscaling.service.MonClient")
+    @mock.patch("osm_policy_module.autoscaling.service.LcmClient")
+    @mock.patch("osm_policy_module.alarming.service.MonClient")
+    @mock.patch("osm_policy_module.alarming.service.LcmClient")
+    @mock.patch.object(AutoscalingService, "handle_alarm")
+    @mock.patch.object(AlarmingService, "handle_alarm")
+    def test_handle_alarm_notification(
+        self,
+        alarming_handle_alarm,
+        autoscaling_handle_alarm,
+        autoscaling_lcm_client,
+        autoscaling_mon_client,
+        alarming_lcm_client,
+        alarming_mon_client,
+    ):
         async def mock_handle_alarm(alarm_uuid, status, payload=None):
             pass
 
@@ -117,24 +118,24 @@ class PolicyAgentTest(unittest.TestCase):
         assert alarming_lcm_client.called
         assert alarming_mon_client.called
         content = {
-            'notify_details': {
-                'alarm_uuid': 'test_alarm_uuid',
-                'metric_name': 'test_metric_name',
-                'operation': 'test_operation',
-                'threshold_value': 'test_threshold_value',
-                'vdu_name': 'test_vdu_name',
-                'vnf_member_index': 'test_vnf_member_index',
-                'ns_id': 'test_nsr_id',
-                'status': 'alarm'
+            "notify_details": {
+                "alarm_uuid": "test_alarm_uuid",
+                "metric_name": "test_metric_name",
+                "operation": "test_operation",
+                "threshold_value": "test_threshold_value",
+                "vdu_name": "test_vdu_name",
+                "vnf_member_index": "test_vnf_member_index",
+                "ns_id": "test_nsr_id",
+                "status": "alarm",
             }
         }
         autoscaling_handle_alarm.side_effect = mock_handle_alarm
         alarming_handle_alarm.side_effect = mock_handle_alarm
 
         self.loop.run_until_complete(agent._handle_alarm_notification(content))
-        autoscaling_handle_alarm.assert_called_with('test_alarm_uuid', 'alarm')
-        alarming_handle_alarm.assert_called_with('test_alarm_uuid', 'alarm', content)
+        autoscaling_handle_alarm.assert_called_with("test_alarm_uuid", "alarm")
+        alarming_handle_alarm.assert_called_with("test_alarm_uuid", "alarm", content)
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     unittest.main()