catalog/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 from catalog.pub.utils.jobutil import JobUtil
  17
  18 from catalog.pub.database.models import JobModel, JobStatusModel
  19
  20
  21 class JobsViewTest(TestCase):
  22     def setUp(self):
  23         self.job_id = 'test_job_id'
  24         self.client = Client()
  25
  26     def tearDown(self):
  27         JobModel.objects.all().delete()
  28
  29     def test_job_normal(self):
  30         JobModel(jobid=self.job_id, jobtype='VNF', jobaction='INST', resid='1').save()
  31         JobStatusModel(indexid=1, jobid=self.job_id, status='inst', errcode='0', progress=20, descp='inst').save()
  32         response = self.client.get("/api/catalog/v1/jobs/%s" % self.job_id)
  33         self.assertEqual(status.HTTP_200_OK, response.status_code)
  34
  35     def test_job_when_jobid_not_exist(self):
  36         job_id = 'test_new_job_id'
  37         JobModel(jobid=self.job_id, jobtype='VNF', jobaction='INST', resid='1').save()
  38         JobStatusModel(indexid=1, jobid=self.job_id, status='inst', progress=20, descp='inst').save()
  39         response = self.client.get("/api/catalog/v1/jobs/%s" % job_id)
  40         self.assertIn('jobId', response.data)
  41         self.assertNotIn('responseDescriptor', response.data)
  42
  43     def test_job_normal_multijobstatus(self):
  44         JobUtil.create_job(
  45             inst_type='test_new_job_id',
  46             jobaction='test_jobaction',
  47             inst_id="test_jobinstid",
  48             job_id=self.job_id)
  49         response = self.client.post("/api/catalog/v1/jobs/%s" % self.job_id, {"progress": "10", "desc": "10%", "errcode": "true"},
  50                                     format='json')
  51         print(response)
  52         self.assertEqual(status.HTTP_202_ACCEPTED, response.status_code)
  53         response = self.client.post("/api/catalog/v1/jobs/%s" % self.job_id, {"progress": "50", "desc": "50%", "errcode": "true"},
  54                                     format='json')
  55         self.assertEqual(status.HTTP_202_ACCEPTED, response.status_code)
  56         response = self.client.post("/api/catalog/v1/jobs/%s" % self.job_id, {"progress": "100", "desc": "100%", "errcode": "true"},
  57                                     format='json')
  58         self.assertEqual(status.HTTP_202_ACCEPTED, response.status_code)
  59         response = self.client.get("/api/catalog/v1/jobs/%s" % self.job_id)
  60         print(response)
  61         self.assertEqual(status.HTTP_200_OK, response.status_code)