From 6c0420d6c5e17fab36b15adf8970b760aaaac3f3 Mon Sep 17 00:00:00 2001
From: richard-dds <richard-ctr@friends.dds.mil>
Date: Mon, 9 Sep 2019 10:45:38 -0400
Subject: [PATCH] Read config to determine mock CSP type

---
 atst/domain/csp/__init__.py   | 12 ++++++++----
 atst/domain/csp/cloud.py      | 18 ++++++++++++------
 config/test.ini               |  1 +
 tests/domain/test_mock_csp.py |  4 ++--
 4 files changed, 23 insertions(+), 12 deletions(-)

diff --git a/atst/domain/csp/__init__.py b/atst/domain/csp/__init__.py
index 1e288928..8c460ea1 100644
--- a/atst/domain/csp/__init__.py
+++ b/atst/domain/csp/__init__.py
@@ -4,22 +4,24 @@ from .reports import MockReportingProvider
 
 
 class MockCSP:
-    def __init__(self, app):
-        self.cloud = MockCloudProvider()
+    def __init__(self, app, test_mode=False):
+        self.cloud = MockCloudProvider(
+            app.config, with_delay=(not test_mode), with_failure=(not test_mode)
+        )
         self.files = MockUploader(app)
         self.reports = MockReportingProvider()
 
 
 class AzureCSP:
     def __init__(self, app):
-        self.cloud = MockCloudProvider()
+        self.cloud = MockCloudProvider(app.config)
         self.files = AzureUploader(app.config)
         self.reports = MockReportingProvider()
 
 
 class AwsCSP:
     def __init__(self, app):
-        self.cloud = MockCloudProvider()
+        self.cloud = MockCloudProvider(app.config)
         self.files = AwsUploader(app.config)
         self.reports = MockReportingProvider()
 
@@ -29,5 +31,7 @@ def make_csp_provider(app, csp=None):
         app.csp = AwsCSP(app)
     elif csp == "azure":
         app.csp = AzureCSP(app)
+    elif csp == "mock-test":
+        app.csp = MockCSP(app, test_mode=True)
     else:
         app.csp = MockCSP(app)
diff --git a/atst/domain/csp/cloud.py b/atst/domain/csp/cloud.py
index 9a1f2258..5f8689c1 100644
--- a/atst/domain/csp/cloud.py
+++ b/atst/domain/csp/cloud.py
@@ -12,6 +12,9 @@ class GeneralCSPException(Exception):
 
 
 class CloudProviderInterface:
+    def root_creds() -> Dict:
+        raise NotImplementedError()
+
     def create_environment(
         self, auth_credentials: Dict, user: User, environment: Environment
     ) -> str:
@@ -128,7 +131,7 @@ class MockCloudProvider(CloudProviderInterface):
     ENV_CREATE_FAILURE_PCT = 12
     ATAT_ADMIN_CREATE_FAILURE_PCT = 12
 
-    def __init__(self, with_delay=True, with_failure=True):
+    def __init__(self, config, with_delay=True, with_failure=True):
         from time import sleep
         import random
 
@@ -137,6 +140,9 @@ class MockCloudProvider(CloudProviderInterface):
         self._sleep = sleep
         self._random = random
 
+    def root_creds(self):
+        return self._auth_credentials
+
     def create_environment(self, auth_credentials, user, environment):
         self._authorize(auth_credentials)
 
@@ -159,7 +165,7 @@ class MockCloudProvider(CloudProviderInterface):
             GeneralCSPException("Could not create admin user."),
         )
 
-        return {"id": self._id(), "credentials": {}}
+        return {"id": self._id(), "credentials": self._auth_credentials}
 
     def create_environment_baseline(self, auth_credentials, csp_environment_id):
         self._authorize(auth_credentials)
@@ -172,10 +178,10 @@ class MockCloudProvider(CloudProviderInterface):
         )
 
         return {
-            CSPRole.BASIC_ACCESS: self._id(),
-            CSPRole.NETWORK_ADMIN: self._id(),
-            CSPRole.BUSINESS_READ: self._id(),
-            CSPRole.TECHNICAL_READ: self._id(),
+            CSPRole.BASIC_ACCESS.value: self._id(),
+            CSPRole.NETWORK_ADMIN.value: self._id(),
+            CSPRole.BUSINESS_READ.value: self._id(),
+            CSPRole.TECHNICAL_READ.value: self._id(),
         }
 
     def create_or_update_user(self, auth_credentials, user_info, csp_role_id):
diff --git a/config/test.ini b/config/test.ini
index c323f599..37a9e629 100644
--- a/config/test.ini
+++ b/config/test.ini
@@ -6,3 +6,4 @@ CRL_STORAGE_CONTAINER = tests/fixtures/crl
 WTF_CSRF_ENABLED = false
 STORAGE_PROVIDER=LOCAL
 PRESERVE_CONTEXT_ON_EXCEPTION = false
+CSP=mock-test
diff --git a/tests/domain/test_mock_csp.py b/tests/domain/test_mock_csp.py
index 927bf304..27495f4c 100644
--- a/tests/domain/test_mock_csp.py
+++ b/tests/domain/test_mock_csp.py
@@ -2,12 +2,12 @@ import pytest
 
 from atst.domain.csp import MockCloudProvider
 
-CREDENTIALS = MockCloudProvider()._auth_credentials
+CREDENTIALS = MockCloudProvider(config={})._auth_credentials
 
 
 @pytest.fixture
 def mock_csp():
-    return MockCloudProvider(with_delay=False, with_failure=False)
+    return MockCloudProvider(config={}, with_delay=False, with_failure=False)
 
 
 def test_create_environment(mock_csp: MockCloudProvider):