diff --git a/tests/unit/fixtures.py b/tests/unit/fixtures.py index 5cd18f49bc0..cfcc1636b18 100644 --- a/tests/unit/fixtures.py +++ b/tests/unit/fixtures.py @@ -26,6 +26,7 @@ def model_node(): metrics=[], depends_on=DependsOn(), description="", + primary_key=[], database="test_db", schema="test_schema", alias="bar", diff --git a/tests/unit/test_contracts_graph_compiled.py b/tests/unit/test_contracts_graph_compiled.py index e345899e4d6..e0228cc28f6 100644 --- a/tests/unit/test_contracts_graph_compiled.py +++ b/tests/unit/test_contracts_graph_compiled.py @@ -149,6 +149,7 @@ def basic_compiled_dict(): "depends_on": {"macros": [], "nodes": []}, "database": "test_db", "description": "", + "primary_key": [], "schema": "test_schema", "alias": "bar", "tags": [], diff --git a/tests/unit/test_contracts_graph_parsed.py b/tests/unit/test_contracts_graph_parsed.py index 3a1861af462..7a62c394b22 100644 --- a/tests/unit/test_contracts_graph_parsed.py +++ b/tests/unit/test_contracts_graph_parsed.py @@ -167,6 +167,7 @@ def base_parsed_model_dict(): "depends_on": {"macros": [], "nodes": []}, "database": "test_db", "description": "", + "primary_key": [], "schema": "test_schema", "alias": "bar", "tags": [], @@ -220,6 +221,7 @@ def basic_parsed_model_object(): metrics=[], depends_on=DependsOn(), description="", + primary_key=[], database="test_db", schema="test_schema", alias="bar", @@ -274,6 +276,7 @@ def complex_parsed_model_dict(): "depends_on": {"macros": [], "nodes": ["model.test.bar"]}, "database": "test_db", "description": "My parsed node", + "primary_key": [], "schema": "test_schema", "alias": "bar", "tags": ["tag"],