From 9cb3e574931e16533d87e79b4f9b326f8ea7cf05 Mon Sep 17 00:00:00 2001 From: Chenyu Li Date: Wed, 12 Jun 2024 12:40:07 -0700 Subject: [PATCH] update on_skip to adjust for node that do not have schema --- core/dbt/task/base.py | 2 +- tests/unit/task/test_build.py | 14 ++++++++++++++ 2 files changed, 15 insertions(+), 1 deletion(-) create mode 100644 tests/unit/task/test_build.py diff --git a/core/dbt/task/base.py b/core/dbt/task/base.py index dfade18f1ab..df09a507e16 100644 --- a/core/dbt/task/base.py +++ b/core/dbt/task/base.py @@ -410,7 +410,7 @@ def _skip_caused_by_ephemeral_failure(self): return self.skip_cause.node.is_ephemeral_model def on_skip(self): - schema_name = self.node.schema + schema_name = getattr(self.node, "schema", "") node_name = self.node.name error_message = None diff --git a/tests/unit/task/test_build.py b/tests/unit/task/test_build.py new file mode 100644 index 00000000000..87d7b081ae8 --- /dev/null +++ b/tests/unit/task/test_build.py @@ -0,0 +1,14 @@ +from dbt.contracts.graph.nodes import SavedQuery +from dbt.task.build import SavedQueryRunner + + +def test_saved_query_runner_on_skip(saved_query: SavedQuery): + runner = SavedQueryRunner( + config=None, + adapter=None, + node=saved_query, + node_index=None, + num_nodes=None, + ) + # on_skip would work + runner.on_skip()