Fix tests for dbt-labs/dbt-core#3955 (#32)

* Fix tests for dbt-labs/dbt-core#3955 * Methods renamed in dbt-labs/dbt-core#4101 * Missed one
dbt-labs · Oct 27, 2021 · 2769ae5 · 2769ae5
1 parent 8a19724
commit 2769ae5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/tests/integration/docs_generate_tests/test_docs_generate.py b/tests/integration/docs_generate_tests/test_docs_generate.py
@@ -1096,7 +1096,7 @@ def expected_seeded_manifest(self, model_database=None, quote_model=False):
                     'database': self.default_database,
                     'description': 'My table',
                     'external': None,
-                    'freshness': {'error_after': None, 'warn_after': None, 'filter': None},
+                    'freshness': {'error_after': {'count': None, 'period': None}, 'warn_after': {'count': None, 'period': None}, 'filter': None},
                     'identifier': 'seed',
                     'loaded_at_field': None,
                     'loader': 'a_loader',

diff --git a/tests/unit/test_context.py b/tests/unit/test_context.py
@@ -403,7 +403,7 @@ def test_query_header_context(config, manifest_fx):
 
 
 def test_macro_runtime_context(config, manifest_fx, get_adapter, get_include_paths):
-    ctx = providers.generate_runtime_macro(
+    ctx = providers.generate_runtime_macro_context(
         macro=manifest_fx.macros['macro.root.macro_a'],
         config=config,
         manifest=manifest_fx,
@@ -413,7 +413,7 @@ def test_macro_runtime_context(config, manifest_fx, get_adapter, get_include_pat
 
 
 def test_model_parse_context(config, manifest_fx, get_adapter, get_include_paths):
-    ctx = providers.generate_parser_model(
+    ctx = providers.generate_parser_model_context(
         model=mock_model(),
         config=config,
         manifest=manifest_fx,
@@ -423,7 +423,7 @@ def test_model_parse_context(config, manifest_fx, get_adapter, get_include_paths
 
 
 def test_model_runtime_context(config, manifest_fx, get_adapter, get_include_paths):
-    ctx = providers.generate_runtime_model(
+    ctx = providers.generate_runtime_model_context(
         model=mock_model(),
         config=config,
         manifest=manifest_fx,
@@ -432,15 +432,15 @@ def test_model_runtime_context(config, manifest_fx, get_adapter, get_include_pat
 
 
 def test_docs_runtime_context(config):
-    ctx = docs.generate_runtime_docs(config, mock_model(), [], 'root')
+    ctx = docs.generate_runtime_docs_context(config, mock_model(), [], 'root')
     assert_has_keys(REQUIRED_DOCS_KEYS, MAYBE_KEYS, ctx)
 
 
 def test_resolve_specific(config, manifest_extended, redshift_adapter, get_include_paths):
     rs_macro = manifest_extended.macros['macro.dbt_redshift.redshift__some_macro']
     package_rs_macro = manifest_extended.macros['macro.root.redshift__some_macro']
 
-    ctx = providers.generate_runtime_model(
+    ctx = providers.generate_runtime_model_context(
         model=mock_model(),
         config=config,
         manifest=manifest_extended,