X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=osm_policy_module%2Ftests%2Fintegration%2Ftest_policy_agent.py;h=2c86b12796a5fee9b2e961e6565762870d7e3ccb;hb=4584f8e86a492d67d120bfea1195eff1475c0a65;hp=9fe5e2ad62758359329feb29c0891c1fd0f0580d;hpb=e8ee171063d81a02033729273793f5ba2c2b71cc;p=osm%2FPOL.git diff --git a/osm_policy_module/tests/integration/test_policy_agent.py b/osm_policy_module/tests/integration/test_policy_agent.py index 9fe5e2a..2c86b12 100644 --- a/osm_policy_module/tests/integration/test_policy_agent.py +++ b/osm_policy_module/tests/integration/test_policy_agent.py @@ -38,8 +38,14 @@ from osm_policy_module.common.mon_client import MonClient from osm_policy_module.core import database from osm_policy_module.core.agent import PolicyModuleAgent from osm_policy_module.core.config import Config -from osm_policy_module.core.database import ScalingGroup, ScalingAlarm, ScalingPolicy, ScalingCriteria, VnfAlarm, \ - AlarmAction +from osm_policy_module.core.database import ( + ScalingGroup, + ScalingAlarm, + ScalingPolicy, + ScalingCriteria, + VnfAlarm, + AlarmAction, +) log = logging.getLogger() log.level = logging.INFO @@ -56,9 +62,7 @@ nsr_record_mock = { "_id": "d7c8bd3c-eb39-4514-8847-19f01345524f", "_admin": { "created": 1535392246.499733, - "userDefinedData": { - - }, + "userDefinedData": {}, "usageSate": "NOT_IN_USE", "storage": { "zipfile": "package.tar.gz", @@ -66,66 +70,68 @@ nsr_record_mock = { "path": "/app/storage/", "folder": "d7c8bd3c-eb39-4514-8847-19f01345524f", "pkg-dir": "cirros_nsd", - "descriptor": "cirros_nsd/cirros_vdu_scaling_nsd.yaml" + "descriptor": "cirros_nsd/cirros_vdu_scaling_nsd.yaml", }, "onboardingState": "ONBOARDED", "modified": 1535392246.499733, - "projects_read": [ - "admin" - ], + "projects_read": ["admin"], "operationalState": "ENABLED", - "projects_write": [ - "admin" - ] + "projects_write": ["admin"], }, "id": "cirros_vdu_scaling_ns", "name": "cirros_vdu_scaling_ns", "description": "Simple NS example with a cirros_vdu_scaling_vnf", - "designer": "OSM", "version": "1.0", + "designer": "OSM", + "version": "1.0", "vnfd-id": ["cirros_vdu_scaling_vnf"], - "df": [{ - "id": "default-df", - "vnf-profile": [ - { - "id": "1", - "vnfd-id": "cirros_vdu_scaling_vnf", - "virtual-link-connectivity": [{ - "virtual-link-profile-id": "cirros_nsd_vld1", - "constituent-cpd-id": [{ - "constituent-base-element-id": "1", - "constituent-cpd-id": "eth0-ext" - }] - }] - }, { - "id": "2", - "vnfd-id": "cirros_vdu_scaling_vnf", - "virtual-link-connectivity": [{ - "virtual-link-profile-id": "cirros_nsd_vld1", - "constituent-cpd-id": [{ - "constituent-base-element-id": "2", - "constituent-cpd-id": "eth0-ext" - }] - }] - } - ] - }], - "virtual-link-desc": [{ - "id": "cirros_nsd_vld1", "mgmt-network": "true" - }] + "df": [ + { + "id": "default-df", + "vnf-profile": [ + { + "id": "1", + "vnfd-id": "cirros_vdu_scaling_vnf", + "virtual-link-connectivity": [ + { + "virtual-link-profile-id": "cirros_nsd_vld1", + "constituent-cpd-id": [ + { + "constituent-base-element-id": "1", + "constituent-cpd-id": "eth0-ext", + } + ], + } + ], + }, + { + "id": "2", + "vnfd-id": "cirros_vdu_scaling_vnf", + "virtual-link-connectivity": [ + { + "virtual-link-profile-id": "cirros_nsd_vld1", + "constituent-cpd-id": [ + { + "constituent-base-element-id": "2", + "constituent-cpd-id": "eth0-ext", + } + ], + } + ], + }, + ], + } + ], + "virtual-link-desc": [{"id": "cirros_nsd_vld1", "mgmt-network": "true"}], }, "id": "87776f33-b67c-417a-8119-cb08e4098951", "config-status": "configured", "operational-events": [], "_admin": { "created": 1535392482.0084584, - "projects_read": [ - "admin" - ], + "projects_read": ["admin"], "nsState": "INSTANTIATED", "modified": 1535392482.0084584, - "projects_write": [ - "admin" - ], + "projects_write": ["admin"], "deployed": { "RO": { "vnfd_id": { @@ -133,9 +139,9 @@ nsr_record_mock = { }, "nsd_id": "92c56cf0-f8fa-488c-9afb-9f3d78ae6bbb", "nsr_id": "637e12cd-c201-4c44-8ebd-70fb57a4dcee", - "nsr_status": "BUILD" + "nsr_status": "BUILD", } - } + }, }, "nsd-ref": "cirros_vdu_scaling_ns", "name": "cirros_ns", @@ -145,29 +151,21 @@ nsr_record_mock = { "nsdId": "d7c8bd3c-eb39-4514-8847-19f01345524f", "nsr_id": "87776f33-b67c-417a-8119-cb08e4098951", "nsName": "cirros_ns", - "vimAccountId": "be48ae31-1d46-4892-a4b4-d69abd55714b" + "vimAccountId": "be48ae31-1d46-4892-a4b4-d69abd55714b", }, "description": "default description", "constituent-vnfr-ref": [ "0d9d06ad-3fc2-418c-9934-465e815fafe2", - "3336eb44-77df-4c4f-9881-d2828d259864" + "3336eb44-77df-4c4f-9881-d2828d259864", ], "admin-status": "ENABLED", "detailed-status": "done", "datacenter": "be48ae31-1d46-4892-a4b4-d69abd55714b", - "orchestration-progress": { - - }, + "orchestration-progress": {}, "short-name": "cirros_ns", "ns-instance-config-ref": "87776f33-b67c-417a-8119-cb08e4098951", "nsd-name-ref": "cirros_vdu_scaling_ns", - "admin": { - "deployed": { - "RO": { - "nsr_status": "ACTIVE" - } - } - } + "admin": {"deployed": {"RO": {"nsr_status": "ACTIVE"}}}, } vnfr_record_mocks = [ @@ -182,7 +180,7 @@ vnfr_record_mocks = [ { "mac-address": "fa:16:3e:71:fd:b8", "name": "eth0", - "ip-address": "192.168.160.2" + "ip-address": "192.168.160.2", } ], "status": "ACTIVE", @@ -190,7 +188,7 @@ vnfr_record_mocks = [ "name": "cirros_ns-1-cirros_vnfd-VM-1", "status-detailed": None, "ip-address": "192.168.160.2", - "vdu-id-ref": "cirros_vnfd-VM" + "vdu-id-ref": "cirros_vnfd-VM", } ], "id": "0d9d06ad-3fc2-418c-9934-465e815fafe2", @@ -198,23 +196,13 @@ vnfr_record_mocks = [ "vnfd-id": "63f44c41-45ee-456b-b10d-5f08fb1796e0", "_admin": { "created": 1535392482.0067868, - "projects_read": [ - "admin" - ], + "projects_read": ["admin"], "modified": 1535392482.0067868, - "projects_write": [ - "admin" - ] + "projects_write": ["admin"], }, "nsr-id-ref": "87776f33-b67c-417a-8119-cb08e4098951", "member-vnf-index-ref": "1", - "connection-point": [ - { - "name": "eth0", - "id": None, - "connection-point-id": None - } - ] + "connection-point": [{"name": "eth0", "id": None, "connection-point-id": None}], }, { "_id": "3336eb44-77df-4c4f-9881-d2828d259864", @@ -227,7 +215,7 @@ vnfr_record_mocks = [ { "mac-address": "fa:16:3e:1e:76:e8", "name": "eth0", - "ip-address": "192.168.160.10" + "ip-address": "192.168.160.10", } ], "status": "ACTIVE", @@ -235,7 +223,7 @@ vnfr_record_mocks = [ "name": "cirros_ns-2-cirros_vnfd-VM-1", "status-detailed": None, "ip-address": "192.168.160.10", - "vdu-id-ref": "cirros_vnfd-VM" + "vdu-id-ref": "cirros_vnfd-VM", } ], "id": "3336eb44-77df-4c4f-9881-d2828d259864", @@ -243,189 +231,192 @@ vnfr_record_mocks = [ "vnfd-id": "63f44c41-45ee-456b-b10d-5f08fb1796e0", "_admin": { "created": 1535392482.0076294, - "projects_read": [ - "admin" - ], + "projects_read": ["admin"], "modified": 1535392482.0076294, - "projects_write": [ - "admin" - ] + "projects_write": ["admin"], }, "nsr-id-ref": "87776f33-b67c-417a-8119-cb08e4098951", "member-vnf-index-ref": "2", - "connection-point": [ - { - "name": "eth0", - "id": None, - "connection-point-id": None - } - ]}] + "connection-point": [{"name": "eth0", "id": None, "connection-point-id": None}], + }, +] nsd_record_mock = { - 'id': 'cirros_vdu_scaling_ns', - 'name': 'cirros_vdu_scaling_ns', - 'description': 'Simple NS example with a cirros_vdu_scaling_vnf', - 'designer': 'OSM', - 'version': '1.0', - 'vnfd-id': ['cirros_vdu_scaling_vnf'], - 'df': [{ - 'id': 'default-df', - 'vnf-profile': [ - { - 'id': '1', - 'vnfd-id': 'cirros_vdu_scaling_vnf', - 'virtual-link-connectivity': [{ - 'virtual-link-profile-id': 'cirros_nsd_vld1', - 'constituent-cpd-id': [{ - 'constituent-base-element-id': '1', - 'constituent-cpd-id': 'eth0-ext' - }] - }] - }, { - 'id': '2', - 'vnfd-id': 'cirros_vdu_scaling_vnf', - 'virtual-link-connectivity': [{ - 'virtual-link-profile-id': 'cirros_nsd_vld1', - 'constituent-cpd-id': [{ - 'constituent-base-element-id': '2', - 'constituent-cpd-id': 'eth0-ext' - }] - }] - } - ] - }], - 'virtual-link-desc': [{ - 'id': 'cirros_nsd_vld1', - 'mgmt-network': 'true' - }] + "id": "cirros_vdu_scaling_ns", + "name": "cirros_vdu_scaling_ns", + "description": "Simple NS example with a cirros_vdu_scaling_vnf", + "designer": "OSM", + "version": "1.0", + "vnfd-id": ["cirros_vdu_scaling_vnf"], + "df": [ + { + "id": "default-df", + "vnf-profile": [ + { + "id": "1", + "vnfd-id": "cirros_vdu_scaling_vnf", + "virtual-link-connectivity": [ + { + "virtual-link-profile-id": "cirros_nsd_vld1", + "constituent-cpd-id": [ + { + "constituent-base-element-id": "1", + "constituent-cpd-id": "eth0-ext", + } + ], + } + ], + }, + { + "id": "2", + "vnfd-id": "cirros_vdu_scaling_vnf", + "virtual-link-connectivity": [ + { + "virtual-link-profile-id": "cirros_nsd_vld1", + "constituent-cpd-id": [ + { + "constituent-base-element-id": "2", + "constituent-cpd-id": "eth0-ext", + } + ], + } + ], + }, + ], + } + ], + "virtual-link-desc": [{"id": "cirros_nsd_vld1", "mgmt-network": "true"}], } vnfd_record_mock = { - 'id': 'cirros_vdu_scaling_vnf', + "id": "cirros_vdu_scaling_vnf", "_id": "63f44c41-45ee-456b-b10d-5f08fb1796e0", - 'product-name': 'cirros_vdu_scaling_vnf', - 'description': 'Simple VNF example with a cirros and a scaling group descriptor', - 'provider': 'OSM', - 'version': '1.0', - 'mgmt-cp': 'eth0-ext', - 'virtual-storage-desc': [{ - 'id': 'cirros_vnfd-VM-storage', - 'size-of-storage': 2 - }], - 'virtual-compute-desc': [{ - 'id': 'cirros_vnfd-VM-compute', - 'virtual-cpu': { - 'num-virtual-cpu': 1 - }, - 'virtual-memory': { - 'size': 0.25 + "product-name": "cirros_vdu_scaling_vnf", + "description": "Simple VNF example with a cirros and a scaling group descriptor", + "provider": "OSM", + "version": "1.0", + "mgmt-cp": "eth0-ext", + "virtual-storage-desc": [{"id": "cirros_vnfd-VM-storage", "size-of-storage": 2}], + "virtual-compute-desc": [ + { + "id": "cirros_vnfd-VM-compute", + "virtual-cpu": {"num-virtual-cpu": 1}, + "virtual-memory": {"size": 0.25}, } - }], - 'sw-image-desc': [{ - 'id': 'cirros034', - 'name': 'cirros034', - 'image': 'cirros034' - }], - 'vdu': [{ - 'id': 'cirros_vnfd-VM', - 'description': 'cirros_vnfd-VM', - 'name': 'cirros_vnfd-VM', - 'alarm': [{ - 'value': 20.0, - 'actions': { - 'insufficient-data': [{ - 'url': 'localhost:9090' - }], - 'ok': [{ - 'url': 'localhost:9090' - }], - 'alarm': [{ - 'url': 'localhost:9090' - }] - }, - 'alarm-id': 'alarm-1', - 'operation': 'LT', - 'vnf-monitoring-param-ref': 'cirros_vnf_memory_util' - }], - 'sw-image-desc': 'cirros034', - 'virtual-compute-desc': 'cirros_vnfd-VM-compute', - 'virtual-storage-desc': ['cirros_vnfd-VM-storage'], - 'int-cpd': [{ - 'id': 'eth0-int', - 'virtual-network-interface-requirement': [{ - 'name': 'eth0', - 'virtual-interface': { - 'bandwidth': '0', - 'type': 'VIRTIO', - 'vpci': '0000:00:0a.0'} - }] - }], - 'monitoring-parameter': [{ - 'id': 'cirros_vnf_memory_util', - 'name': 'cirros_vnf_memory_util', - 'performance-metric': 'average_memory_utilization' - }] - }], - 'df': [{ - 'id': 'default-df', - 'vdu-profile': [{ - 'id': 'cirros_vnfd-VM', - 'min-number-of-instances': 1, - 'max-number-of-instances': 10, - 'vdu-configuration-id': 'cirros_vnfd-VM-vdu-configuration' - }], - 'instantiation-level': [{ - 'id': 'default-instantiation-level', - 'vdu-level': [{ - 'vdu-id': 'cirros_vnfd-VM', - 'number-of-instances': 1}] - }], - 'scaling-aspect': [{ - 'id': 'scale_cirros_vnfd-VM', - 'name': 'scale_cirros_vnfd-VM', - 'max-scale-level': 10, - 'scaling-policy': [{ - 'name': 'auto_memory_util_above_threshold', - 'scaling-type': 'automatic', - 'cooldown-time': 60, - 'threshold-time': 10, - 'scaling-criteria': [{ - 'name': 'group1_memory_util_above_threshold', - 'vnf-monitoring-param-ref': 'cirros_vnf_memory_util', - 'scale-out-threshold': 80, - 'scale-out-relational-operation': 'GT', - 'scale-in-relational-operation': 'LT', - 'scale-in-threshold': 20 - }] - }], - 'aspect-delta-details': { - 'deltas': [{ - 'id': 'scale_cirros_vnfd-VM-delta', - 'vdu-delta': [{ - 'number-of-instances': 1, - 'id': 'cirros_vnfd-VM'}] - }] - } - }] - }], - 'ext-cpd': [{ - 'id': 'eth0-ext', - 'int-cpd': { - 'vdu-id': 'cirros_vnfd-VM', - 'cpd': 'eth0-int' + ], + "sw-image-desc": [{"id": "cirros034", "name": "cirros034", "image": "cirros034"}], + "vdu": [ + { + "id": "cirros_vnfd-VM", + "description": "cirros_vnfd-VM", + "name": "cirros_vnfd-VM", + "alarm": [ + { + "value": 20.0, + "actions": { + "insufficient-data": [{"url": "localhost:9090"}], + "ok": [{"url": "localhost:9090"}], + "alarm": [{"url": "localhost:9090"}], + }, + "alarm-id": "alarm-1", + "operation": "LT", + "vnf-monitoring-param-ref": "cirros_vnf_memory_util", + } + ], + "sw-image-desc": "cirros034", + "virtual-compute-desc": "cirros_vnfd-VM-compute", + "virtual-storage-desc": ["cirros_vnfd-VM-storage"], + "int-cpd": [ + { + "id": "eth0-int", + "virtual-network-interface-requirement": [ + { + "name": "eth0", + "virtual-interface": { + "bandwidth": "0", + "type": "VIRTIO", + "vpci": "0000:00:0a.0", + }, + } + ], + } + ], + "monitoring-parameter": [ + { + "id": "cirros_vnf_memory_util", + "name": "cirros_vnf_memory_util", + "performance-metric": "average_memory_utilization", + } + ], } - }], - 'vdu-configuration': [{ - 'juju': { - 'charm': 'testmetrics', - 'proxy': True - }, - 'metrics': [{ - 'name': 'users' - }], - 'id': 'cirros_vnfd-VM-vdu-configuration' - }], + ], + "df": [ + { + "id": "default-df", + "vdu-profile": [ + { + "id": "cirros_vnfd-VM", + "min-number-of-instances": 1, + "max-number-of-instances": 10, + "vdu-configuration-id": "cirros_vnfd-VM-vdu-configuration", + } + ], + "instantiation-level": [ + { + "id": "default-instantiation-level", + "vdu-level": [ + {"vdu-id": "cirros_vnfd-VM", "number-of-instances": 1} + ], + } + ], + "scaling-aspect": [ + { + "id": "scale_cirros_vnfd-VM", + "name": "scale_cirros_vnfd-VM", + "max-scale-level": 10, + "scaling-policy": [ + { + "name": "auto_memory_util_above_threshold", + "scaling-type": "automatic", + "cooldown-time": 60, + "threshold-time": 10, + "scaling-criteria": [ + { + "name": "group1_memory_util_above_threshold", + "vnf-monitoring-param-ref": "cirros_vnf_memory_util", + "scale-out-threshold": 80, + "scale-out-relational-operation": "GT", + "scale-in-relational-operation": "LT", + "scale-in-threshold": 20, + } + ], + } + ], + "aspect-delta-details": { + "deltas": [ + { + "id": "scale_cirros_vnfd-VM-delta", + "vdu-delta": [ + {"number-of-instances": 1, "id": "cirros_vnfd-VM"} + ], + } + ] + }, + } + ], + } + ], + "ext-cpd": [ + {"id": "eth0-ext", "int-cpd": {"vdu-id": "cirros_vnfd-VM", "cpd": "eth0-int"}} + ], + "vdu-configuration": [ + { + "juju": {"charm": "testmetrics", "proxy": True}, + "metrics": [{"name": "users"}], + "id": "cirros_vnfd-VM-vdu-configuration", + } + ], "_admin": { "created": 1535392242.6281035, "modified": 1535392242.6281035, @@ -435,30 +426,31 @@ vnfd_record_mock = { "path": "/app/storage/", "folder": "63f44c41-45ee-456b-b10d-5f08fb1796e0", "fs": "local", - "descriptor": "cirros_vnf/cirros_vdu_scaling_vnfd.yaml" + "descriptor": "cirros_vnf/cirros_vdu_scaling_vnfd.yaml", }, "usageSate": "NOT_IN_USE", "onboardingState": "ONBOARDED", - "userDefinedData": { - - }, - "projects_read": [ - "admin" - ], + "userDefinedData": {}, + "projects_read": ["admin"], "operationalState": "ENABLED", - "projects_write": [ - "admin" - ] - } + "projects_write": ["admin"], + }, } -MODELS = [ScalingGroup, ScalingPolicy, ScalingCriteria, ScalingAlarm, VnfAlarm, AlarmAction] +MODELS = [ + ScalingGroup, + ScalingPolicy, + ScalingCriteria, + ScalingAlarm, + VnfAlarm, + AlarmAction, +] class PolicyModuleAgentTest(unittest.TestCase): def setUp(self): super() - database.db.initialize(connect('sqlite:///test_db.sqlite')) + database.db.initialize(connect("sqlite:///test_db.sqlite")) database.db.bind(MODELS) database.db.connect() database.db.drop_tables(MODELS) @@ -468,19 +460,21 @@ class PolicyModuleAgentTest(unittest.TestCase): def tearDown(self): super() - os.remove('test_db.sqlite') + os.remove("test_db.sqlite") - @patch.object(DbMongo, 'db_connect', Mock()) - @patch.object(KafkaProducer, '__init__') - @patch.object(MonClient, 'create_alarm') - @patch.object(CommonDbClient, 'get_vnfd') - @patch.object(CommonDbClient, 'get_nsr') - @patch.object(CommonDbClient, 'get_vnfr') - def test_configure_scaling_groups(self, get_vnfr, get_nsr, get_vnfd, create_alarm, kafka_producer_init): + @patch.object(DbMongo, "db_connect", Mock()) + @patch.object(KafkaProducer, "__init__") + @patch.object(MonClient, "create_alarm") + @patch.object(CommonDbClient, "get_vnfd") + @patch.object(CommonDbClient, "get_nsr") + @patch.object(CommonDbClient, "get_vnfr") + def test_configure_scaling_groups( + self, get_vnfr, get_nsr, get_vnfd, create_alarm, kafka_producer_init + ): def _test_configure_scaling_groups_get_vnfr(*args, **kwargs): - if '1' in args[1]: + if "1" in args[1]: return vnfr_record_mocks[0] - if '2' in args[1]: + if "2" in args[1]: return vnfr_record_mocks[1] def assert_not_called_with(*args, **kwargs): @@ -488,7 +482,7 @@ class PolicyModuleAgentTest(unittest.TestCase): create_alarm.assert_called_with(*args, **kwargs) except AssertionError: return - raise AssertionError('Expected to not have been called.') + raise AssertionError("Expected to not have been called.") async def _test_configure_scaling_groups_create_alarm(*args, **kwargs): return uuid.uuid4() @@ -501,46 +495,58 @@ class PolicyModuleAgentTest(unittest.TestCase): create_alarm.assert_not_called_with = assert_not_called_with config = Config() agent = PolicyModuleAgent(config, self.loop) - self.loop.run_until_complete(agent.autoscaling_service.configure_scaling_groups("test_nsr_id")) - create_alarm.assert_any_call(metric_name='average_memory_utilization', - ns_id='test_nsr_id', - operation='GT', - threshold=80, - vdu_name='cirros_ns-1-cirros_vnfd-VM-1', - vnf_member_index='1') - create_alarm.assert_not_called_with(metric_name='average_memory_utilization', - ns_id='test_nsr_id', - operation='LT', - threshold=20, - vdu_name='cirros_ns-1-cirros_vnfd-VM-1', - vnf_member_index='1') - create_alarm.assert_any_call(metric_name='average_memory_utilization', - ns_id='test_nsr_id', - operation='GT', - threshold=80, - vdu_name='cirros_ns-2-cirros_vnfd-VM-1', - vnf_member_index='2') - create_alarm.assert_not_called_with(metric_name='average_memory_utilization', - ns_id='test_nsr_id', - operation='LT', - threshold=20, - vdu_name='cirros_ns-2-cirros_vnfd-VM-1', - vnf_member_index='2') + self.loop.run_until_complete( + agent.autoscaling_service.configure_scaling_groups("test_nsr_id") + ) + create_alarm.assert_any_call( + metric_name="average_memory_utilization", + ns_id="test_nsr_id", + operation="GT", + threshold=80, + vdu_name="cirros_ns-1-cirros_vnfd-VM-1", + vnf_member_index="1", + ) + create_alarm.assert_not_called_with( + metric_name="average_memory_utilization", + ns_id="test_nsr_id", + operation="LT", + threshold=20, + vdu_name="cirros_ns-1-cirros_vnfd-VM-1", + vnf_member_index="1", + ) + create_alarm.assert_any_call( + metric_name="average_memory_utilization", + ns_id="test_nsr_id", + operation="GT", + threshold=80, + vdu_name="cirros_ns-2-cirros_vnfd-VM-1", + vnf_member_index="2", + ) + create_alarm.assert_not_called_with( + metric_name="average_memory_utilization", + ns_id="test_nsr_id", + operation="LT", + threshold=20, + vdu_name="cirros_ns-2-cirros_vnfd-VM-1", + vnf_member_index="2", + ) scaling_record = ScalingGroup.get() - self.assertEqual(scaling_record.name, 'scale_cirros_vnfd-VM') - self.assertEqual(scaling_record.nsr_id, 'test_nsr_id') + self.assertEqual(scaling_record.name, "scale_cirros_vnfd-VM") + self.assertEqual(scaling_record.nsr_id, "test_nsr_id") - @patch.object(DbMongo, 'db_connect', Mock()) - @patch.object(KafkaProducer, '__init__') - @patch.object(MonClient, 'create_alarm') - @patch.object(CommonDbClient, 'get_vnfd') - @patch.object(CommonDbClient, 'get_nsr') - @patch.object(CommonDbClient, 'get_vnfr') - def test_configure_vnf_alarms(self, get_vnfr, get_nsr, get_vnfd, create_alarm, kafka_producer_init): + @patch.object(DbMongo, "db_connect", Mock()) + @patch.object(KafkaProducer, "__init__") + @patch.object(MonClient, "create_alarm") + @patch.object(CommonDbClient, "get_vnfd") + @patch.object(CommonDbClient, "get_nsr") + @patch.object(CommonDbClient, "get_vnfr") + def test_configure_vnf_alarms( + self, get_vnfr, get_nsr, get_vnfd, create_alarm, kafka_producer_init + ): def _test_configure_scaling_groups_get_vnfr(*args, **kwargs): - if '1' in args[1]: + if "1" in args[1]: return vnfr_record_mocks[0] - if '2' in args[1]: + if "2" in args[1]: return vnfr_record_mocks[1] async def _test_configure_vnf_alarms_create_alarm(*args, **kwargs): @@ -553,20 +559,26 @@ class PolicyModuleAgentTest(unittest.TestCase): create_alarm.side_effect = _test_configure_vnf_alarms_create_alarm config = Config() agent = PolicyModuleAgent(config, self.loop) - self.loop.run_until_complete(agent.alarming_service.configure_vnf_alarms("test_nsr_id")) - create_alarm.assert_any_call(metric_name='average_memory_utilization', - ns_id='test_nsr_id', - vdu_name='cirros_ns-1-cirros_vnfd-VM-1', - vnf_member_index='1', - threshold=20.0, - operation='LT') - create_alarm.assert_any_call(metric_name='average_memory_utilization', - ns_id='test_nsr_id', - vdu_name='cirros_ns-2-cirros_vnfd-VM-1', - vnf_member_index='2', - threshold=20.0, - operation='LT') + self.loop.run_until_complete( + agent.alarming_service.configure_vnf_alarms("test_nsr_id") + ) + create_alarm.assert_any_call( + metric_name="average_memory_utilization", + ns_id="test_nsr_id", + vdu_name="cirros_ns-1-cirros_vnfd-VM-1", + vnf_member_index="1", + threshold=20.0, + operation="LT", + ) + create_alarm.assert_any_call( + metric_name="average_memory_utilization", + ns_id="test_nsr_id", + vdu_name="cirros_ns-2-cirros_vnfd-VM-1", + vnf_member_index="2", + threshold=20.0, + operation="LT", + ) -if __name__ == '__main__': +if __name__ == "__main__": unittest.main()