nits

dbt-labs · Dec 19, 2024 · 4c388e4 · 4c388e4
1 parent 9df8c62
commit 4c388e4
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/tests/unit/contracts/graph/test_manifest.py b/tests/unit/contracts/graph/test_manifest.py
@@ -97,6 +97,7 @@
         "defer_relation",
         "time_spine",
         "batch",
+        "freshness",
     }
 )
 

diff --git a/tests/unit/parser/test_parser.py b/tests/unit/parser/test_parser.py
@@ -8,7 +8,10 @@
 
 from dbt import tracking
 from dbt.artifacts.resources import ModelConfig, RefArgs
-from dbt.artifacts.resources.v1.model import ModelBuildAfter
+from dbt.artifacts.resources.v1.model import (
+    ModelBuildAfter,
+    ModelFreshnessDependsOnOptions,
+)
 from dbt.context.context_config import ContextConfig
 from dbt.contracts.files import FileHash, FilePath, SchemaSourceFile, SourceFile
 from dbt.contracts.graph.manifest import Manifest
@@ -728,7 +731,9 @@ def test__parse_model_freshness(self):
 
         assert self.parser.manifest.nodes[
             "model.root.my_model"
-        ].freshness.build_after == ModelBuildAfter(count=1, period="day", depends_on="any")
+        ].freshness.build_after == ModelBuildAfter(
+            count=1, period="day", depends_on=ModelFreshnessDependsOnOptions.any
+        )
 
     def test__parse_model_freshness_depend_on(self):
         block = self.file_block_for(SINGLE_TALBE_MODEL_FRESHNESS_ONLY_DEPEND_ON, "test_one.yml")
@@ -738,7 +743,9 @@ def test__parse_model_freshness_depend_on(self):
         self.assert_has_manifest_lengths(self.parser.manifest, nodes=1)
         assert self.parser.manifest.nodes[
             "model.root.my_model"
-        ].freshness.build_after == ModelBuildAfter(count=0, period="hour", depends_on="all")
+        ].freshness.build_after == ModelBuildAfter(
+            count=0, period="hour", depends_on=ModelFreshnessDependsOnOptions.all
+        )
 
     def test__read_basic_model_tests_wrong_severity(self):
         block = self.yaml_block_for(SINGLE_TABLE_MODEL_TESTS_WRONG_SEVERITY, "test_one.yml")