add ut for vnf terminate

Change-Id: I5d37e4b56e77a5f080ff17653ddb9b78f2b228ad
Issue-ID: VFC-1489
Signed-off-by: yangyan <yangyanyj@chinamobile.com>

diff --git a/lcm/ns_vnfs/tests/vnf_tests.py b/lcm/ns_vnfs/tests/vnf_tests.py
index 73e9e27..6b25e44 100644 (file)
--- a/lcm/ns_vnfs/tests/vnf_tests.py
+++ b/lcm/ns_vnfs/tests/vnf_tests.py
@@ -14,9 +14,12 @@
 
 import unittest
 import json
+
 import mock
 from rest_framework.test import APIClient
 from rest_framework import status
+
+from lcm.ns_vnfs.biz.terminate_nfs import TerminateVnfs
 from lcm.ns_vnfs.tests.const import GRANT_DATA, VNF_LCM_OP_OCC_NOTIFICATION_DATA, \
     VNF_IDENTIFIER_CREATION_NOTIFICATION_DATA, VNF_IDENTIFIER_DELETION_NOTIFICATION_DATA
 from lcm.pub.database.models import NfInstModel
@@ -198,3 +201,13 @@ class VnfGrantViewTest(unittest.TestCase):
                                     data=VNF_IDENTIFIER_DELETION_NOTIFICATION_DATA,
                                     format='json')
         self.assertEqual(status.HTTP_204_NO_CONTENT, response.status_code)
+
+    @mock.patch.object(TerminateVnfs, 'run')
+    def test_vnf_terminate(self, mock_run):
+        vnf_instance_id = '628fd152-0089-4c20-b549-f35cb2fd4933'
+        data = {
+            'terminationType': 'FORCEFUL',
+            'gracefulTerminationTimeout': 600
+        }
+        response = self.client.post("/api/nslcm/v1/ns/terminatevnf/%s" % vnf_instance_id, data=data)
+        self.assertEqual(status.HTTP_202_ACCEPTED, response.status_code, response.data)