Add query for environments pending creation

tests/test_jobs.py

@@ -1,18 +1,26 @@
+import pendulum
 import pytest
 
 from atst.jobs import RecordEnvironmentFailure, RecordEnvironmentRoleFailure
 from tests.factories import EnvironmentFactory, EnvironmentRoleFactory, UserFactory
 from uuid import uuid4
 from unittest.mock import Mock
-
+from tests.factories import (
+    UserFactory,
+    PortfolioFactory,
+    CLINFactory,
+    TaskOrderFactory,
+    EnvironmentFactory
+)
+from atst.domain.csp.cloud import MockCloudProvider
 from atst.jobs import (
     do_create_environment,
     do_create_atat_admin_user,
     do_create_environment_baseline,
+    environments_to_create,
 )
 from atst.models import Environment
 
-
 def test_environment_job_failure(celery_app, celery_worker):
     @celery_app.task(bind=True, base=RecordEnvironmentFailure)
     def _fail_hard(self, environment_id=None):
@@ -47,8 +55,10 @@ def test_environment_role_job_failure(celery_app, celery_worker):
     assert role.job_failures
     job_failure = role.job_failures[0]
     assert job_failure.task == task
-from tests.factories import EnvironmentFactory, UserFactory
-from atst.domain.csp.cloud import MockCloudProvider
+
+now = pendulum.now()
+yesterday = now.subtract(days=1)
+tomorrow = now.add(days=1)
 
 
 @pytest.fixture(autouse=True, scope="function")
@@ -99,3 +109,40 @@ def test_create_environment_baseline(csp, session):
     updated_environment = session.query(Environment).get(environment_id)
 
     assert updated_environment.baseline_info
+
+
+def create_portfolio_with_clins(start_and_end_dates):
+    return PortfolioFactory.create(
+        applications=[
+            {
+                "name": "Mos Eisley",
+                "description": "Where Han shot first",
+                "environments": [{"name": "thebar"}],
+            }
+        ],
+        task_orders=[
+            TaskOrderFactory.create(
+                clins=[
+                    CLINFactory.create(start_date=start_date, end_date=end_date)
+                    for (start_date, end_date) in start_and_end_dates
+                ]
+            )
+        ],
+    )
+
+
+def test_dispatch_query_with_expired_clins(session):
+    create_portfolio_with_clins([(yesterday, yesterday)])
+    assert len(environments_to_create(pendulum.now())) == 0
+
+
+def test_dispatch_query_with_active_clins(session):
+    portfolio = create_portfolio_with_clins([(yesterday, tomorrow)])
+    environments_to_create(pendulum.now()) == [
+        portfolio.applications[0].environments[0].id
+    ]
+
+
+def test_dispatch_query_with_future_clins(session):
+    create_portfolio_with_clins([(tomorrow, tomorrow)])
+    assert len(environments_to_create(pendulum.now())) == 0