ADD UT for ns_sfcs Issue-ID: VFC-1429 Signed-off-by: zhuerlei <zhu.erlei@zte.com.cn>

[vfc/nfvo/lcm.git] / lcm / ns_sfcs / tests / test_sfc_instance.py

diff --git a/lcm/ns_sfcs/tests/test_sfc_instance.py b/lcm/ns_sfcs/tests/test_sfc_instance.py
index f2e7d39..f4cf768 100644 (file)
--- a/lcm/ns_sfcs/tests/test_sfc_instance.py
+++ b/lcm/ns_sfcs/tests/test_sfc_instance.py
@@ -11,37 +11,57 @@
 # 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.
-# import json
-# from rest_framework import status
-# from test_data import nsd_model
+import json
+import copy
+from rest_framework import status
 from django.test import Client
 from django.test import TestCase
 from lcm.pub.database.models import FPInstModel, VNFFGInstModel
+from lcm.ns_sfcs.tests.test_data import nsd_model
 
 
-class TestSfc(TestCase):
+class TestSfcInstance(TestCase):
     def setUp(self):
         self.client = Client()
+        self.data = {
+            "nsInstanceId": "ns_inst_1",
+            "context": json.dumps(nsd_model),
+            "fpindex": "fpd_1",
+            "sdnControllerId": "sdnControllerId_1"
+        }
         VNFFGInstModel.objects.all().delete()
         FPInstModel.objects.all().delete()
         VNFFGInstModel(vnffgdid="vnffg_id1", vnffginstid="vnffg_inst_1", nsinstid="ns_inst_1", endpointnumber=2,
-                       vllist="vlinst1", cplist="cp1", vnflist="vnf1,vnf2").save()
+                       vllist="vlinst1", cplist="cp1", vnflist="vnf1,vnf2", fplist="fp1").save()
 
     def tearDown(self):
         VNFFGInstModel.objects.all().delete()
         FPInstModel.objects.all().delete()
 
-    # def test_sfc_instance_success(self):
-    #    data = {
-    #        "nsInstanceId": "ns_inst_1",
-    #        "context": json.dumps(nsd_model),
-    #        "fpindex": "fpd_1",
-    #        "sdnControllerId": "sdnControllerId_1"
-    #    }
-    #    resp = self.client.post("/api/nslcm/v1/ns/sfc_instance", data, format='json')
-
-    #    self.assertEqual(resp.status_code, status.HTTP_200_OK, resp.data)
-    #    vnffg = VNFFGInstModel.objects.get(vnffginstid="vnffg_inst_1")
-    #    ret = FPInstModel.objects.get(fpinstid=resp.data["fpinstid"])
-    #    self.assertEqual(vnffg.fplist, resp.data["fpinstid"])
-    #    self.assertIsNotNone(ret)
+    def test_sfc_instance_success(self):
+        resp = self.client.post("/api/nslcm/v1/ns/sfc_instance", self.data, format="json")
+
+        self.assertEqual(resp.status_code, status.HTTP_200_OK)
+        vnffg = VNFFGInstModel.objects.get(vnffginstid="vnffg_inst_1")
+        self.assertEqual(vnffg.fplist, "fp1," + resp.data["fpinstid"])
+        ret = FPInstModel.objects.get(fpinstid=resp.data["fpinstid"])
+        self.assertIsNotNone(ret)
+
+    def test_sfc_instance_when_request_data_is_not_valid(self):
+        data = {}
+        resp = self.client.post("/api/nslcm/v1/ns/sfc_instance", data, format="json")
+        self.assertEqual(resp.status_code, status.HTTP_500_INTERNAL_SERVER_ERROR)
+
+    def test_sfc_instance_when_error_request_data(self):
+        data = copy.deepcopy(self.data)
+        data.pop("fpindex")
+        resp = self.client.post("/api/nslcm/v1/ns/sfc_instance", data, format="json")
+
+        self.assertEqual(resp.status_code, status.HTTP_500_INTERNAL_SERVER_ERROR)
+
+    def test_sfc_instance_when_no_fp_model(self):
+        data = copy.deepcopy(self.data)
+        data["context"] = json.dumps({"fps": []})
+        resp = self.client.post("/api/nslcm/v1/ns/sfc_instance", data, format="json")
+
+        self.assertEqual(resp.status_code, status.HTTP_500_INTERNAL_SERVER_ERROR)