fix: prevent past ttl values for environment and snapshot

Author: eakmanrq

sqlmesh/core/config/root.py

@@ -41,6 +41,7 @@
 from sqlmesh.core.loader import Loader, SqlMeshLoader
 from sqlmesh.core.notification_target import NotificationTarget
 from sqlmesh.core.user import User
+from sqlmesh.utils.date import to_timestamp, now, now_timestamp
 from sqlmesh.utils.errors import ConfigError
 from sqlmesh.utils.pydantic import field_validator, model_validator
 
@@ -97,8 +98,8 @@ class Config(BaseConfig):
     default_gateway: str = ""
     notification_targets: t.List[NotificationTarget] = []
     project: str = ""
-    snapshot_ttl: str = c.DEFAULT_SNAPSHOT_TTL
-    environment_ttl: t.Optional[str] = c.DEFAULT_ENVIRONMENT_TTL
+    snapshot_ttl: NoPastTTLString = c.DEFAULT_SNAPSHOT_TTL
+    environment_ttl: t.Optional[NoPastTTLString] = c.DEFAULT_ENVIRONMENT_TTL
     ignore_patterns: t.List[str] = c.IGNORE_PATTERNS
     time_column_format: str = c.DEFAULT_TIME_COLUMN_FORMAT
     users: t.List[User] = []
@@ -302,3 +303,20 @@ def dialect(self) -> t.Optional[str]:
     @property
     def fingerprint(self) -> str:
         return str(zlib.crc32(pickle.dumps(self.dict(exclude={"loader", "notification_targets"}))))
+
+
+def validate_no_past_ttl(v: str) -> str:
+    current_time = now()
+    if to_timestamp(v, relative_base=current_time) < to_timestamp(current_time):
+        raise ValueError(
+            f"TTL '{v}' is in the past. Please specify a relative time in the future. Ex: `in 1 week` instead of `1 week`."
+        )
+    return v
+
+
+if t.TYPE_CHECKING:
+    NoPastTTLString = str
+else:
+    from pydantic.functional_validators import BeforeValidator
+
+    NoPastTTLString = t.Annotated[str, BeforeValidator(validate_no_past_ttl)]

tests/integrations/github/cicd/test_config.py

@@ -200,3 +200,48 @@ def test_validation(tmp_path):
         ValueError, match="merge_method must be set if enable_deploy_command is True"
     ):
         load_config_from_paths(Config, project_paths=[tmp_path / "config.yaml"])
+
+
+def test_ttl_in_past(tmp_path):
+    create_temp_file(
+        tmp_path,
+        pathlib.Path("config.yaml"),
+        """
+environment_ttl: in 1 week
+model_defaults:
+    dialect: duckdb
+""",
+    )
+
+    config = load_config_from_paths(Config, project_paths=[tmp_path / "config.yaml"])
+    assert config.environment_ttl == "in 1 week"
+
+    create_temp_file(
+        tmp_path,
+        pathlib.Path("config.yaml"),
+        """
+environment_ttl: 1 week
+model_defaults:
+    dialect: duckdb
+""",
+    )
+    with pytest.raises(
+        ValueError,
+        match="TTL '1 week' is in the past. Please specify a relative time in the future. Ex: `in 1 week` instead of `1 week`.",
+    ):
+        load_config_from_paths(Config, project_paths=[tmp_path / "config.yaml"])
+
+        create_temp_file(
+            tmp_path,
+            pathlib.Path("config.yaml"),
+            """
+snapshot_ttl: 1 week
+model_defaults:
+    dialect: duckdb
+    """,
+        )
+        with pytest.raises(
+            ValueError,
+            match="TTL '1 week' is in the past. Please specify a relative time in the future. Ex: `in 1 week` instead of `1 week`.",
+        ):
+            load_config_from_paths(Config, project_paths=[tmp_path / "config.yaml"])