from lcm.ns.biz.ns_terminate import TerminateNsService
from lcm.pub.database.models import NfInstModel, NSInstModel
from lcm.pub.utils import restcall
-from lcm.pub.utils.jobutil import JOB_MODEL_STATUS
-from lcm.pub.utils.jobutil import JobUtil, JOB_TYPE
+from lcm.pub.enum import JOB_MODEL_STATUS, JOB_TYPE
+from lcm.pub.utils.jobutil import JobUtil
class TestTerminateNsViews(TestCase):
"terminationType": "forceful",
"gracefulTerminationTimeout": "600"}
response = self.client.post("/api/nslcm/v1/ns/%s/terminate" % self.ns_inst_id, data=req_data)
- self.failUnlessEqual(status.HTTP_202_ACCEPTED, response.status_code, response.data)
+ self.assertEqual(status.HTTP_202_ACCEPTED, response.status_code, response.data)
response = self.client.delete("/api/nslcm/v1/ns/%s" % self.ns_inst_id)
- self.failUnlessEqual(status.HTTP_204_NO_CONTENT, response.status_code)
+ self.assertEqual(status.HTTP_204_NO_CONTENT, response.status_code)
@mock.patch.object(restcall, 'call_req')
def test_terminate_vnf(self, mock_call_req):
return mock_vals[args[4]]
mock_call_req.side_effect = side_effect
-
- TerminateNsService(self.nf_inst_id, "forceful", "600", job_id).run()
+ req_data = {
+ "terminationType": "FORCEFUL",
+ "gracefulTerminationTimeout": "600"}
+ TerminateNsService(self.nf_inst_id, job_id, req_data).run()
nsinst = NSInstModel.objects.get(id=self.ns_inst_id)
if nsinst:
self.assertTrue(1, 0)
"terminationType": "forceful",
"gracefulTerminationTimeout": "600"}
response = self.client.post("/api/nslcm/v1/ns/%s/terminate" % ns_inst_id, data=req_data)
- self.failUnlessEqual(status.HTTP_202_ACCEPTED, response.status_code, response.data)
+ self.assertEqual(status.HTTP_202_ACCEPTED, response.status_code, response.data)
self.assertRaises(NSInstModel.DoesNotExist, NSInstModel.objects.get, id=ns_inst_id)