Fix and test environment dispatch tasks

tests/routes/task_orders/test_new.py

@@ -329,7 +329,7 @@ def test_task_orders_submit_task_order(client, user_session, task_order):
         ({"_pdf": None, "number": "", "clins": []}, "step_1"),
         ({"number": "", "clins": []}, "step_2"),
         ({"number": "1234567890123", "clins": []}, "step_3"),
-        ({"number": "1234567890123", "create_clins": [1]}, "step_4"),
+        ({"number": "1234567890123", "create_clins": [{"number": 1}]}, "step_4"),
     ],
 )
 def test_task_orders_edit_redirects_to_latest_incomplete_step(

tests/test_access.py

@@ -541,7 +541,7 @@ def test_task_orders_new_get_routes(get_url_assert_status):
     task_order = TaskOrderFactory.create(
         creator=owner,
         portfolio=portfolio,
-        create_clins=["1234567890123456789012345678901234567890123"],
+        create_clins=[{"number": "1234567890123456789012345678901234567890123"}],
     )
 
     for route in get_routes:

tests/test_jobs.py

@@ -11,8 +11,16 @@ from atst.jobs import (
     do_create_environment,
     do_create_atat_admin_user,
     do_create_environment_baseline,
+    dispatch_create_environment,
+    dispatch_create_atat_admin_user,
+    dispatch_create_environment_baseline,
+)
+from tests.factories import (
+    EnvironmentFactory,
+    EnvironmentRoleFactory,
+    UserFactory,
+    PortfolioFactory,
 )
-from tests.factories import EnvironmentFactory, EnvironmentRoleFactory, UserFactory
 
 
 def test_environment_job_failure(celery_app, celery_worker):
@@ -105,3 +113,86 @@ def test_create_environment_baseline(csp, session):
     updated_environment = session.query(Environment).get(environment_id)
 
     assert updated_environment.baseline_info
+
+
+def test_dispatch_create_environment(session, monkeypatch):
+    portfolio = PortfolioFactory.create(
+        applications=[{"environments": [{}]}],
+        task_orders=[
+            {
+                "create_clins": [
+                    {
+                        "start_date": pendulum.now().subtract(days=1),
+                        "end_date": pendulum.now().add(days=1),
+                    }
+                ]
+            }
+        ],
+    )
+    mock = Mock()
+    monkeypatch.setattr("atst.jobs.create_environment", mock)
+    environment = portfolio.applications[0].environments[0]
+
+    dispatch_create_environment.run()
+
+    mock.delay.assert_called_once_with(
+        environment_id=environment.id, atat_user_id=environment.creator_id
+    )
+
+
+def test_dispatch_create_atat_admin_user(session, monkeypatch):
+    portfolio = PortfolioFactory.create(
+        applications=[
+            {"environments": [{"cloud_id": uuid4().hex, "root_user_info": None}]}
+        ],
+        task_orders=[
+            {
+                "create_clins": [
+                    {
+                        "start_date": pendulum.now().subtract(days=1),
+                        "end_date": pendulum.now().add(days=1),
+                    }
+                ]
+            }
+        ],
+    )
+    mock = Mock()
+    monkeypatch.setattr("atst.jobs.create_atat_admin_user", mock)
+    environment = portfolio.applications[0].environments[0]
+
+    dispatch_create_atat_admin_user.run()
+
+    mock.delay.assert_called_once_with(environment_id=environment.id)
+
+
+def test_dispatch_create_environment_baseline(session, monkeypatch):
+    portfolio = PortfolioFactory.create(
+        applications=[
+            {
+                "environments": [
+                    {
+                        "cloud_id": uuid4().hex,
+                        "root_user_info": {},
+                        "baseline_info": None,
+                    }
+                ]
+            }
+        ],
+        task_orders=[
+            {
+                "create_clins": [
+                    {
+                        "start_date": pendulum.now().subtract(days=1),
+                        "end_date": pendulum.now().add(days=1),
+                    }
+                ]
+            }
+        ],
+    )
+    mock = Mock()
+    monkeypatch.setattr("atst.jobs.create_environment_baseline", mock)
+    environment = portfolio.applications[0].environments[0]
+
+    dispatch_create_environment_baseline.run()
+
+    mock.delay.assert_called_once_with(environment_id=environment.id)