Merge "Add tests for commands"

contrib/fuel_bootstrap/fuel_bootstrap_cli/fuel_bootstrap/tests/test_activate_command.py

@@ -0,0 +1,31 @@
+# -*- coding: utf-8 -*-
+
+#    Copyright 2016 Mirantis, Inc.
+#
+#    Licensed under the Apache License, Version 2.0 (the "License"); you may
+#    not use this file except in compliance with the License. You may obtain
+#    a copy of the License at
+#
+#         http://www.apache.org/licenses/LICENSE-2.0
+#
+#    Unless required by applicable law or agreed to in writing, software
+#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+#    License for the specific language governing permissions and limitations
+#    under the License.
+import mock
+
+from fuel_bootstrap.tests import base
+
+UUID = 'fake_uuid'
+
+
+class TestActivateCommand(base.BaseTest):
+
+    @mock.patch('fuel_bootstrap.utils.bootstrap_image.activate',
+                return_value=UUID)
+    def test_parser(self, mock_activate):
+        self.app.run(['activate', UUID])
+        mock_activate.assert_called_once_with(UUID)
+        self.assertEqual("Bootstrap image {0} has been activated.\n"
+                         .format(UUID), self.app.stdout.getvalue())

contrib/fuel_bootstrap/fuel_bootstrap_cli/fuel_bootstrap/tests/test_build_command.py

@@ -0,0 +1,75 @@
+# -*- coding: utf-8 -*-
+
+#    Copyright 2016 Mirantis, Inc.
+#
+#    Licensed under the Apache License, Version 2.0 (the "License"); you may
+#    not use this file except in compliance with the License. You may obtain
+#    a copy of the License at
+#
+#         http://www.apache.org/licenses/LICENSE-2.0
+#
+#    Unless required by applicable law or agreed to in writing, software
+#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+#    License for the specific language governing permissions and limitations
+#    under the License.
+import copy
+
+import mock
+
+from fuel_bootstrap.tests import base
+
+PARSED_ARGS = {'extend_kopts': None,
+               'no_compress': False,
+               'output_dir': None,
+               'image_build_dir': None,
+               'post_script_file': None,
+               'root_ssh_authorized_file': None,
+               'activate': False,
+               'ubuntu_release': None,
+               'root_password': None,
+               'no_default_direct_repo_addr': False,
+               'https_proxy': None,
+               'http_proxy': None,
+               'direct_repo_addr': None,
+               'label': None,
+               'repos': None,
+               'kernel_flavor': None,
+               'certs': None,
+               'extra_dirs': None,
+               'no_default_packages': False,
+               'no_default_extra_dirs': False,
+               'packages': None}
+
+UUID = 'fake_uuid'
+PATH = 'fake_path'
+
+
+class TestBuildCommand(base.BaseTest):
+    @mock.patch('fuel_bootstrap.utils.bootstrap_image.make_bootstrap',
+                return_value=(UUID, PATH))
+    def test_parser(self, mock_make_bootstrap):
+        self.app.run(['build'])
+        mock_make_bootstrap.assert_called_once_with(PARSED_ARGS)
+        self.assertEqual("Bootstrap image {0} has been built: {1}\n"
+                         .format(UUID, PATH),
+                         self.app.stdout.getvalue())
+
+    @mock.patch('fuel_bootstrap.utils.bootstrap_image.activate',
+                return_value=(UUID, PATH))
+    @mock.patch('fuel_bootstrap.utils.bootstrap_image.import_image',
+                return_value=UUID)
+    @mock.patch('fuel_bootstrap.utils.bootstrap_image.make_bootstrap',
+                return_value=(UUID, PATH))
+    def test_parser_activate(self, mock_make_bootstrap,
+                             mock_import, mock_activate):
+        self.app.run(['build', '--activate'])
+        parsed_args = copy.deepcopy(PARSED_ARGS)
+        parsed_args['activate'] = True
+        mock_make_bootstrap.assert_called_once_with(parsed_args)
+        mock_import.assert_called_once_with(PATH)
+        mock_activate.assert_called_once_with(UUID)
+        self.assertEqual("Bootstrap image {0} has been built: {1}\n"
+                         "Bootstrap image {0} has been activated.\n"
+                         .format(UUID, PATH),
+                         self.app.stdout.getvalue())

contrib/fuel_bootstrap/fuel_bootstrap_cli/fuel_bootstrap/tests/test_delete_command.py

@@ -0,0 +1,32 @@
+# -*- coding: utf-8 -*-
+
+#    Copyright 2016 Mirantis, Inc.
+#
+#    Licensed under the Apache License, Version 2.0 (the "License"); you may
+#    not use this file except in compliance with the License. You may obtain
+#    a copy of the License at
+#
+#         http://www.apache.org/licenses/LICENSE-2.0
+#
+#    Unless required by applicable law or agreed to in writing, software
+#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+#    License for the specific language governing permissions and limitations
+#    under the License.
+
+import mock
+
+from fuel_bootstrap.tests import base
+
+UUID = 'fake_uuid'
+
+
+class TestDeleteCommand(base.BaseTest):
+
+    @mock.patch('fuel_bootstrap.utils.bootstrap_image.delete',
+                return_value=UUID)
+    def test_parser(self, mock_delete):
+        self.app.run(['delete', UUID])
+        mock_delete.assert_called_once_with(UUID)
+        self.assertEqual("Bootstrap image {0} has been deleted.\n"
+                         .format(UUID), self.app.stdout.getvalue())

contrib/fuel_bootstrap/fuel_bootstrap_cli/fuel_bootstrap/tests/test_import_command.py

@@ -0,0 +1,45 @@
+# -*- coding: utf-8 -*-
+
+#    Copyright 2016 Mirantis, Inc.
+#
+#    Licensed under the Apache License, Version 2.0 (the "License"); you may
+#    not use this file except in compliance with the License. You may obtain
+#    a copy of the License at
+#
+#         http://www.apache.org/licenses/LICENSE-2.0
+#
+#    Unless required by applicable law or agreed to in writing, software
+#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+#    License for the specific language governing permissions and limitations
+#    under the License.
+
+import mock
+
+from fuel_bootstrap.tests import base
+
+UUID = 'fake_uuid'
+PATH = 'fake_path'
+
+
+class TestImportCommand(base.BaseTest):
+
+    @mock.patch('fuel_bootstrap.utils.bootstrap_image.import_image',
+                return_value=UUID)
+    def test_parser(self, mock_import):
+        self.app.run(['import', PATH])
+        mock_import.assert_called_once_with(PATH)
+        self.assertEqual("Bootstrap image {0} has been imported.\n"
+                         .format(UUID), self.app.stdout.getvalue())
+
+    @mock.patch('fuel_bootstrap.utils.bootstrap_image.activate',
+                return_value=UUID)
+    @mock.patch('fuel_bootstrap.utils.bootstrap_image.import_image',
+                return_value=UUID)
+    def test_parser_activate(self, mock_import, mock_activate):
+        self.app.run(['import', PATH, '--activate'])
+        mock_import.assert_called_once_with(PATH)
+        mock_activate.assert_called_once_with(UUID)
+        self.assertEqual("Bootstrap image {0} has been imported.\n"
+                         "Bootstrap image {0} has been activated\n"
+                         .format(UUID), self.app.stdout.getvalue())