genericparser/jobs/tests/tests.py

   1 # Copyright 2017 ZTE Corporation.
   2 #
   3 # Licensed under the Apache License, Version 2.0 (the "License");
   4 # you may not use this file except in compliance with the License.
   5 # You may obtain a copy of the License at
   6 #
   7 #         http://www.apache.org/licenses/LICENSE-2.0
   8 #
   9 # Unless required by applicable law or agreed to in writing, software
  10 # distributed under the License is distributed on an "AS IS" BASIS,
  11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12 # See the License for the specific language governing permissions and
  13 # limitations under the License.
  14 from django.test import TestCase, Client
  15 from rest_framework import status
  16
  17 from genericparser.pub.database.models import JobModel, JobStatusModel
  18
  19
  20 class JobsViewTest(TestCase):
  21     def setUp(self):
  22         self.job_id = 'test_job_id'
  23         self.client = Client()
  24
  25     def tearDown(self):
  26         JobModel.objects.all().delete()
  27
  28     def test_job_normal(self):
  29         JobModel(jobid=self.job_id, jobtype='VNF', jobaction='INST', resid='1').save()
  30         JobStatusModel(indexid=1, jobid=self.job_id, status='inst', errcode='0', progress=20, descp='inst').save()
  31         response = self.client.get("/api/genericparser/v1/jobs/%s" % self.job_id)
  32         self.failUnlessEqual(status.HTTP_200_OK, response.status_code)
  33
  34     def test_job_when_jobid_not_exist(self):
  35         job_id = 'test_new_job_id'
  36         JobModel(jobid=self.job_id, jobtype='VNF', jobaction='INST', resid='1').save()
  37         JobStatusModel(indexid=1, jobid=self.job_id, status='inst', progress=20, descp='inst').save()
  38         response = self.client.get("/api/genericparser/v1/jobs/%s" % job_id)
  39         self.assertIn('jobId', response.data)
  40         self.assertNotIn('responseDescriptor', response.data)