systest/testcases/vnfs/test_vnfs.py

   1 # Copyright 2017 Sandvine
   2 #
   3 # All Rights Reserved.
   4 #
   5 #    Licensed under the Apache License, Version 2.0 (the "License"); you may
   6 #    not use this file except in compliance with the License. You may obtain
   7 #    a copy of the License at
   8 #
   9 #         http://www.apache.org/licenses/LICENSE-2.0
  10 #
  11 #    Unless required by applicable law or agreed to in writing, software
  12 #    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
  13 #    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
  14 #    License for the specific language governing permissions and limitations
  15 #    under the License.
  16
  17 import pytest
  18 import pprint
  19 import time
  20 from osmclient.common import utils
  21 import pprint
  22
  23 class TestClass(object):
  24
  25     @pytest.fixture(scope='function')
  26     def cleanup_test_vnf(self,osm,request):
  27         vnfd_file_list = osm.vnfd_descriptors_list
  28         nsd_file_list = osm.nsd_descriptors_list
  29
  30         # cleanup all ns/packages that might have been left around
  31         def teardown():
  32
  33             # first delete all nsd's
  34             for file in nsd_file_list:
  35                 try:
  36                     desc = osm.get_api().package.get_key_val_from_pkg(file)
  37                     ns_name=osm.ns_name_prefix+desc['name']
  38                     osm.get_api().ns.delete(ns_name)
  39                 except:
  40                     pass
  41
  42             # delete all nsd packages
  43             for file in nsd_file_list:
  44                 try:
  45                     desc = osm.get_api().package.get_key_val_from_pkg(file)
  46                     osm.get_api().nsd.delete(desc['name'])
  47                 except:
  48                     pass
  49
  50             # delete all vnfd packages
  51             for file in vnfd_file_list:
  52                 try:
  53                     desc = osm.get_api().package.get_key_val_from_pkg(file)
  54                     osm.get_api().vnfd.delete(desc['name'])
  55                 except:
  56                     pass
  57
  58         request.addfinalizer(teardown)
  59
  60     def vnf_upload_packages(self, osm, vnfd_file_list, nsd_file_list ):
  61         vnfd_descriptors=[]
  62         for file in vnfd_file_list:
  63             assert not osm.get_api().package.upload(file)
  64             assert not osm.get_api().package.wait_for_upload(file)
  65             desc = osm.get_api().package.get_key_val_from_pkg(file)
  66             assert desc
  67             vnfd_descriptors.append(desc)
  68
  69         nsd_descriptors=[]
  70         for file in nsd_file_list:
  71             assert not osm.get_api().package.upload(file)
  72             assert not osm.get_api().package.wait_for_upload(file)
  73             desc = osm.get_api().package.get_key_val_from_pkg(file)
  74             assert desc
  75             nsd_descriptors.append(desc)
  76         # TODO/HACK: need to figure out why this is necessary.
  77         # vnfd/nsd is there (seen on ping_pong), but the ns fails that nsd is not there,
  78         # another way to check if the nsd is really ready via API?
  79         time.sleep(5)
  80
  81     def vnf_test(self,osm, openstack, vim, vmware, vnfd_file_list, nsd_file_list, ns_scale=False):
  82         for file in nsd_file_list:
  83             nsd_desc = osm.get_api().package.get_key_val_from_pkg(file)
  84
  85             ns_name=osm.ns_name_prefix+nsd_desc['name']
  86
  87             assert osm.get_api().ns.create(nsd_desc['name'],ns_name,vim.vim_name)
  88
  89             if not utils.wait_for_value(lambda: osm.get_api().ns.get_field(ns_name,'operational-status'),result='init', wait_time=10):
  90                 nsr=osm.get_api().ns.get(ns_name)
  91                 pprint.pprint(nsr)
  92                 assert Fail, "operational-status != init"
  93
  94             # make sure ns is running
  95             if not utils.wait_for_value(lambda: osm.get_api().ns.get_field(ns_name,'operational-status'),result='running',wait_time=30):
  96                 nsr=osm.get_api().ns.get(ns_name)
  97                 pprint.pprint(nsr)
  98                 assert Fail, "operational-status != running"
  99
 100             if ns_scale:
 101                 # for each descriptor, scale it
 102                 for scale in nsd_desc['scaling-group-descriptor']:
 103                     # scale it.
 104                     assert not osm.get_api().ns.scale(ns_name, scale['name'], 1)
 105
 106                     # ensure ns is scaling-out
 107                     assert utils.wait_for_value(lambda: osm.get_api().ns.get_field(ns_name,'operational-status'),result='scaling-out',wait_time=120)
 108
 109                     # wait for ns to be in running-state
 110                     assert utils.wait_for_value(lambda: osm.get_api().ns.get_field(ns_name,'operational-status'),result='running',wait_time=300)
 111
 112             time.sleep(10)
 113
 114             assert not osm.get_api().ns.delete(ns_name)
 115
 116             assert not osm.get_api().nsd.delete(nsd_desc['name'])
 117
 118         for file in vnfd_file_list:
 119             vnfd_desc = osm.get_api().package.get_key_val_from_pkg(file)
 120             assert not osm.get_api().vnfd.delete(vnfd_desc['name'])
 121
 122     @pytest.mark.openstack
 123     @pytest.mark.vnf
 124     @pytest.mark.vmware
 125     def test_vnf(self,osm, vim, openstack, vmware, cleanup_test_vnf):
 126         vnfd_file_list = osm.vnfd_descriptors_list
 127         nsd_file_list = osm.nsd_descriptors_list
 128
 129         self.vnf_upload_packages(osm, vnfd_file_list, nsd_file_list )
 130         self.vnf_test(osm,openstack, vim, vmware, vnfd_file_list, nsd_file_list)
 131
 132     @pytest.mark.openstack
 133     @pytest.mark.ns_scale
 134     @pytest.mark.vmware
 135     def test_scale_vnf(self,osm, vim, openstack, vmware, cleanup_test_vnf):
 136         vnfd_file_list = osm.vnfd_descriptors_list
 137         nsd_file_list = osm.nsd_descriptors_list
 138
 139         self.vnf_upload_packages(osm, vnfd_file_list, nsd_file_list )
 140         self.vnf_test(osm,openstack, vim, vmware, vnfd_file_list, nsd_file_list, ns_scale=True)