Fix the renamed relations code (#1125)

* Add test and move semantics. * Add field import --------- Co-authored-by: Mila Page <[email protected]> Co-authored-by: Mike Alfare <[email protected]> Co-authored-by: Mike Alfare <[email protected]> (cherry picked from commit 4002483)
dbt-labs · Mar 21, 2024 · c668ca5 · c668ca5
1 parent 56949d8
commit c668ca5
Show file tree

Hide file tree

Showing 3 changed files with 39 additions and 4 deletions.
diff --git a/.changes/unreleased/Under the Hood-20240227-004659.yaml b/.changes/unreleased/Under the Hood-20240227-004659.yaml
@@ -0,0 +1,6 @@
+kind: Under the Hood
+body: Add unit test for transaction semantics.
+time: 2024-02-27T00:46:59.188231-08:00
+custom:
+  Author: versusfacit
+  Issue: "1123"
diff --git a/dbt/adapters/bigquery/relation.py b/dbt/adapters/bigquery/relation.py
@@ -1,4 +1,4 @@
-from dataclasses import dataclass
+from dataclasses import dataclass, field
 from typing import FrozenSet, Optional, TypeVar
 
 from itertools import chain, islice
@@ -25,9 +25,22 @@
 class BigQueryRelation(BaseRelation):
     quote_character: str = "`"
     location: Optional[str] = None
-    renameable_relations: FrozenSet[RelationType] = frozenset({RelationType.Table})
-    replaceable_relations: FrozenSet[RelationType] = frozenset(
-        {RelationType.Table, RelationType.View}
+
+    renameable_relations: FrozenSet[RelationType] = field(
+        default_factory=lambda: frozenset(
+            {
+                RelationType.Table,
+            }
+        )
+    )
+
+    replaceable_relations: FrozenSet[RelationType] = field(
+        default_factory=lambda: frozenset(
+            {
+                RelationType.View,
+                RelationType.Table,
+            }
+        )
     )
 
     def matches(

diff --git a/tests/unit/test_renamed_relations.py b/tests/unit/test_renamed_relations.py
@@ -0,0 +1,16 @@
+from dbt.adapters.bigquery.relation import BigQueryRelation
+from dbt.adapters.contracts.relation import RelationType
+
+
+def test_renameable_relation():
+    relation = BigQueryRelation.create(
+        database="my_db",
+        schema="my_schema",
+        identifier="my_table",
+        type=RelationType.Table,
+    )
+    assert relation.renameable_relations == frozenset(
+        {
+            RelationType.Table,
+        }
+    )