Merge "Add vfc-nslcm log integration config"
[vfc/nfvo/lcm.git] / lcm / pub / tests / test_scaleaspect.py
index 53d3a7d..38d4ad2 100644 (file)
@@ -1,10 +1,24 @@
+# Copyright 2016-2018 ZTE Corporation.
+#
+# 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.
+
 from django.test import TestCase
 from lcm.pub.utils.scaleaspect import get_json_data
 from lcm.pub.utils.scaleaspect import get_nsdId
 from lcm.pub.utils.scaleaspect import get_scale_vnf_data_from_json
 from lcm.pub.utils.scaleaspect import get_scale_vnf_data_info_list
 from lcm.pub.utils.scaleaspect import set_scacle_vnf_instance_id
-from lcm.pub.utils.scaleaspect import get_and_check_params
+from lcm.pub.utils.scaleaspect import check_and_set_params
 from lcm.pub.utils.scaleaspect import set_scaleVnfData_type
 from lcm.pub.database.models import NfInstModel
 from lcm.pub.database.models import NSInstModel
@@ -39,11 +53,27 @@ class TestScaleAspect(TestCase):
         self.ns_scale_aspect = "TIC_EDGE_IMS"
         self.ns_scale_steps = "1"
         self.ns_scale_direction = "SCALE_IN"
-        self.scaleNsData = {
-            "aspectId": self.ns_scale_aspect,
-            "numberOfSteps": self.ns_scale_steps,
-            "scalingDirection": self.ns_scale_direction
-        }
+        self.scaleNsData = [{
+            "scaleNsByStepsData": [
+                {
+                    "aspectId": self.ns_scale_aspect,
+                    "numberOfSteps": self.ns_scale_steps,
+                    "scalingDirection": self.ns_scale_direction
+                }
+            ]
+        }]
+
+        self.ns_scale_aspect2 = "TIC_EDGE_HW"
+        self.ns_scale_steps2 = "4"
+        self.scaleNsData2 = [{
+            "scaleNsByStepsData": [
+                {
+                    "aspectId": self.ns_scale_aspect2,
+                    "numberOfSteps": self.ns_scale_steps2,
+                    "scalingDirection": self.ns_scale_direction
+                }
+            ]
+        }]
 
     def init_scaling_map_json(self):
         curdir_path = os.path.dirname(
@@ -151,12 +181,63 @@ class TestScaleAspect(TestCase):
                        '"description": "PGW VNFD description",'
                        '"isShared":true,"vnfExtendType":"driver"}}')
 
+    def add_new_vnf_instance(self):
+        # Create a third vnf instance
+        nf_inst_id = "241"
+        package_id = "nf_hw_cscf"
+        nf_uuid = "ab34-3g5j-de13-aa85-ij93"
+
+        NfInstModel.objects.create(
+            nfinstid=nf_inst_id,
+            nf_name=self.nf_name,
+            vnf_id=self.vnf_id,
+            vnfm_inst_id=self.vnfm_inst_id,
+            ns_inst_id=self.ns_inst_id,
+            max_cpu='14',
+            max_ram='12296',
+            max_hd='101',
+            max_shd="20",
+            max_net=10,
+            status='active',
+            mnfinstid=nf_uuid,
+            package_id=package_id,
+            vnfd_model='{"metadata": {"vnfdId": "1","vnfdName": "PGW001",'
+                       '"vnfProvider": "zte","vnfdVersion": "V00001","vnfVersion": "V5.10.20",'
+                       '"productType": "CN","vnfType": "PGW",'
+                       '"description": "PGW VNFD description",'
+                       '"isShared":true,"vnfExtendType":"driver"}}')
+
+        # Create a third vnf instance
+        nf_inst_id = "242"
+        package_id = "nf_hw_hss"
+        nf_uuid = "ab34-3g5j-de13-aa85-id93"
+
+        NfInstModel.objects.create(
+            nfinstid=nf_inst_id,
+            nf_name=self.nf_name,
+            vnf_id=self.vnf_id,
+            vnfm_inst_id=self.vnfm_inst_id,
+            ns_inst_id=self.ns_inst_id,
+            max_cpu='14',
+            max_ram='12296',
+            max_hd='101',
+            max_shd="20",
+            max_net=10,
+            status='active',
+            mnfinstid=nf_uuid,
+            package_id=package_id,
+            vnfd_model='{"metadata": {"vnfdId": "1","vnfdName": "PGW001",'
+                       '"vnfProvider": "zte","vnfdVersion": "V00001","vnfVersion": "V5.10.20",'
+                       '"productType": "CN","vnfType": "PGW",'
+                       '"description": "PGW VNFD description",'
+                       '"isShared":true,"vnfExtendType":"driver"}}')
+
     def tearDown(self):
         NSInstModel().clean()
         NfInstModel().clean()
 
     def test_get_and_check_params(self):
