don't fail for no owner

databricks · Mar 7, 2025 · 87dffb3 · 87dffb3
1 parent ab6065b
commit 87dffb3
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/tests/functional/adapter/basic/fixtures.py b/tests/functional/adapter/basic/fixtures.py
@@ -15,3 +15,10 @@
 union all
 select cast(2 as bigint) as id, 'goodbye' as msg
 """
+
+
+class AnyStringOrNone:
+    """Any string. Use this in assert calls"""
+
+    def __eq__(self, other):
+        return isinstance(other, str) or other is None
diff --git a/tests/functional/adapter/basic/test_docs_generate.py b/tests/functional/adapter/basic/test_docs_generate.py
@@ -2,7 +2,7 @@
 
 from dbt.tests.adapter.basic import expected_catalog
 from dbt.tests.adapter.basic.test_docs_generate import BaseDocsGenerate, BaseDocsGenReferences
-from dbt.tests.util import AnyString
+from tests.functional.adapter.basic.fixtures import AnyStringOrNone
 from tests.functional.adapter.basic.typing import AnyLongType, StatsLikeDict
 
 
@@ -11,7 +11,7 @@ class TestDocsGenerate(BaseDocsGenerate):
     def expected_catalog(self, project):
         return expected_catalog.base_expected_catalog(
             project,
-            role=AnyString(),
+            role=AnyStringOrNone(),
             id_type=AnyLongType(),
             text_type="string",
             time_type="timestamp",
@@ -26,7 +26,7 @@ class TestDocsGenReferences(BaseDocsGenReferences):
     def expected_catalog(self, project):
         return self.expected_references_catalog(
             project,
-            role=AnyString(),
+            role=AnyStringOrNone(),
             id_type=AnyLongType(),
             text_type="string",
             time_type="timestamp",