From 31c75fc92b278ea1ff2c8d42ebd2b76bf1923379 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ensar=20Saraj=C4=8Di=C4=87?= <dev@ensarsarajcic.com>
Date: Thu, 19 Dec 2024 10:46:25 +0100
Subject: [PATCH] Ensure `ConfigDiff` takes tables into account

---
 src/config/diff.rs | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/config/diff.rs b/src/config/diff.rs
index da5ed54faeb9d..1a9097c71e59d 100644
--- a/src/config/diff.rs
+++ b/src/config/diff.rs
@@ -31,12 +31,16 @@ impl ConfigDiff {
         self.sources.flip();
         self.transforms.flip();
         self.sinks.flip();
+        self.enrichment_tables.flip();
         self
     }
 
     /// Checks whether or not the given component is present at all.
     pub fn contains(&self, key: &ComponentKey) -> bool {
-        self.sources.contains(key) || self.transforms.contains(key) || self.sinks.contains(key)
+        self.sources.contains(key)
+            || self.transforms.contains(key)
+            || self.sinks.contains(key)
+            || self.enrichment_tables.contains(key)
     }
 
     /// Checks whether or not the given component is changed.
@@ -44,6 +48,7 @@ impl ConfigDiff {
         self.sources.is_changed(key)
             || self.transforms.is_changed(key)
             || self.sinks.is_changed(key)
+            || self.enrichment_tables.contains(key)
     }
 
     /// Checks whether or not the given component is removed.
@@ -51,6 +56,7 @@ impl ConfigDiff {
         self.sources.is_removed(key)
             || self.transforms.is_removed(key)
             || self.sinks.is_removed(key)
+            || self.enrichment_tables.contains(key)
     }
 }