Add test to restore transactions behavior. (#24)

Co-authored-by: Mila Page <[email protected]> Co-authored-by: Mike Alfare <[email protected]> Co-authored-by: Mike Alfare <[email protected]>
dbt-labs · Mar 21, 2024 · 23880f4 · 23880f4
1 parent 618c85d
commit 23880f4
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 5 deletions.
diff --git a/.changes/unreleased/Under the Hood-20240226-225642.yaml b/.changes/unreleased/Under the Hood-20240226-225642.yaml
@@ -0,0 +1,6 @@
+kind: Under the Hood
+body: Add unit test for transaction semantics.
+time: 2024-02-26T22:56:42.202429-08:00
+custom:
+  Author: versusfacit
+  Issue: "23"
diff --git a/dbt/adapters/postgres/relation.py b/dbt/adapters/postgres/relation.py
@@ -1,4 +1,4 @@
-from dataclasses import dataclass
+from dataclasses import dataclass, field
 from typing import FrozenSet, Optional, Set
 
 from dbt.adapters.base.relation import BaseRelation
@@ -20,19 +20,19 @@
 
 @dataclass(frozen=True, eq=False, repr=False)
 class PostgresRelation(BaseRelation):
-    renameable_relations = frozenset(
+    renameable_relations: FrozenSet[RelationType] = field(default_factory=lambda: frozenset(
         {
             RelationType.View,
             RelationType.Table,
             RelationType.MaterializedView,
         }
-    )
-    replaceable_relations = frozenset(
+    ))
+    replaceable_relations: FrozenSet[RelationType] = field(default_factory=lambda: frozenset(
         {
             RelationType.View,
             RelationType.Table,
         }
-    )
+    ))
 
     def __post_init__(self):
         # Check for length of Postgres table/view names.

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