Feature 10986: Autoheal switch and Autoscale switch

[osm/POL.git] / osm_policy_module / tests / integration / test_policy_agent.py

diff --git a/osm_policy_module/tests/integration/test_policy_agent.py b/osm_policy_module/tests/integration/test_policy_agent.py
index 0329b7c..e96c7e6 100644 (file)
--- a/osm_policy_module/tests/integration/test_policy_agent.py
+++ b/osm_policy_module/tests/integration/test_policy_agent.py
@@ -375,7 +375,7 @@ vnfd_record_mock = {
                             "recovery-type": "automatic",
                             "action-on-recovery": "REDEPLOY_ONLY",
                             "cooldown-time": 180,
-                            "day1": False
+                            "day1": False,
                         }
                     ],
                 }
@@ -513,7 +513,7 @@ class PolicyModuleAgentTest(unittest.TestCase):
         create_alarm.side_effect = _test_configure_scaling_groups_create_alarm
         create_alarm.assert_not_called_with = assert_not_called_with
         config = Config()
-        agent = PolicyModuleAgent(config, self.loop)
+        agent = PolicyModuleAgent(config)
         self.loop.run_until_complete(
             agent.autoscaling_service.configure_scaling_groups("test_nsr_id")
         )
@@ -581,7 +581,7 @@ class PolicyModuleAgentTest(unittest.TestCase):
         get_vnfd.return_value = vnfd_record_mock
         create_alarm.side_effect = _test_configure_vnf_alarms_create_alarm
         config = Config()
-        agent = PolicyModuleAgent(config, self.loop)
+        agent = PolicyModuleAgent(config)
         self.loop.run_until_complete(
             agent.alarming_service.configure_vnf_alarms("test_nsr_id")
         )
@@ -592,7 +592,7 @@ class PolicyModuleAgentTest(unittest.TestCase):
             vnf_member_index="1",
             threshold=20.0,
             operation="LT",
-            action="{'webhook': ['localhost:9090', 'localhost:9090', 'localhost:9090']}"
+            action="{'webhook': ['localhost:9090', 'localhost:9090', 'localhost:9090']}",
         )
         create_alarm.assert_any_call(
             metric_name="average_memory_utilization",
@@ -601,7 +601,7 @@ class PolicyModuleAgentTest(unittest.TestCase):
             vnf_member_index="2",
             threshold=20.0,
             operation="LT",
-            action="{'webhook': ['localhost:9090', 'localhost:9090', 'localhost:9090']}"
+            action="{'webhook': ['localhost:9090', 'localhost:9090', 'localhost:9090']}",
         )
 
     @patch.object(DbMongo, "db_connect", Mock())
@@ -628,7 +628,7 @@ class PolicyModuleAgentTest(unittest.TestCase):
         get_vnfd.return_value = vnfd_record_mock
         create_alarm.side_effect = _test_configure_healing_alarms_create_alarm
         config = Config()
-        agent = PolicyModuleAgent(config, self.loop)
+        agent = PolicyModuleAgent(config)
         self.loop.run_until_complete(
             agent.healing_service.configure_healing_alarms("test_nsr_id")
         )