From b364a1d2a6ed59b07716000e1502f2f096d0962c Mon Sep 17 00:00:00 2001
From: dandds <dan-ctr@friends.dds.mil>
Date: Sun, 2 Feb 2020 13:58:41 -0500
Subject: [PATCH] Celery wrapper for creating a user.

---
 atst/domain/csp/cloud/mock_cloud_provider.py |  5 +++++
 atst/jobs.py                                 | 21 ++++++++++----------
 2 files changed, 16 insertions(+), 10 deletions(-)

diff --git a/atst/domain/csp/cloud/mock_cloud_provider.py b/atst/domain/csp/cloud/mock_cloud_provider.py
index 5b9a8b34..96648b52 100644
--- a/atst/domain/csp/cloud/mock_cloud_provider.py
+++ b/atst/domain/csp/cloud/mock_cloud_provider.py
@@ -463,6 +463,11 @@ class MockCloudProvider(CloudProviderInterface):
             id=f"{AZURE_MGMNT_PATH}{payload.management_group_name}"
         )
 
+    def create_user(self, payload: UserCSPPayload):
+        self._maybe_raise(self.UNAUTHORIZED_RATE, GeneralCSPException)
+
+        return UserCSPResult(id=str(uuid4()))
+
     def get_credentials(self, scope="portfolio", tenant_id=None):
         return self.root_creds()
 
diff --git a/atst/jobs.py b/atst/jobs.py
index 6a91b03e..0ce664da 100644
--- a/atst/jobs.py
+++ b/atst/jobs.py
@@ -1,22 +1,23 @@
-from flask import current_app as app
 import pendulum
+from flask import current_app as app
 
 from atst.database import db
-from atst.queue import celery
-from atst.models import JobFailure
-from atst.domain.csp.cloud.exceptions import GeneralCSPException
-from atst.domain.csp.cloud import CloudProviderInterface
-from atst.domain.applications import Applications
-from atst.domain.environments import Environments
-from atst.domain.portfolios import Portfolios
 from atst.domain.application_roles import ApplicationRoles
-from atst.models.utils import claim_for_update, claim_many_for_update
-from atst.utils.localization import translate
+from atst.domain.applications import Applications
+from atst.domain.csp.cloud import CloudProviderInterface
+from atst.domain.csp.cloud.exceptions import GeneralCSPException
 from atst.domain.csp.cloud.models import (
     ApplicationCSPPayload,
     EnvironmentCSPPayload,
     UserCSPPayload,
 )
+from atst.domain.environment_roles import EnvironmentRoles
+from atst.domain.environments import Environments
+from atst.domain.portfolios import Portfolios
+from atst.models import JobFailure
+from atst.models.utils import claim_for_update, claim_many_for_update
+from atst.queue import celery
+from atst.utils.localization import translate
 
 
 class RecordFailure(celery.Task):