Add scenario client + tests

Change-Id: I9e90e58c5d6f327b0b947c2c9270ef9cdbe16f45
Signed-off-by: thuva4 <tharma.thuva@gmail.com>

7 files changed, 602 insertions, 0 deletions

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)
diff --git a/testapi/testapi-client/testapiclient/tests/unit/test_scenario_custom_client.py b/testapi/testapi-client/testapiclient/tests/unit/test_scenario_custom_client.py
new file mode 100644
index 0000000..7c6d62c
--- /dev/null
+++ b/testapi/testapi-client/testapiclient/tests/unit/test_scenario_custom_client.py
@@ -0,0 +1,106 @@
+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
+
+
+class CustomClientTest(utils.TestCommand):
+    def setUp(self):
+        super(CustomClientTest, self).setUp()
+        self.scenario_name = 'scenrio1'
+        self.base_url = parse.urljoin(
+            self.api_url,
+            'scenarios/{}/customs'.format(self.scenario_name))
+        self.custom_raw = 'custom'
+        self.custom_input = ['custom']
+        self.installer_name = 'installer'
+        self.version_name = 'version'
+        self.project_name = 'project'
+        self.custom_client = scenarios.CustomsClient()
+
+
+class CustomClientCreateTest(CustomClientTest):
+
+    def setUp(self):
+        super(CustomClientCreateTest, self).setUp()
+        self.succ_rsp = {
+            'href': '{}/{}'.format(
+                self.base_url, self.scenario_name)
+        }
+
+    def test_create_success(self):
+        self.post_mock.return_value = fakes.FakeResponse(
+            data=self.succ_rsp)
+        self.custom_client.create(
+            self.scenario_name, self.installer_name,
+            self.version_name, self.project_name,
+            self.custom_raw)
+        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.custom_client.create(
+                self.scenario_name, self.installer_name,
+                self.version_name, self.project_name,
+                self.custom_raw)
+
+
+class CustomClientDeleteTest(CustomClientTest):
+
+    def setUp(self):
+        super(CustomClientDeleteTest, self).setUp()
+
+    def test_delete_success(self):
+        self.delete_mock.return_value = fakes.FakeResponse()
+        self.custom_client.delete(
+            self.scenario_name, self.installer_name,
+            self.version_name, self.project_name,
+            self.custom_raw)
+        kall = self.delete_mock.call_args
+        args, kwargs = kall
+        self.assert_url(
+            args[0],
+            self.base_url +
+            '?installer=installer&version=version&project=project')
+
+    def test_delete_failure(self):
+        with testtools.ExpectedException(Exception, 'Delete failed: Error'):
+            self.delete_mock.return_value = utils.FAKE_FAILURE
+            self.custom_client.delete(
+                self.scenario_name, self.installer_name,
+                self.version_name, self.project_name,
+                self.custom_raw)
+
+
+class CustomClientUpdateTest(CustomClientTest):
+
+    def setUp(self):
+        super(CustomClientUpdateTest, self).setUp()
+
+    def test_update_success(self):
+        self.put_mock.return_value = fakes.FakeResponse()
+        self.custom_client.update(
+            self.scenario_name,
+            self.installer_name,
+            self.version_name,
+            self.project_name,
+            self.custom_raw)
+        kall = self.put_mock.call_args
+        args, kwargs = kall
+        self.assert_url(
+            args[0],
+            self.base_url +
+            '?installer=installer&version=version&project=project')
+
+    def test_update_failure(self):
+        with testtools.ExpectedException(Exception, 'Update failed: Error'):
+            self.put_mock.return_value = utils.FAKE_FAILURE
+            self.custom_client.update(
+                self.scenario_name,
+                self.installer_name,
+                self.version_name,
+                self.project_name,
+                self.custom_raw)
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)
diff --git a/testapi/testapi-client/testapiclient/tests/unit/test_scenario_project_client.py b/testapi/testapi-client/testapiclient/tests/unit/test_scenario_project_client.py
new file mode 100644
index 0000000..f8c3d60
--- /dev/null
+++ b/testapi/testapi-client/testapiclient/tests/unit/test_scenario_project_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
+
+
+class ProjectClientTest(utils.TestCommand):
+    def setUp(self):
+        super(ProjectClientTest, self).setUp()
+        self.scenario_name = 'scenrio1'
+        self.base_url = parse.urljoin(
+            self.api_url,
+            'scenarios/{}/projects'.format(self.scenario_name))
+        self.project_json = {
+            'trust_indicators': [],
+            'project': 'test-project',
+            'scores': [],
+            'customs': []
+        }
+        self.installer_name = 'installer'
+        self.version_name = 'version'
+        self.project_client = scenarios.ProjectsClient()
+        self.project_string = json.dumps(self.project_json)
+
+
+class ProjectClientCreateTest(ProjectClientTest):
+
+    def setUp(self):
+        super(ProjectClientCreateTest, self).setUp()
+        self.succ_rsp = {
+            'href': '{}/{}'.format(
+                self.base_url, self.project_json.get('name'))
+        }
+
+    def test_create_success(self):
+        self.post_mock.return_value = fakes.FakeResponse(
+            data=self.succ_rsp)
+        self.project_client.create(
+            self.scenario_name, self.installer_name,
+            self.version_name, self.project_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.project_client.create(
+                self.scenario_name, self.installer_name,
+                self.version_name, self.project_json)
+
+
+class ProjectClientDeleteTest(ProjectClientTest):
+
+    def setUp(self):
+        super(ProjectClientDeleteTest, self).setUp()
+
+    def test_delete_success(self):
+        self.delete_mock.return_value = fakes.FakeResponse()
+        self.project_client.delete(
+            self.scenario_name, self.installer_name,
+            self.version_name, 'def')
+        kall = self.delete_mock.call_args
+        args, kwargs = kall
+        self.assert_url(
+            args[0],
+            self.base_url + '?installer=installer&version=version')
+
+    def test_delete_failure(self):
+        with testtools.ExpectedException(Exception, 'Delete failed: Error'):
+            self.delete_mock.return_value = utils.FAKE_FAILURE
+            self.project_client.delete(
+                self.scenario_name, self.installer_name,
+                self.version_name, 'def')
+
+
+class ProjectClientUpdateTest(ProjectClientTest):
+
+    def setUp(self):
+        super(ProjectClientUpdateTest, self).setUp()
+
+    def test_update_success(self):
+        self.put_mock.return_value = fakes.FakeResponse()
+        self.project_client.update(
+            self.scenario_name,
+            self.installer_name,
+            self.version_name,
+            self.project_json)
+        kall = self.put_mock.call_args
+        args, kwargs = kall
+        self.assert_url(
+            args[0],
+            self.base_url + '?installer=installer&version=version')
+
+    def test_update_failure(self):
+        with testtools.ExpectedException(Exception, 'Update failed: Error'):
+            self.put_mock.return_value = utils.FAKE_FAILURE
+            self.project_client.update(
+                self.scenario_name,
+                self.installer_name,
+                self.version_name,
+                self.project_json)
diff --git a/testapi/testapi-client/testapiclient/tests/unit/test_scenario_score_client.py b/testapi/testapi-client/testapiclient/tests/unit/test_scenario_score_client.py
new file mode 100644
index 0000000..beebd47
--- /dev/null
+++ b/testapi/testapi-client/testapiclient/tests/unit/test_scenario_score_client.py
@@ -0,0 +1,53 @@
+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
+
+
+class ScoreClientTest(utils.TestCommand):
+    def setUp(self):
+        super(ScoreClientTest, self).setUp()
+        self.scenario_name = 'scenrio1'
+        self.base_url = parse.urljoin(
+            self.api_url,
+            'scenarios/{}/scores'.format(self.scenario_name))
+        self.score_json = {
+            'score': 'test_score1',
+            'date': '2018/01/2'
+        }
+        self.installer_name = 'installer'
+        self.version_name = 'version'
+        self.project_name = 'project'
+        self.score_client = scenarios.ScoresClient()
+        self.score_string = json.dumps(self.score_json)
+
+
+class ScoreClientCreateTest(ScoreClientTest):
+
+    def setUp(self):
+        super(ScoreClientCreateTest, self).setUp()
+        self.succ_rsp = {
+            'href': '{}/{}'.format(
+                self.base_url, self.score_json.get('name'))
+        }
+
+    def test_create_success(self):
+        self.post_mock.return_value = fakes.FakeResponse(
+            data=self.succ_rsp)
+        self.score_client.create(
+            self.scenario_name, self.installer_name,
+            self.version_name, self.project_name,
+            self.score_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.score_client.create(
+                self.scenario_name, self.installer_name,
+                self.version_name, self.project_name,
+                self.score_json)
diff --git a/testapi/testapi-client/testapiclient/tests/unit/test_scenario_trust_indicator_client.py b/testapi/testapi-client/testapiclient/tests/unit/test_scenario_trust_indicator_client.py
new file mode 100644
index 0000000..e44e2d2
--- /dev/null
+++ b/testapi/testapi-client/testapiclient/tests/unit/test_scenario_trust_indicator_client.py
@@ -0,0 +1,53 @@
+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
+
+
+class TrustIndicatorClientTest(utils.TestCommand):
+    def setUp(self):
+        super(TrustIndicatorClientTest, self).setUp()
+        self.scenario_name = 'scenrio1'
+        self.base_url = parse.urljoin(
+            self.api_url,
+            'scenarios/{}/trust_indicators'.format(self.scenario_name))
+        self.trust_indicator_json = {
+            'status': 'test_status',
+            'date': '2018/01/2'
+        }
+        self.installer_name = 'installer'
+        self.version_name = 'version'
+        self.project_name = 'project'
+        self.trust_indicator_client = scenarios.TrustIndicatorsClient()
+        self.trust_indicator_string = json.dumps(self.trust_indicator_json)
+
+
+class TrustIndicatorClientCreateTest(TrustIndicatorClientTest):
+
+    def setUp(self):
+        super(TrustIndicatorClientCreateTest, self).setUp()
+        self.succ_rsp = {
+            'href': '{}/{}'.format(
+                self.base_url, self.trust_indicator_json.get('name'))
+        }
+
+    def test_create_success(self):
+        self.post_mock.return_value = fakes.FakeResponse(
+            data=self.succ_rsp)
+        self.trust_indicator_client.create(
+            self.scenario_name, self.installer_name,
+            self.version_name, self.project_name,
+            self.trust_indicator_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.trust_indicator_client.create(
+                self.scenario_name, self.installer_name,
+                self.version_name, self.project_name,
+                self.trust_indicator_json)
diff --git a/testapi/testapi-client/testapiclient/tests/unit/test_scenario_version_client.py b/testapi/testapi-client/testapiclient/tests/unit/test_scenario_version_client.py
new file mode 100644
index 0000000..1ae2409
--- /dev/null
+++ b/testapi/testapi-client/testapiclient/tests/unit/test_scenario_version_client.py
@@ -0,0 +1,96 @@
+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 VersionClientTest(utils.TestCommand):
+    def setUp(self):
+        super(VersionClientTest, self).setUp()
+        self.scenario_name = 'scenrio1'
+        self.base_url = parse.urljoin(
+            self.api_url,
+            'scenarios/{}/versions'.format(self.scenario_name))
+        self.version_json = {
+            'projects': [],
+            'version': 'test-version',
+            'owner': 'test_owner'
+        }
+        self.installer_name = 'installer'
+        self.version_client = scenarios.VersionsClient()
+        self.version_string = json.dumps(self.version_json)
+
+
+class VersionClientCreateTest(VersionClientTest):
+
+    def setUp(self):
+        super(VersionClientCreateTest, self).setUp()
+        self.succ_rsp = {
+            'href': '{}/{}'.format(
+                self.base_url, self.version_json.get('name'))
+        }
+
+    def test_create_success(self):
+        self.post_mock.return_value = fakes.FakeResponse(
+            data=self.succ_rsp)
+        self.version_client.create(
+            self.scenario_name, self.installer_name, self.version_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.version_client.create(
+                self.scenario_name, self.installer_name, self.version_json)
+
+
+class VersionClientDeleteTest(VersionClientTest):
+
+    def setUp(self):
+        super(VersionClientDeleteTest, self).setUp()
+
+    def test_delete_success(self):
+        self.delete_mock.return_value = fakes.FakeResponse()
+        self.version_client.delete(
+            self.scenario_name, self.installer_name, 'def')
+        kall = self.delete_mock.call_args
+        args, kwargs = kall
+        self.assert_url(
+            args[0],
+            self.base_url + '?installer=installer')
+
+    def test_delete_failure(self):
+        with testtools.ExpectedException(Exception, 'Delete failed: Error'):
+            self.delete_mock.return_value = utils.FAKE_FAILURE
+            self.version_client.delete(
+                self.scenario_name, self.installer_name, 'def')
+
+
+class VersionClientUpdateTest(VersionClientTest):
+
+    def setUp(self):
+        super(VersionClientUpdateTest, self).setUp()
+
+    def test_update_success(self):
+        self.put_mock.return_value = fakes.FakeResponse()
+        self.version_client.update(
+            self.scenario_name,
+            self.installer_name,
+            self.version_json)
+        self.put_mock.assert_called_once_with(
+            self.base_url + '?installer=installer',
+            data=self.version_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.version_client.update(
+                self.scenario_name,
+                self.installer_name,
+                self.version_json)