NOT IN query updated

Author: MohamedFaheemM

src/sql_data_guard/restriction_verification.py

@@ -110,14 +110,8 @@ def _verify_restriction(
     Returns:
         bool: True if the restriction is satisfied, False otherwise.
     """
-    if isinstance(exp, expr.Not) and isinstance(exp.this, expr.In):
-        expr_values = [int(val.this) for val in exp.this.expressions]  # Convert to int
-        restriction_values = [
-            int(val) for val in restriction["values"]
-        ]  # Convert to int
-
-        # NOT IN should be valid if it only excludes values in the restriction
-        return all(val in restriction_values for val in expr_values)
+    if isinstance(exp, expr.Not):
+        return False
 
     if isinstance(exp, expr.Paren):
         return _verify_restriction(restriction, from_table, exp.this)

test/test_sql_guard_updates_unit.py

@@ -728,8 +728,12 @@ def test_not_in_operator(self, config, cnn):
         verify_sql_test(
             """SELECT id FROM products1 WHERE id NOT IN (324, 224)""",
             config,
+            errors={
+                "Missing restriction for table: products1 column: id value: [324, 224]"
+            },
+            fix="SELECT id FROM products1 WHERE (NOT id IN (324, 224)) AND id IN (324, 224)",
             cnn=cnn,
-            data=[("325",), ("435",), ("445",)],
+            data=[],
         )
 
     def test_in_operator_with_numeric_values(self, config, cnn):