--- /dev/null
+"""Test the collection of charm metrics.
+ 1. Deploy a charm w/metrics to a unit
+ 2. Collect metrics or wait for collection to run
+ 3. Execute n2vc.GetMetrics()
+ 5. Destroy Juju unit
+"""
+import asyncio
+import functools
+import logging
+import sys
+import time
+import unittest
+from .. import utils
+
+NSD_YAML = """
+nsd:nsd-catalog:
+ nsd:
+ - id: singlecharmvdu-ns
+ name: singlecharmvdu-ns
+ short-name: singlecharmvdu-ns
+ description: NS with 1 VNFs singlecharmvdu-vnf connected by datanet and mgmtnet VLs
+ version: '1.0'
+ logo: osm.png
+ constituent-vnfd:
+ - vnfd-id-ref: singlecharmvdu-vnf
+ member-vnf-index: '1'
+ vld:
+ - id: mgmtnet
+ name: mgmtnet
+ short-name: mgmtnet
+ type: ELAN
+ mgmt-network: 'true'
+ vim-network-name: mgmt
+ vnfd-connection-point-ref:
+ - vnfd-id-ref: singlecharmvdu-vnf
+ member-vnf-index-ref: '1'
+ vnfd-connection-point-ref: vnf-mgmt
+ - vnfd-id-ref: singlecharmvdu-vnf
+ member-vnf-index-ref: '2'
+ vnfd-connection-point-ref: vnf-mgmt
+ - id: datanet
+ name: datanet
+ short-name: datanet
+ type: ELAN
+ vnfd-connection-point-ref:
+ - vnfd-id-ref: singlecharmvdu-vnf
+ member-vnf-index-ref: '1'
+ vnfd-connection-point-ref: vnf-data
+ - vnfd-id-ref: singlecharmvdu-vnf
+ member-vnf-index-ref: '2'
+ vnfd-connection-point-ref: vnf-data
+"""
+
+VNFD_YAML = """
+vnfd:vnfd-catalog:
+ vnfd:
+ - id: singlecharmvdu-vnf
+ name: singlecharmvdu-vnf
+ short-name: singlecharmvdu-vnf
+ version: '1.0'
+ description: A VNF consisting of 2 VDUs w/charms connected to an internal VL, and one VDU with cloud-init
+ logo: osm.png
+ connection-point:
+ - id: vnf-mgmt
+ name: vnf-mgmt
+ short-name: vnf-mgmt
+ type: VPORT
+ - id: vnf-data
+ name: vnf-data
+ short-name: vnf-data
+ type: VPORT
+ mgmt-interface:
+ cp: vnf-mgmt
+ internal-vld:
+ - id: internal
+ name: internal
+ short-name: internal
+ type: ELAN
+ internal-connection-point:
+ - id-ref: mgmtVM-internal
+ - id-ref: dataVM-internal
+ vdu:
+ - id: mgmtVM
+ name: mgmtVM
+ image: xenial
+ count: '1'
+ vm-flavor:
+ vcpu-count: '1'
+ memory-mb: '1024'
+ storage-gb: '10'
+ interface:
+ - name: mgmtVM-eth0
+ position: '1'
+ type: EXTERNAL
+ virtual-interface:
+ type: VIRTIO
+ external-connection-point-ref: vnf-mgmt
+ - name: mgmtVM-eth1
+ position: '2'
+ type: INTERNAL
+ virtual-interface:
+ type: VIRTIO
+ internal-connection-point-ref: mgmtVM-internal
+ internal-connection-point:
+ - id: mgmtVM-internal
+ name: mgmtVM-internal
+ short-name: mgmtVM-internal
+ type: VPORT
+ cloud-init-file: cloud-config.txt
+ vnf-configuration:
+ juju:
+ charm: metrics-ci
+ config-primitive:
+ - name: touch
+ parameter:
+ - name: filename
+ data-type: STRING
+ default-value: '/home/ubuntu/touched'
+"""
+
+
+class PythonTest(unittest.TestCase):
+ n2vc = None
+ charm = None
+
+ def setUp(self):
+ self.log = logging.getLogger()
+ self.log.level = logging.DEBUG
+
+ self.stream_handler = logging.StreamHandler(sys.stdout)
+ self.log.addHandler(self.stream_handler)
+
+ self.loop = asyncio.get_event_loop()
+
+ self.n2vc = utils.get_n2vc()
+
+ # Parse the descriptor
+ self.log.debug("Parsing the descriptor")
+ self.nsd = utils.get_descriptor(NSD_YAML)
+ self.vnfd = utils.get_descriptor(VNFD_YAML)
+
+
+ # Build the charm
+
+ vnf_config = self.vnfd.get("vnf-configuration")
+ if vnf_config:
+ juju = vnf_config['juju']
+ charm = juju['charm']
+
+ self.log.debug("Building charm {}".format(charm))
+ self.charm = utils.build_charm(charm)
+
+ def tearDown(self):
+ self.loop.run_until_complete(self.n2vc.logout())
+ self.log.removeHandler(self.stream_handler)
+
+ def n2vc_callback(self, model_name, application_name, workload_status,\
+ workload_message, task=None):
+ """We pass the vnfd when setting up the callback, so expect it to be
+ returned as a tuple."""
+ self.log.debug("status: {}; task: {}".format(workload_status, task))
+
+ # if workload_status in ["stop_test"]:
+ # # Stop the test
+ # self.log.debug("Stopping the test1")
+ # self.loop.call_soon_threadsafe(self.loop.stop)
+ # return
+
+ if workload_status:
+ if workload_status in ["active"] and not task:
+ # Force a run of the metric collector, so we don't have
+ # to wait for it's normal 5 minute interval run.
+ # NOTE: this shouldn't be done outside of CI
+ utils.collect_metrics(application_name)
+
+ # get the current metrics
+ task = asyncio.ensure_future(
+ self.n2vc.GetMetrics(
+ model_name,
+ application_name,
+ )
+ )
+ task.add_done_callback(
+ functools.partial(
+ self.n2vc_callback,
+ model_name,
+ application_name,
+ "collect_metrics",
+ task,
+ )
+ )
+
+ elif workload_status in ["collect_metrics"]:
+
+ if task:
+ # Check if task returned metrics
+ results = task.result()
+
+ foo = utils.parse_metrics(application_name, results)
+ if 'load' in foo:
+ self.log.debug("Removing charm")
+ task = asyncio.ensure_future(
+ self.n2vc.RemoveCharms(model_name, application_name, self.n2vc_callback)
+ )
+ task.add_done_callback(
+ functools.partial(
+ self.n2vc_callback,
+ model_name,
+ application_name,
+ "stop_test",
+ task,
+ )
+ )
+ return
+
+ # No metrics are available yet, so try again in a minute.
+ self.log.debug("Sleeping for 60 seconds")
+ time.sleep(60)
+ task = asyncio.ensure_future(
+ self.n2vc.GetMetrics(
+ model_name,
+ application_name,
+ )
+ )
+ task.add_done_callback(
+ functools.partial(
+ self.n2vc_callback,
+ model_name,
+ application_name,
+ "collect_metrics",
+ task,
+ )
+ )
+ elif workload_status in ["stop_test"]:
+ # Stop the test
+ self.log.debug("Stopping the test2")
+ self.loop.call_soon_threadsafe(self.loop.stop)
+
+ def test_deploy_application(self):
+ """Deploy proxy charm to a unit."""
+ if self.nsd and self.vnfd:
+ params = {}
+ vnf_index = 0
+
+ def deploy():
+ """An inner function to do the deployment of a charm from
+ either a vdu or vnf.
+ """
+ charm_dir = "{}/builds/{}".format(utils.get_charm_path(), charm)
+
+ # Setting this to an IP that will fail the initial config.
+ # This will be detected in the callback, which will execute
+ # the "config" primitive with the right IP address.
+ # mgmtaddr = self.container.state().network['eth0']['addresses']
+ # params['rw_mgmt_ip'] = mgmtaddr[0]['address']
+
+ # Legacy method is to set the ssh-private-key config
+ # with open(utils.get_juju_private_key(), "r") as f:
+ # pkey = f.readline()
+ # params['ssh-private-key'] = pkey
+
+ ns_name = "default"
+
+ vnf_name = self.n2vc.FormatApplicationName(
+ ns_name,
+ self.vnfd['name'],
+ str(vnf_index),
+ )
+
+ self.loop.run_until_complete(
+ self.n2vc.DeployCharms(
+ ns_name,
+ vnf_name,
+ self.vnfd,
+ charm_dir,
+ params,
+ {},
+ self.n2vc_callback
+ )
+ )
+
+ # Check if the VDUs in this VNF have a charm
+ # for vdu in vnfd['vdu']:
+ # vdu_config = vdu.get('vdu-configuration')
+ # if vdu_config:
+ # juju = vdu_config['juju']
+ # self.assertIsNotNone(juju)
+ #
+ # charm = juju['charm']
+ # self.assertIsNotNone(charm)
+ #
+ # params['initial-config-primitive'] = vdu_config['initial-config-primitive']
+ #
+ # deploy()
+ # vnf_index += 1
+ #
+ # # Check if this VNF has a charm
+ vnf_config = self.vnfd.get("vnf-configuration")
+ if vnf_config:
+ juju = vnf_config['juju']
+ self.assertIsNotNone(juju)
+
+ charm = juju['charm']
+ self.assertIsNotNone(charm)
+
+ if 'initial-config-primitive' in vnf_config:
+ params['initial-config-primitive'] = vnf_config['initial-config-primitive']
+
+ deploy()
+ vnf_index += 1
+
+ self.loop.run_forever()
+ # while self.loop.is_running():
+ # # await asyncio.sleep(1)
+ # time.sleep(1)