From 590c5ef0ffe80247b02a2653d189fa62a091ae28 Mon Sep 17 00:00:00 2001
From: Michelle Ark <michelle.ark@dbtlabs.com>
Date: Fri, 3 Mar 2023 10:47:23 -0500
Subject: [PATCH] first pass

---
 core/dbt/contracts/graph/nodes.py  | 43 ++++++++++++++++++++++++++++++
 core/dbt/graph/selector_methods.py |  1 +
 2 files changed, 44 insertions(+)

diff --git a/core/dbt/contracts/graph/nodes.py b/core/dbt/contracts/graph/nodes.py
index 368636c44d1..e5ef0f6a8f5 100644
--- a/core/dbt/contracts/graph/nodes.py
+++ b/core/dbt/contracts/graph/nodes.py
@@ -154,6 +154,21 @@ class ColumnInfo(AdditionalPropertiesMixin, ExtensibleDbtClassMixin, Replaceable
     tags: List[str] = field(default_factory=list)
     _extra: Dict[str, Any] = field(default_factory=dict)
 
+    def same_contract(self, old: "ColumnInfo") -> bool:
+        """contract includes name and data_type"""
+        breaking_changes = []
+        if self.name != old.name:
+            breaking_changes.append(f"Column {old.name} renamed to {self.name}.")
+        elif self.data_type != old.data_type:
+            breaking_changes.append(
+                f"Column {self.name} updated data_type from {old.data_type} to {self.data_type}"
+            )
+
+        if breaking_changes:
+            raise Exception(breaking_changes)
+
+        return True
+
 
 # Metrics, exposures,
 @dataclass
@@ -458,6 +473,34 @@ def __post_serialize__(self, dct):
                 del dct["compiled_code"]
         return dct
 
+    def same_contract(self, old) -> bool:
+        if old is None:
+            return False
+
+        breaking_changes = []
+        old_column_contract = [(column.name, column.data_type) for column in old.columns.values()]
+        new_column_contract = [(column.name, column.data_type) for column in self.columns.values()]
+
+        if old.contract and not self.contract:
+            breaking_changes.append("Updated config from contract: true to contract: false")
+
+        for column in old.columns:
+            if column not in self.columns.keys():
+                breaking_changes.append(f"Column {column} removed")
+            else:
+                try:
+                    self.columns[column].same_contract(old.columns[column])
+                except Exception as e:
+                    breaking_changes.append(str(e).strip("[']"))
+
+        if breaking_changes and (self.contract or old.contract):
+            breaking_changes_formatted = "\n * ".join(breaking_changes)
+            raise Exception(
+                f"breaking changes in {self.unique_id}!!\n * {breaking_changes_formatted}"
+            )
+
+        return old_column_contract == new_column_contract
+
     @property
     def depends_on_nodes(self):
         return self.depends_on.nodes
diff --git a/core/dbt/graph/selector_methods.py b/core/dbt/graph/selector_methods.py
index 1828f7f48d4..5eb7cca5b84 100644
--- a/core/dbt/graph/selector_methods.py
+++ b/core/dbt/graph/selector_methods.py
@@ -523,6 +523,7 @@ def search(self, included_nodes: Set[UniqueId], selector: str) -> Iterator[Uniqu
             "modified": self.check_modified_content,
             "modified.body": self.check_modified_factory("same_body"),
             "modified.configs": self.check_modified_factory("same_config"),
+            "modified.contract": self.check_modified_factory("same_contract"),
             "modified.persisted_descriptions": self.check_modified_factory(
                 "same_persisted_description"
             ),