-        aspect, numberOfSteps, scale_type = get_and_check_params(
+        aspect, numberOfSteps, scale_type = check_and_set_params(
             self.scaleNsData, "1")
         self.assertEqual(aspect, self.ns_scale_aspect)
         self.assertEqual(numberOfSteps, self.ns_scale_steps)
@@ -167,6 +248,14 @@ class TestScaleAspect(TestCase):
             self.scaling_map_json, "23", "TIC_EDGE_IMS", "1")
         self.assertIsNotNone(vnf_data_package)
         self.assertEqual(2, vnf_data_package.__len__())
+        self.assertIsNotNone(vnf_data_package)
+        self.assertEqual(2, vnf_data_package.__len__())
+        self.assertEqual("nf_zte_cscf", vnf_data_package[0]["vnfd_id"])
+        self.assertEqual("1", vnf_data_package[0]["numberOfSteps"])
+        self.assertEqual("gsu", vnf_data_package[0]["vnf_scaleAspectId"])
+        self.assertEqual("nf_zte_hss", vnf_data_package[1]["vnfd_id"])
+        self.assertEqual("3", vnf_data_package[1]["numberOfSteps"])
+        self.assertEqual("gpu", vnf_data_package[1]["vnf_scaleAspectId"])
 
     def test_get_scale_vnf_data_from_json_2(self):
         vnf_data_package = get_scale_vnf_data_from_json(
@@ -250,16 +339,13 @@ class TestScaleAspect(TestCase):
         vnf_scale_list = set_scacle_vnf_instance_id(self.vnf_scale_info_list)
         result = set_scaleVnfData_type(vnf_scale_list, self.ns_scale_direction)
         self.assertEqual(2, result.__len__())
-
         self.assertNotIn("scaleByStepData", result)
-
         self.assertEqual(
             self.ns_scale_direction,
             result[0]["scaleByStepData"]["type"])
         self.assertEqual("mpu", result[0]["scaleByStepData"]["aspectId"])
         self.assertNotIn("vnf_scaleAspectId", result[0]["scaleByStepData"])
         self.assertEqual("1", result[0]["scaleByStepData"]["numberOfSteps"])
-
         self.assertEqual(
             self.ns_scale_direction,
             result[1]["scaleByStepData"]["type"])
@@ -279,3 +365,34 @@ class TestScaleAspect(TestCase):
         scale_vnf_data = get_scale_vnf_data_info_list(self.scaleNsData, "1")
         self.assertIsNotNone(scale_vnf_data)
         self.assertEqual(2, scale_vnf_data.__len__())
+
+    @mock.patch.object(catalog, 'get_scalingmap_json_package')
+    def test_get_scale_vnf_data_info_list_2(
+            self, mock_get_scalingmap_json_package):
+        mock_get_scalingmap_json_package.return_value = self.scaling_map_json
+
+        scale_vnf_data = None
+        is_exception_caught = False
+        try:
+            scale_vnf_data = get_scale_vnf_data_info_list(
+                self.scaleNsData2, "1")
+        except Exception:
+            is_exception_caught = True
+        self.assertTrue(is_exception_caught)
+        self.assertIsNone(scale_vnf_data)
+
+    @mock.patch.object(catalog, 'get_scalingmap_json_package')
+    def test_get_scale_vnf_data_info_list_3(
+            self, mock_get_scalingmap_json_package):
+        mock_get_scalingmap_json_package.return_value = self.scaling_map_json
+        self.add_new_vnf_instance()
+
+        scale_vnf_data = None
+        is_exception_caught = False
+        try:
+            scale_vnf_data = get_scale_vnf_data_info_list(
+                self.scaleNsData2, "1")
+        except Exception:
+            is_exception_caught = True
+        self.assertFalse(is_exception_caught)
+        self.assertEqual(2, scale_vnf_data.__len__())