1 files changed, 122 insertions, 0 deletions
diff --git a/functest/tests/unit/opnfv_tests/vnf/ims/test_orchestrator_cloudify.py b/functest/tests/unit/opnfv_tests/vnf/ims/test_orchestrator_cloudify.py
new file mode 100644
index 000000000..620b0216f
--- /dev/null
+++ b/functest/tests/unit/opnfv_tests/vnf/ims/test_orchestrator_cloudify.py
@@ -0,0 +1,122 @@
+#!/usr/bin/env python
+
+# All rights reserved. This program and the accompanying materials
+# are made available under the terms of the Apache License, Version 2.0
+# which accompanies this distribution, and is available at
+# http://www.apache.org/licenses/LICENSE-2.0
+
+import logging
+import unittest
+
+import mock
+
+from functest.opnfv_tests.vnf.ims import orchestrator_cloudify
+
+
+class ImsVnfTesting(unittest.TestCase):
+
+    logging.disable(logging.CRITICAL)
+
+    def setUp(self):
+        self.orchestrator = orchestrator_cloudify.Orchestrator('test_dir')
+        self.bp = {'file_name': 'test_file',
+                   'destination_folder': 'test_folder',
+                   'url': 'test_url',
+                   'branch': 'test_branch'}
+
+    def test_download_manager_blueprint_download_blueprint_failed(self):
+        self.orchestrator.manager_blueprint = False
+        with mock.patch.object(self.orchestrator, '_download_blueprints',
+                               return_value=False), \
+            mock.patch('functest.opnfv_tests.vnf.ims.orchestrator_cloudify.'
+                       'exit') as mock_exit:
+            self.orchestrator.download_manager_blueprint('test_url',
+                                                         'test_branch')
+            mock_exit.assert_any_call(-1)
+
+    def test_download_manager_blueprint_download_blueprint_passed(self):
+        self.orchestrator.manager_blueprint = False
+        with mock.patch.object(self.orchestrator, '_download_blueprints',
+                               return_value=True):
+            self.orchestrator.download_manager_blueprint('test_url',
+                                                         'test_branch')
+            self.assertEqual(self.orchestrator.manager_blueprint,
+                             True)
+
+    def test_deploy_manager_failed(self):
+        self.orchestrator.manager_blueprint = True
+        with mock.patch('__builtin__.open', mock.mock_open()), \
+            mock.patch('functest.opnfv_tests.vnf.ims.orchestrator_cloudify.'
+                       'os.remove'), \
+            mock.patch('functest.opnfv_tests.vnf.ims.orchestrator_cloudify.'
+                       'execute_command', return_value='error'):
+            self.assertEqual(self.orchestrator.deploy_manager(),
+                             'error')
+            self.assertEqual(self.orchestrator.manager_up,
+                             False)
+
+    def test_deploy_manager_passed(self):
+        self.orchestrator.manager_blueprint = True
+        with mock.patch('__builtin__.open', mock.mock_open()), \
+            mock.patch('functest.opnfv_tests.vnf.ims.orchestrator_cloudify.'
+                       'os.remove'), \
+            mock.patch('functest.opnfv_tests.vnf.ims.orchestrator_cloudify.'
+                       'execute_command', return_value=''):
+            self.orchestrator.deploy_manager()
+            self.assertEqual(self.orchestrator.manager_up,
+                             True)
+
+    def test_undeploy_manager_passed(self):
+        with mock.patch('functest.opnfv_tests.vnf.ims.orchestrator_cloudify.'
+                        'execute_command', return_value=''):
+            self.orchestrator.deploy_manager()
+            self.assertEqual(self.orchestrator.manager_up,
+                             False)
+
+    def test_dwnld_upload_and_depl_blueprint_dwnld_blueprint_failed(self):
+        with mock.patch.object(self.orchestrator, '_download_blueprints',
+                               return_value=False), \
+            mock.patch('functest.opnfv_tests.vnf.ims.orchestrator_cloudify.'
+                       'exit', side_effect=Exception) as mock_exit, \
+                self.assertRaises(Exception):
+            self.orchestrator.download_upload_and_deploy_blueprint(self.bp,
+                                                                   'cfig',
+                                                                   'bpn',
+                                                                   'dpn')
+            mock_exit.assert_any_call(-1)
+
+    def test_dwnld_upload_and_depl_blueprint_failed(self):
+        with mock.patch.object(self.orchestrator, '_download_blueprints',
+                               return_value=True), \
+            mock.patch('__builtin__.open', mock.mock_open()), \
+            mock.patch('functest.opnfv_tests.vnf.ims.orchestrator_cloudify.'
+                       'execute_command', return_value='error'):
+            r = self.orchestrator.download_upload_and_deploy_blueprint(self.bp,
+                                                                       'cfig',
+                                                                       'bpn',
+                                                                       'dpn')
+            self.assertEqual(r, 'error')
+
+    def test__download_blueprints_failed(self):
+        with mock.patch('functest.opnfv_tests.vnf.ims.orchestrator_cloudify.'
+                        'shutil.rmtree'), \
+            mock.patch('functest.opnfv_tests.vnf.ims.orchestrator_cloudify.'
+                       'Repo.clone_from', side_effect=Exception):
+            self.assertEqual(self.orchestrator._download_blueprints('bp_url',
+                                                                    'branch',
+                                                                    'dest'),
+                             False)
+
+    def test__download_blueprints_passed(self):
+        with mock.patch('functest.opnfv_tests.vnf.ims.orchestrator_cloudify.'
+                        'shutil.rmtree'), \
+            mock.patch('functest.opnfv_tests.vnf.ims.orchestrator_cloudify.'
+                       'Repo.clone_from'):
+            self.assertEqual(self.orchestrator._download_blueprints('bp_url',
+                                                                    'branch',
+                                                                    'dest'),
+                             True)
+
+
+if __name__ == "__main__":
+    unittest.main(verbosity=2)