Merge pull request #1062 from dod-ccpo/env-provisioning-task

Environment provisioning background jobs

tests/models/test_environments.py

@@ -1,6 +1,7 @@
+import pytest
+
 from atst.models import AuditEvent
 from atst.models.environment_role import CSPRole
-from atst.domain.environments import Environments
 from atst.domain.applications import Applications
 
 from tests.factories import *
@@ -12,7 +13,11 @@ def test_add_user_to_environment():
 
     portfolio = PortfolioFactory.create(owner=owner)
     application = Applications.create(
-        portfolio, "my test application", "It's mine.", ["dev", "staging", "prod"]
+        portfolio.owner,
+        portfolio,
+        "my test application",
+        "It's mine.",
+        ["dev", "staging", "prod"],
     )
     dev_environment = application.environments[0]
 
@@ -42,3 +47,29 @@ def test_audit_event_for_environment_deletion(session):
     before, after = update_event.changed_state["deleted"]
     assert not before
     assert after
+
+
+@pytest.mark.parametrize(
+    "env_data,expected_status",
+    [
+        [
+            {"cloud_id": None, "root_user_info": None, "baseline_info": None},
+            Environment.ProvisioningStatus.PENDING,
+        ],
+        [
+            {"cloud_id": 1, "root_user_info": None, "baseline_info": None},
+            Environment.ProvisioningStatus.PENDING,
+        ],
+        [
+            {"cloud_id": 1, "root_user_info": {}, "baseline_info": None},
+            Environment.ProvisioningStatus.PENDING,
+        ],
+        [
+            {"cloud_id": 1, "root_user_info": {}, "baseline_info": {}},
+            Environment.ProvisioningStatus.COMPLETED,
+        ],
+    ],
+)
+def test_environment_provisioning_status(env_data, expected_status):
+    environment = EnvironmentFactory.create(**env_data)
+    assert environment.provisioning_status == expected_status