diff --git a/tests/unit/contracts/graph/test_manifest.py b/tests/unit/contracts/graph/test_manifest.py index 3505ee80037..526f46c7884 100644 --- 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 index 2921d19af0b..d6b2d53655f 100644 --- 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")