Merge "Add scenario client + tests"

1 files changed, 86 insertions, 0 deletions

diff --git a/testapi/testapi-client/testapiclient/tests/unit/test_scenario_installer_client.py b/testapi/testapi-client/testapiclient/tests/unit/test_scenario_installer_client.py
new file mode 100644
index 0000000..71ba150
--- /dev/null
+++ b/testapi/testapi-client/testapiclient/tests/unit/test_scenario_installer_client.py
@@ -0,0 +1,86 @@
+import json
+
+from six.moves.urllib import parse
+import testtools
+
+from testapiclient.client import scenarios
+from testapiclient.tests.unit import fakes
+from testapiclient.tests.unit import utils
+from testapiclient.utils import clientmanager
+
+
+class InstallerClientTest(utils.TestCommand):
+    def setUp(self):
+        super(InstallerClientTest, self).setUp()
+        self.scenario_name = 'scenrio1'
+        self.base_url = parse.urljoin(
+            self.api_url,
+            'scenarios/{}/installers'.format(self.scenario_name))
+        self.installer_json = {
+            'versions': [],
+            'installer': 'test-installer',
+        }
+        self.installer_client = scenarios.InstallersClient()
+        self.installer_string = json.dumps(self.installer_json)
+
+
+class InstallerClientCreateTest(InstallerClientTest):
+
+    def setUp(self):
+        super(InstallerClientCreateTest, self).setUp()
+        self.succ_rsp = {
+            'href': '{}/{}'.format(
+                self.base_url, self.installer_json.get('name'))
+        }
+
+    def test_create_success(self):
+        self.post_mock.return_value = fakes.FakeResponse(
+            data=self.succ_rsp)
+        self.installer_client.create(self.scenario_name, self.installer_json)
+        self.post_mock.assert_called_once()
+
+    def test_create_failure(self):
+        with testtools.ExpectedException(Exception, 'Create failed: Error'):
+            self.post_mock.return_value = utils.FAKE_FAILURE
+            self.installer_client.create(
+                self.scenario_name, self.installer_json)
+
+
+class InstallerClientDeleteTest(InstallerClientTest):
+
+    def setUp(self):
+        super(InstallerClientDeleteTest, self).setUp()
+
+    def test_delete_success(self):
+        self.delete_mock.return_value = fakes.FakeResponse()
+        self.installer_client.delete(self.scenario_name, 'def')
+        self.delete_mock.assert_called_once_with(
+            self.base_url,
+            data=json.dumps(['def']),
+            headers=clientmanager.ClientManager.headers)
+
+    def test_delete_failure(self):
+        with testtools.ExpectedException(Exception, 'Delete failed: Error'):
+            self.delete_mock.return_value = utils.FAKE_FAILURE
+            self.installer_client.delete(self.scenario_name, 'def')
+
+
+class InstallerClientUpdateTest(InstallerClientTest):
+
+    def setUp(self):
+        super(InstallerClientUpdateTest, self).setUp()
+
+    def test_update_success(self):
+        self.put_mock.return_value = fakes.FakeResponse()
+        self.installer_client.update(
+            self.scenario_name,
+            self.installer_json)
+        self.put_mock.assert_called_once_with(
+            self.base_url,
+            data=self.installer_string,
+            headers=clientmanager.ClientManager.headers)
+
+    def test_update_failure(self):
+        with testtools.ExpectedException(Exception, 'Update failed: Error'):
+            self.put_mock.return_value = utils.FAKE_FAILURE
+            self.installer_client.update('def', self.installer_json)