vfclcm upgrade from python2 to python3

[vfc/gvnfm/vnflcm.git] / lcm / lcm / nf / tests / test_change_flavour.py

diff --git a/lcm/lcm/nf/tests/test_change_flavour.py b/lcm/lcm/nf/tests/test_change_flavour.py
index b4330a3..442fd8d 100644 (file)
--- a/lcm/lcm/nf/tests/test_change_flavour.py
+++ b/lcm/lcm/nf/tests/test_change_flavour.py
@@ -15,18 +15,57 @@ from django.test import TestCase
 from rest_framework import status
 from rest_framework.test import APIClient
 
+from lcm.pub.database.models import NfInstModel
+
 
 class TestFlavour(TestCase):
     def setUp(self):
         self.client = APIClient()
+        NfInstModel(nfinstid='12345',
+                    nf_name='VNF1',
+                    nf_desc="VNF DESC",
+                    vnfdid="1",
+                    netype="XGW",
+                    vendor="ZTE",
+                    vnfSoftwareVersion="V1",
+                    version="V1",
+                    package_id="2",
+                    status='NOT_INSTANTIATED').save()
+        NfInstModel(nfinstid='345',
+                    nf_name='VNF1',
+                    nf_desc="VNF DESC",
+                    vnfdid="1",
+                    netype="XGW",
+                    vendor="ZTE",
+                    vnfSoftwareVersion="V1",
+                    version="V1",
+                    package_id="2",
+                    status='INSTANTIATED').save()
+        self.req_data = {
+            "newFlavourId": "myFlavour_new"
+        }
 
     def tearDown(self):
-        pass
+        NfInstModel.objects.filter(nfinstid='12345').delete()
+        NfInstModel.objects.filter(nfinstid='345').delete()
 
     def test_change_flavour_not_found(self):
-        req_data = {}
         url = "/api/vnflcm/v1/vnf_instances/12/change_flavour"
         response = self.client.post(url,
-                                    data=req_data,
+                                    data=self.req_data,
+                                    format='json')
+        self.assertEqual(status.HTTP_404_NOT_FOUND, response.status_code)
+
+    def test_change_flavour_conflict(self):
+        url = "/api/vnflcm/v1/vnf_instances/12345/change_flavour"
+        response = self.client.post(url,
+                                    data=self.req_data,
+                                    format='json')
+        self.assertEqual(status.HTTP_409_CONFLICT, response.status_code)
+
+    def test_change_flavour_badreq(self):
+        url = "/api/vnflcm/v1/vnf_instances/345/change_flavour"
+        response = self.client.post(url,
+                                    data={},
                                     format='json')
-        self.failUnlessEqual(status.HTTP_404_NOT_FOUND, response.status_code)
+        self.assertEqual(status.HTTP_400_BAD_REQUEST, response.status_code)