lcm/lcm/nf/tests/test_scale_vnf.py

   1 # Copyright (C) 2019 ZTE. All Rights Reserved.
   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
  15 from rest_framework import status
  16 from rest_framework.test import APIClient
  17
  18 from lcm.pub.database.models import NfInstModel
  19
  20
  21 class TestNfScale(TestCase):
  22     def setUp(self):
  23         self.client = APIClient()
  24         NfInstModel(nfinstid='12345',
  25                     nf_name='VNF1',
  26                     nf_desc="VNF DESC",
  27                     vnfdid="1",
  28                     netype="XGW",
  29                     vendor="ZTE",
  30                     vnfSoftwareVersion="V1",
  31                     version="V1",
  32                     package_id="2",
  33                     status='NOT_INSTANTIATED').save()
  34         self.req_data = {
  35             "type": "SCALE_IN",
  36             "aspectId": "sunshine_aspect"
  37         }
  38
  39     def tearDown(self):
  40         NfInstModel.objects.filter(nfinstid='12345').delete()
  41
  42     def test_scale_vnf_not_found(self):
  43         url = "/api/vnflcm/v1/vnf_instances/12/scale"
  44         response = self.client.post(url,
  45                                     data=self.req_data,
  46                                     format='json')
  47         self.failUnlessEqual(status.HTTP_404_NOT_FOUND, response.status_code)
  48
  49     def test_scale_vnf_conflict(self):
  50         url = "/api/vnflcm/v1/vnf_instances/12345/scale"
  51         response = self.client.post(url,
  52                                     data=self.req_data,
  53                                     format='json')
  54         self.failUnlessEqual(status.HTTP_409_CONFLICT, response.status_code)