Migrates POL code from MON repo
Adds support for VDU metric autoscaling
Modifies env var names
Signed-off-by: Benjamin Diaz <bdiaz@whitestack.com>
Change-Id: If9587e1b8eacaf6fb297306050a97d33c8a63ead
diff --git a/osm_policy_module/tests/__init__.py b/osm_policy_module/tests/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/osm_policy_module/tests/__init__.py
diff --git a/osm_policy_module/tests/examples/cirros_vdu_scaling_nsd.yaml b/osm_policy_module/tests/examples/cirros_vdu_scaling_nsd.yaml
new file mode 100644
index 0000000..a9e0abd
--- /dev/null
+++ b/osm_policy_module/tests/examples/cirros_vdu_scaling_nsd.yaml
@@ -0,0 +1,42 @@
+nsd:nsd-catalog:
+ nsd:
+ - id: cirros_vdu_scaling_ns
+ name: cirros_vdu_scaling_ns
+ short-name: cirros_vdu_scaling_ns
+ description: Simple NS example with a cirros_vdu_scaling_vnf
+ vendor: OSM
+ version: '1.0'
+
+ # Place the logo as png in icons directory and provide the name here
+ logo: osm.png
+
+ # Specify the VNFDs that are part of this NSD
+ constituent-vnfd:
+ # The member-vnf-index needs to be unique, starting from 1
+ # vnfd-id-ref is the id of the VNFD
+ # Multiple constituent VNFDs can be specified
+ - member-vnf-index: '1'
+ vnfd-id-ref: cirros_vdu_scaling_vnf
+ - member-vnf-index: '2'
+ vnfd-id-ref: cirros_vdu_scaling_vnf
+ vld:
+ # Networks for the VNFs
+ - id: cirros_nsd_vld1
+ name: cirros_nsd_vld1
+ type: ELAN
+ mgmt-network: 'true'
+ # vim-network-name: <update>
+ # provider-network:
+ # segmentation_id: <update>
+ vnfd-connection-point-ref:
+ # Specify the constituent VNFs
+ # member-vnf-index-ref - entry from constituent vnf
+ # vnfd-id-ref - VNFD id
+ # vnfd-connection-point-ref - connection point name in the VNFD
+ - member-vnf-index-ref: 1
+ vnfd-id-ref: cirros_vdu_scaling_vnf
+ # NOTE: Validate the entry below
+ vnfd-connection-point-ref: eth0
+ - member-vnf-index-ref: 2
+ vnfd-id-ref: cirros_vdu_scaling_vnf
+ vnfd-connection-point-ref: eth0
\ No newline at end of file
diff --git a/osm_policy_module/tests/examples/cirros_vdu_scaling_vnfd.yaml b/osm_policy_module/tests/examples/cirros_vdu_scaling_vnfd.yaml
new file mode 100644
index 0000000..dc599f5
--- /dev/null
+++ b/osm_policy_module/tests/examples/cirros_vdu_scaling_vnfd.yaml
@@ -0,0 +1,74 @@
+vnfd:vnfd-catalog:
+ vnfd:
+ - id: cirros_vdu_scaling_vnf
+ name: cirros_vdu_scaling_vnf
+ short-name: cirros_vdu_scaling_vnf
+ description: Simple VNF example with a cirros and a scaling group descriptor
+ vendor: OSM
+ version: '1.0'
+ # Place the logo as png in icons directory and provide the name here
+ logo: cirros-64.png
+ # Management interface
+ mgmt-interface:
+ cp: eth0
+ # Atleast one VDU need to be specified
+ vdu:
+ - id: cirros_vnfd-VM
+ name: cirros_vnfd-VM
+ description: cirros_vnfd-VM
+ count: 1
+
+ # Flavour of the VM to be instantiated for the VDU
+ # flavor below can fit into m1.micro
+ vm-flavor:
+ vcpu-count: 1
+ memory-mb: 256
+ storage-gb: 2
+ # Image/checksum or image including the full path
+ image: 'cirros034'
+ #checksum:
+ interface:
+ # Specify the external interfaces
+ # There can be multiple interfaces defined
+ - name: eth0
+ type: EXTERNAL
+ virtual-interface:
+ type: VIRTIO
+ bandwidth: '0'
+ vpci: 0000:00:0a.0
+ external-connection-point-ref: eth0
+ monitoring-param:
+ - id: "cirros_vnfd-VM_memory_util"
+ nfvi-metric: "average_memory_utilization" # The associated NFVI metric to be monitored. Id of the metric
+ #interface-name-ref: reference to interface name, required for some metrics
+ connection-point:
+ - name: eth0
+ type: VPORT
+ scaling-group-descriptor:
+ - name: "scale_cirros_vnfd-VM"
+ min-instance-count: 1
+ max-instance-count: 10
+ scaling-policy:
+ - name: "auto_memory_util_above_threshold"
+ scaling-type: "automatic"
+ threshold-time: 10
+ cooldown-time: 60
+ scaling-criteria:
+ - name: "group1_memory_util_above_threshold"
+ scale-in-threshold: 20
+ scale-in-relational-operation: "LT"
+ scale-out-threshold: 80
+ scale-out-relational-operation: "GT"
+ vnf-monitoring-param-ref: "cirros_vnf_memory_util"
+ vdu:
+ - vdu-id-ref: cirros_vnfd-VM
+ count: 1
+ # scaling-config-action: # Para utilizar charms
+ # - trigger: post-scale-out
+ # vnf-config-primitive-name-ref:
+ monitoring-param:
+ - id: "cirros_vnf_memory_util"
+ name: "cirros_vnf_memory_util"
+ aggregation-type: AVERAGE
+ vdu-ref: "cirros_vnfd-VM"
+ vdu-monitoring-param-ref: "cirros_vnfd-VM_memory_util"
\ No newline at end of file
diff --git a/osm_policy_module/tests/examples/instantiated.json b/osm_policy_module/tests/examples/instantiated.json
new file mode 100644
index 0000000..88b610d
--- /dev/null
+++ b/osm_policy_module/tests/examples/instantiated.json
@@ -0,0 +1,4 @@
+{
+ "nsr_id": "test_nsr_id",
+ "nslcmop_id": "test_nslcmop_id"
+}
\ No newline at end of file
diff --git a/osm_policy_module/tests/integration/__init__.py b/osm_policy_module/tests/integration/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/osm_policy_module/tests/integration/__init__.py
diff --git a/osm_policy_module/tests/integration/test_kafka_messages.py b/osm_policy_module/tests/integration/test_kafka_messages.py
new file mode 100644
index 0000000..d17d2b2
--- /dev/null
+++ b/osm_policy_module/tests/integration/test_kafka_messages.py
@@ -0,0 +1,79 @@
+# -*- coding: utf-8 -*-
+
+# Copyright 2018 Whitestack, LLC
+# *************************************************************
+
+# This file is part of OSM Monitoring module
+# All Rights Reserved to Whitestack, LLC
+
+# Licensed under the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License. You may obtain
+# a copy of the License at
+
+# http://www.apache.org/licenses/LICENSE-2.0
+
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+# License for the specific language governing permissions and limitations
+# under the License.
+
+# For those usages not covered by the Apache License, Version 2.0 please
+# contact: bdiaz@whitestack.com or glavado@whitestack.com
+##
+import json
+import logging
+import os
+import sys
+import unittest
+
+from kafka import KafkaProducer, KafkaConsumer
+from kafka.errors import KafkaError
+
+from osm_policy_module.core.agent import PolicyModuleAgent
+from osm_policy_module.core.config import Config
+
+log = logging.getLogger()
+log.level = logging.INFO
+stream_handler = logging.StreamHandler(sys.stdout)
+log.addHandler(stream_handler)
+
+
+class KafkaMessagesTest(unittest.TestCase):
+ def setUp(self):
+ try:
+ cfg = Config.instance()
+ kafka_server = '{}:{}'.format(cfg.OSMPOL_MESSAGE_HOST,
+ cfg.OSMPOL_MESSAGE_PORT)
+ self.producer = KafkaProducer(bootstrap_servers=kafka_server,
+ key_serializer=str.encode,
+ value_serializer=str.encode)
+ self.consumer = KafkaConsumer(bootstrap_servers=kafka_server,
+ key_deserializer=bytes.decode,
+ value_deserializer=bytes.decode,
+ auto_offset_reset='earliest',
+ consumer_timeout_ms=5000)
+ self.consumer.subscribe(['ns'])
+ except KafkaError:
+ self.skipTest('Kafka server not present.')
+
+ def tearDown(self):
+ self.producer.close()
+ self.consumer.close()
+
+ def test_send_instantiated_msg(self):
+ with open(
+ os.path.join(os.path.dirname(__file__), '../examples/instantiated.json')) as file:
+ payload = json.load(file)
+ self.producer.send('ns', json.dumps(payload), key="instantiated")
+ self.producer.flush()
+
+ for message in self.consumer:
+ if message.key == 'instantiated':
+ self.assertIsNotNone(message.value)
+ return
+ self.fail("No message received in consumer")
+
+
+if __name__ == '__main__':
+ unittest.main()
diff --git a/osm_policy_module/tests/integration/test_policy_agent.py b/osm_policy_module/tests/integration/test_policy_agent.py
new file mode 100644
index 0000000..4de0dbb
--- /dev/null
+++ b/osm_policy_module/tests/integration/test_policy_agent.py
@@ -0,0 +1,456 @@
+# -*- coding: utf-8 -*-
+
+# Copyright 2018 Whitestack, LLC
+# *************************************************************
+
+# This file is part of OSM Monitoring module
+# All Rights Reserved to Whitestack, LLC
+
+# Licensed under the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License. You may obtain
+# a copy of the License at
+
+# http://www.apache.org/licenses/LICENSE-2.0
+
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+# License for the specific language governing permissions and limitations
+# under the License.
+
+# For those usages not covered by the Apache License, Version 2.0 please
+# contact: bdiaz@whitestack.com or glavado@whitestack.com
+##
+import logging
+import sys
+import unittest
+import uuid
+from unittest.mock import patch, Mock
+
+from kafka import KafkaProducer
+from osm_common.dbmongo import DbMongo
+from osm_policy_module.core import database
+from peewee import SqliteDatabase
+
+from osm_policy_module.common.mon_client import MonClient
+from osm_policy_module.core.agent import PolicyModuleAgent
+from osm_policy_module.core.database import ScalingRecord, ScalingAlarm, BaseModel
+
+log = logging.getLogger()
+log.level = logging.INFO
+stream_handler = logging.StreamHandler(sys.stdout)
+log.addHandler(stream_handler)
+
+nsr_record_mock = {
+ "_id": "87776f33-b67c-417a-8119-cb08e4098951",
+ "crete-time": 1535392482.0044956,
+ "operational-status": "running",
+ "ssh-authorized-key": None,
+ "name-ref": "cirros_ns",
+ "nsd": {
+ "name": "cirros_vdu_scaling_ns",
+ "_id": "d7c8bd3c-eb39-4514-8847-19f01345524f",
+ "vld": [
+ {
+ "id": "cirros_nsd_vld1",
+ "name": "cirros_nsd_vld1",
+ "type": "ELAN",
+ "mgmt-network": "true",
+ "vnfd-connection-point-ref": [
+ {
+ "vnfd-id-ref": "cirros_vdu_scaling_vnf",
+ "member-vnf-index-ref": 1,
+ "vnfd-connection-point-ref": "eth0"
+ },
+ {
+ "vnfd-id-ref": "cirros_vdu_scaling_vnf",
+ "member-vnf-index-ref": 2,
+ "vnfd-connection-point-ref": "eth0"
+ }
+ ]
+ }
+ ],
+ "vendor": "OSM",
+ "constituent-vnfd": [
+ {
+ "member-vnf-index": "1",
+ "vnfd-id-ref": "cirros_vdu_scaling_vnf"
+ },
+ {
+ "member-vnf-index": "2",
+ "vnfd-id-ref": "cirros_vdu_scaling_vnf"
+ }
+ ],
+ "version": "1.0",
+ "id": "cirros_vdu_scaling_ns",
+ "description": "Simple NS example with a cirros_vdu_scaling_vnf",
+ "logo": "osm.png",
+ "_admin": {
+ "created": 1535392246.499733,
+ "userDefinedData": {
+
+ },
+ "usageSate": "NOT_IN_USE",
+ "storage": {
+ "zipfile": "package.tar.gz",
+ "fs": "local",
+ "path": "/app/storage/",
+ "folder": "d7c8bd3c-eb39-4514-8847-19f01345524f",
+ "pkg-dir": "cirros_nsd",
+ "descriptor": "cirros_nsd/cirros_vdu_scaling_nsd.yaml"
+ },
+ "onboardingState": "ONBOARDED",
+ "modified": 1535392246.499733,
+ "projects_read": [
+ "admin"
+ ],
+ "operationalState": "ENABLED",
+ "projects_write": [
+ "admin"
+ ]
+ },
+ "short-name": "cirros_vdu_scaling_ns"
+ },
+ "id": "87776f33-b67c-417a-8119-cb08e4098951",
+ "config-status": "configured",
+ "operational-events": [],
+ "_admin": {
+ "created": 1535392482.0084584,
+ "projects_read": [
+ "admin"
+ ],
+ "nsState": "INSTANTIATED",
+ "modified": 1535392482.0084584,
+ "projects_write": [
+ "admin"
+ ],
+ "deployed": {
+ "RO": {
+ "vnfd_id": {
+ "cirros_vdu_scaling_vnf": "7445e347-fe2f-431a-abc2-8b9be3d093c6"
+ },
+ "nsd_id": "92c56cf0-f8fa-488c-9afb-9f3d78ae6bbb",
+ "nsr_id": "637e12cd-c201-4c44-8ebd-70fb57a4dcee",
+ "nsr_status": "BUILD"
+ }
+ }
+ },
+ "nsd-ref": "cirros_vdu_scaling_ns",
+ "name": "cirros_ns",
+ "resource-orchestrator": "osmopenmano",
+ "instantiate_params": {
+ "nsDescription": "default description",
+ "nsdId": "d7c8bd3c-eb39-4514-8847-19f01345524f",
+ "nsr_id": "87776f33-b67c-417a-8119-cb08e4098951",
+ "nsName": "cirros_ns",
+ "vimAccountId": "be48ae31-1d46-4892-a4b4-d69abd55714b"
+ },
+ "description": "default description",
+ "constituent-vnfr-ref": [
+ "0d9d06ad-3fc2-418c-9934-465e815fafe2",
+ "3336eb44-77df-4c4f-9881-d2828d259864"
+ ],
+ "admin-status": "ENABLED",
+ "detailed-status": "done",
+ "datacenter": "be48ae31-1d46-4892-a4b4-d69abd55714b",
+ "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"
+ }
+ }
+ }
+}
+
+vnfr_record_mocks = [
+ {
+ "_id": "0d9d06ad-3fc2-418c-9934-465e815fafe2",
+ "ip-address": "192.168.160.2",
+ "created-time": 1535392482.0044956,
+ "vim-account-id": "be48ae31-1d46-4892-a4b4-d69abd55714b",
+ "vdur": [
+ {
+ "interfaces": [
+ {
+ "mac-address": "fa:16:3e:71:fd:b8",
+ "name": "eth0",
+ "ip-address": "192.168.160.2"
+ }
+ ],
+ "status": "ACTIVE",
+ "vim-id": "63a65636-9fc8-4022-b070-980823e6266a",
+ "name": "cirros_ns-1-cirros_vnfd-VM-1",
+ "status-detailed": None,
+ "ip-address": "192.168.160.2",
+ "vdu-id-ref": "cirros_vnfd-VM"
+ }
+ ],
+ "id": "0d9d06ad-3fc2-418c-9934-465e815fafe2",
+ "vnfd-ref": "cirros_vdu_scaling_vnf",
+ "vnfd-id": "63f44c41-45ee-456b-b10d-5f08fb1796e0",
+ "_admin": {
+ "created": 1535392482.0067868,
+ "projects_read": [
+ "admin"
+ ],
+ "modified": 1535392482.0067868,
+ "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
+ }
+ ]
+ },
+ {
+ "_id": "3336eb44-77df-4c4f-9881-d2828d259864",
+ "ip-address": "192.168.160.10",
+ "created-time": 1535392482.0044956,
+ "vim-account-id": "be48ae31-1d46-4892-a4b4-d69abd55714b",
+ "vdur": [
+ {
+ "interfaces": [
+ {
+ "mac-address": "fa:16:3e:1e:76:e8",
+ "name": "eth0",
+ "ip-address": "192.168.160.10"
+ }
+ ],
+ "status": "ACTIVE",
+ "vim-id": "a154b8d3-2b10-421a-a51d-4b391d9bd366",
+ "name": "cirros_ns-2-cirros_vnfd-VM-1",
+ "status-detailed": None,
+ "ip-address": "192.168.160.10",
+ "vdu-id-ref": "cirros_vnfd-VM"
+ }
+ ],
+ "id": "3336eb44-77df-4c4f-9881-d2828d259864",
+ "vnfd-ref": "cirros_vdu_scaling_vnf",
+ "vnfd-id": "63f44c41-45ee-456b-b10d-5f08fb1796e0",
+ "_admin": {
+ "created": 1535392482.0076294,
+ "projects_read": [
+ "admin"
+ ],
+ "modified": 1535392482.0076294,
+ "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
+ }
+ ]}]
+
+nsd_record_mock = {'name': 'cirros_vdu_scaling_ns',
+ 'version': '1.0',
+ 'short-name': 'cirros_vdu_scaling_ns',
+ 'logo': 'osm.png',
+ 'id': 'cirros_vdu_scaling_ns',
+ 'description': 'Simple NS example with a cirros_vdu_scaling_vnf',
+ 'vendor': 'OSM',
+ 'vld': [
+ {'name': 'cirros_nsd_vld1',
+ 'type': 'ELAN',
+ 'id': 'cirros_nsd_vld1',
+ 'mgmt-network': 'true',
+ 'vnfd-connection-point-ref': [
+ {'vnfd-id-ref': 'cirros_vdu_scaling_vnf',
+ 'vnfd-connection-point-ref': 'eth0',
+ 'member-vnf-index-ref': 1},
+ {'vnfd-id-ref': 'cirros_vdu_scaling_vnf',
+ 'vnfd-connection-point-ref': 'eth0',
+ 'member-vnf-index-ref': 2}]}],
+ 'constituent-vnfd': [{'vnfd-id-ref': 'cirros_vdu_scaling_vnf',
+ 'member-vnf-index': '1'},
+ {'vnfd-id-ref': 'cirros_vdu_scaling_vnf',
+ 'member-vnf-index': '2'}]}
+
+vnfd_record_mock = {
+ "_id": "63f44c41-45ee-456b-b10d-5f08fb1796e0",
+ "name": "cirros_vdu_scaling_vnf",
+ "vendor": "OSM",
+ "vdu": [
+ {
+ "name": "cirros_vnfd-VM",
+ "monitoring-param": [
+ {
+ "id": "cirros_vnfd-VM_memory_util",
+ "nfvi-metric": "average_memory_utilization"
+ }
+ ],
+ "vm-flavor": {
+ "vcpu-count": 1,
+ "memory-mb": 256,
+ "storage-gb": 2
+ },
+ "description": "cirros_vnfd-VM",
+ "count": 1,
+ "id": "cirros_vnfd-VM",
+ "interface": [
+ {
+ "name": "eth0",
+ "external-connection-point-ref": "eth0",
+ "type": "EXTERNAL",
+ "virtual-interface": {
+ "bandwidth": "0",
+ "type": "VIRTIO",
+ "vpci": "0000:00:0a.0"
+ }
+ }
+ ],
+ "image": "cirros034"
+ }
+ ],
+ "monitoring-param": [
+ {
+ "id": "cirros_vnf_memory_util",
+ "name": "cirros_vnf_memory_util",
+ "aggregation-type": "AVERAGE",
+ "vdu-monitoring-param-ref": "cirros_vnfd-VM_memory_util",
+ "vdu-ref": "cirros_vnfd-VM"
+ }
+ ],
+ "description": "Simple VNF example with a cirros and a scaling group descriptor",
+ "id": "cirros_vdu_scaling_vnf",
+ "logo": "cirros-64.png",
+ "version": "1.0",
+ "connection-point": [
+ {
+ "name": "eth0",
+ "type": "VPORT"
+ }
+ ],
+ "mgmt-interface": {
+ "cp": "eth0"
+ },
+ "scaling-group-descriptor": [
+ {
+ "name": "scale_cirros_vnfd-VM",
+ "min-instance-count": 1,
+ "vdu": [
+ {
+ "count": 1,
+ "vdu-id-ref": "cirros_vnfd-VM"
+ }
+ ],
+ "max-instance-count": 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
+ }
+ ]
+ }
+ ]
+ }
+ ],
+ "short-name": "cirros_vdu_scaling_vnf",
+ "_admin": {
+ "created": 1535392242.6281035,
+ "modified": 1535392242.6281035,
+ "storage": {
+ "zipfile": "package.tar.gz",
+ "pkg-dir": "cirros_vnf",
+ "path": "/app/storage/",
+ "folder": "63f44c41-45ee-456b-b10d-5f08fb1796e0",
+ "fs": "local",
+ "descriptor": "cirros_vnf/cirros_vdu_scaling_vnfd.yaml"
+ },
+ "usageSate": "NOT_IN_USE",
+ "onboardingState": "ONBOARDED",
+ "userDefinedData": {
+
+ },
+ "projects_read": [
+ "admin"
+ ],
+ "operationalState": "ENABLED",
+ "projects_write": [
+ "admin"
+ ]
+ }
+}
+
+test_db = SqliteDatabase(':memory:')
+
+MODELS = [ScalingRecord, ScalingAlarm]
+
+
+class PolicyModuleAgentTest(unittest.TestCase):
+ def setUp(self):
+ super()
+ database.db = test_db
+ test_db.bind(MODELS)
+ test_db.connect()
+ test_db.drop_tables(MODELS)
+ test_db.create_tables(MODELS)
+
+ def tearDown(self):
+ super()
+
+ @patch.object(DbMongo, 'db_connect', Mock())
+ @patch.object(KafkaProducer, '__init__')
+ @patch.object(MonClient, 'create_alarm')
+ @patch.object(PolicyModuleAgent, '_get_vnfd')
+ @patch.object(PolicyModuleAgent, '_get_nsr')
+ @patch.object(PolicyModuleAgent, '_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]:
+ return vnfr_record_mocks[0]
+ if '2' in args[1]:
+ return vnfr_record_mocks[1]
+
+ def _test_configure_scaling_groups_create_alarm(*args, **kwargs):
+ return uuid.uuid4()
+
+ kafka_producer_init.return_value = None
+ get_vnfr.side_effect = _test_configure_scaling_groups_get_vnfr
+ get_nsr.return_value = nsr_record_mock
+ get_vnfd.return_value = vnfd_record_mock
+ create_alarm.side_effect = _test_configure_scaling_groups_create_alarm
+ agent = PolicyModuleAgent()
+ agent._configure_scaling_groups("test_nsr_id")
+ create_alarm.assert_any_call(metric_name='average_memory_utilization',
+ ns_id='test_nsr_id',
+ operation='GT',
+ statistic='AVERAGE',
+ threshold=80,
+ vdu_name='cirros_vnfd-VM',
+ vnf_member_index='1')
+ scaling_record = ScalingRecord.get()
+ self.assertEqual(scaling_record.name, 'scale_cirros_vnfd-VM')
+ self.assertEqual(scaling_record.nsr_id, 'test_nsr_id')
+ self.assertIsNotNone(scaling_record)
+
+
+if __name__ == '__main__':
+ unittest.main()
diff --git a/osm_policy_module/tests/unit/__init__.py b/osm_policy_module/tests/unit/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/osm_policy_module/tests/unit/__init__.py
diff --git a/osm_policy_module/tests/unit/test_policy_agent.py b/osm_policy_module/tests/unit/test_policy_agent.py
new file mode 100644
index 0000000..5e59901
--- /dev/null
+++ b/osm_policy_module/tests/unit/test_policy_agent.py
@@ -0,0 +1,35 @@
+# -*- coding: utf-8 -*-
+
+# Copyright 2018 Whitestack, LLC
+# *************************************************************
+
+# This file is part of OSM Monitoring module
+# All Rights Reserved to Whitestack, LLC
+
+# Licensed under the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License. You may obtain
+# a copy of the License at
+
+# http://www.apache.org/licenses/LICENSE-2.0
+
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+# License for the specific language governing permissions and limitations
+# under the License.
+
+# For those usages not covered by the Apache License, Version 2.0 please
+# contact: bdiaz@whitestack.com or glavado@whitestack.com
+##
+import unittest
+
+from osm_policy_module.core.agent import PolicyModuleAgent
+
+
+class PolicyAgentTest(unittest.TestCase):
+ def setUp(self):
+ self.agent = PolicyModuleAgent()
+
+
+if __name__ == '__main__':
+ unittest.main()