From c2fd6ffd719616a87922505b3b2ce332a7c6048c Mon Sep 17 00:00:00 2001
From: thuva4 <tharma.thuva@gmail.com>
Date: Mon, 14 May 2018 14:15:49 +0530
Subject: Add scenario client + tests

Change-Id: I9e90e58c5d6f327b0b947c2c9270ef9cdbe16f45
Signed-off-by: thuva4 <tharma.thuva@gmail.com>
---
 .../tests/unit/test_scenario_client.py             | 104 +++++++++++++++++++++
 1 file changed, 104 insertions(+)
 create mode 100644 testapi/testapi-client/testapiclient/tests/unit/test_scenario_client.py

(limited to 'testapi/testapi-client/testapiclient/tests/unit/test_scenario_client.py')

diff --git a/testapi/testapi-client/testapiclient/tests/unit/test_scenario_client.py b/testapi/testapi-client/testapiclient/tests/unit/test_scenario_client.py
new file mode 100644
index 0000000..6e9e0fa
--- /dev/null
+++ b/testapi/testapi-client/testapiclient/tests/unit/test_scenario_client.py
@@ -0,0 +1,104 @@
+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 ScenarioClientTest(utils.TestCommand):
+    def setUp(self):
+        super(ScenarioClientTest, self).setUp()
+        self.base_url = parse.urljoin(self.api_url, 'scenarios')
+        self.scenario_json = {
+            "installers": [],
+            "name": "test_scenario"
+        }
+        self.scenario_client = scenarios.ScenariosClient()
+        self.scenario_string = json.dumps(self.scenario_json)
+
+
+class ScenarioClientGetTest(ScenarioClientTest):
+
+    def setUp(self):
+        super(ScenarioClientGetTest, self).setUp()
+        self.scenarios_rsp = {'scenarios': [self.scenario_json]}
+
+    def test_get(self):
+        self.get_mock.return_value = fakes.FakeResponse(
+            data=self.scenarios_rsp)
+        self.scenario_client.get()
+        self.get_mock.assert_called_once_with(
+            self.base_url,
+            headers=clientmanager.ClientManager.headers)
+
+    def test_get_one(self):
+        self.get_mock.return_value = fakes.FakeResponse(
+            data=self.scenario_json)
+        self.scenario_client.get_one('def')
+        self.get_mock.assert_called_once_with(
+            self.base_url + '/def',
+            headers=clientmanager.ClientManager.headers)
+
+
+class ScenarioClientCreateTest(ScenarioClientTest):
+
+    def setUp(self):
+        super(ScenarioClientCreateTest, self).setUp()
+        self.succ_rsp = {
+            'href': '{}/{}'.format(
+                self.base_url, self.scenario_json.get('name'))
+        }
+
+    def test_create_success(self):
+        self.post_mock.return_value = fakes.FakeResponse(
+            data=self.succ_rsp)
+        self.scenario_client.create(self.scenario_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.scenario_client.create(self.scenario_json)
+
+
+class ScenarioClientDeleteTest(ScenarioClientTest):
+
+    def setUp(self):
+        super(ScenarioClientDeleteTest, self).setUp()
+
+    def test_delete_success(self):
+        self.delete_mock.return_value = fakes.FakeResponse()
+        self.scenario_client.delete('def')
+        self.delete_mock.assert_called_once_with(
+            self.base_url + '/def',
+            data=None,
+            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.scenario_client.delete('def')
+
+
+class ScenarioClientUpdateTest(ScenarioClientTest):
+
+    def setUp(self):
+        super(ScenarioClientUpdateTest, self).setUp()
+
+    def test_update_success(self):
+        self.put_mock.return_value = fakes.FakeResponse()
+        self.scenario_client.update(
+            'def', self.scenario_json)
+        self.put_mock.assert_called_once_with(
+            self.base_url + '/def',
+            data=self.scenario_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.scenario_client.update('def', self.scenario_json)
-- 
cgit 1.2.3-korg