functest/tests/unit/vnf/ims/test_clearwater.py

   1 #!/usr/bin/env python
   2
   3 # All rights reserved. This program and the accompanying materials
   4 # are made available under the terms of the Apache License, Version 2.0
   5 # which accompanies this distribution, and is available at
   6 # http://www.apache.org/licenses/LICENSE-2.0
   7
   8 import logging
   9 import unittest
  10
  11 import mock
  12
  13 from functest.opnfv_tests.vnf.ims import clearwater
  14 from functest.opnfv_tests.vnf.ims import orchestrator_cloudify
  15
  16
  17 class ClearwaterTesting(unittest.TestCase):
  18
  19     def setUp(self):
  20         self.clearwater = clearwater.Clearwater()
  21         self.orchestrator = orchestrator_cloudify.Orchestrator('test_dir')
  22         self.clearwater.orchestrator = self.orchestrator
  23         self.clearwater.dep_name = 'test_dep_name'
  24         self.bp = {'file_name': 'test_file',
  25                    'destination_folder': 'test_folder',
  26                    'url': 'test_url',
  27                    'branch': 'test_branch'}
  28
  29     def test_deploy_vnf_blueprint_download_failed(self):
  30         with mock.patch.object(self.clearwater.orchestrator,
  31                                'download_upload_and_deploy_blueprint',
  32                                return_value='error'):
  33             self.assertEqual(self.clearwater.deploy_vnf(self.bp),
  34                              'error')
  35
  36     def test_deploy_vnf_blueprint_download_passed(self):
  37         with mock.patch.object(self.clearwater.orchestrator,
  38                                'download_upload_and_deploy_blueprint',
  39                                return_value=''):
  40             self.clearwater.deploy_vnf(self.bp)
  41             self.assertEqual(self.clearwater.deploy, True)
  42
  43     def test_undeploy_vnf_deployment_passed(self):
  44         with mock.patch.object(self.clearwater.orchestrator,
  45                                'undeploy_deployment'):
  46             self.clearwater.deploy = True
  47             self.clearwater.undeploy_vnf()
  48             self.assertEqual(self.clearwater.deploy, False)
  49
  50     def test_undeploy_vnf_deployment_with_undeploy(self):
  51         with mock.patch.object(self.clearwater.orchestrator,
  52                                'undeploy_deployment') as m:
  53             self.clearwater.deploy = False
  54             self.clearwater.undeploy_vnf(),
  55             self.assertEqual(self.clearwater.deploy, False)
  56             self.assertFalse(m.called)
  57
  58             self.clearwater.orchestrator = None
  59             self.clearwater.deploy = True
  60             self.clearwater.undeploy_vnf(),
  61             self.assertEqual(self.clearwater.deploy, True)
  62
  63             self.clearwater.deploy = False
  64             self.clearwater.undeploy_vnf(),
  65             self.assertEqual(self.clearwater.deploy, False)
  66
  67     def test_set_methods(self):
  68         self.clearwater.set_orchestrator(self.orchestrator)
  69         self.assertTrue(self.clearwater.orchestrator, self.orchestrator)
  70         self.clearwater.set_flavor_id('test_flavor_id')
  71         self.assertTrue(self.clearwater.config['flavor_id'], 'test_flavor_id')
  72         self.clearwater.set_image_id('test_image_id')
  73         self.assertTrue(self.clearwater.config['image_id'], 'test_image_id')
  74         self.clearwater.set_agent_user('test_user')
  75         self.assertTrue(self.clearwater.config['agent_user'], 'test_user')
  76         self.clearwater.set_external_network_name('test_network')
  77         self.assertTrue(self.clearwater.config['external_network_name'],
  78                         'test_network')
  79         self.clearwater.set_public_domain('test_domain')
  80         self.assertTrue(self.clearwater.config['public_domain'],
  81                         'test_domain')
  82
  83 if __name__ == "__main__":
  84     logging.disable(logging.CRITICAL)
  85     unittest.main(verbosity=2)