init

skills/pyspark-sql-guardrails/scripts/test_sql_guard.py

@@ -0,0 +1,64 @@
+"""Standard test set for sql_guard.assert_select_or_insert.
+
+Run from the scripts/ subdirectory of the skill:
+    cd .claude/skills/pyspark-sql-guardrails/scripts
+    python3 test_sql_guard.py
+
+Or from the project root:
+    python3 .claude/skills/pyspark-sql-guardrails/scripts/test_sql_guard.py
+
+Any change to assert_select_or_insert must be followed by running this
+set: all EXPECT_RAISE cases must raise, all EXPECT_OK cases must return
+cleanly. Failures are loud (non-zero exit).
+"""
+
+import sys
+import os
+
+# Allow running this file directly from the scripts/ subdirectory.
+sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
+
+from sql_guard import assert_select_or_insert
+
+
+def expect_ok(name, sql):
+    try:
+        assert_select_or_insert(sql)
+        print(f"  [OK]  {name}")
+    except ValueError as e:
+        raise AssertionError(f"{name} should have passed but raised: {e}")
+
+
+def expect_raise(name, sql):
+    try:
+        assert_select_or_insert(sql)
+        raise AssertionError(f"{name} should have raised but passed")
+    except ValueError:
+        print(f"  [OK]  {name}: raised as expected")
+
+
+print("=== EXPECT_OK ===")
+expect_ok("plain SELECT",        "SELECT 1")
+expect_ok("WITH ... SELECT",     "WITH t AS (SELECT 1 AS x) SELECT x FROM t")
+expect_ok("trailing semicolon",  "SELECT 1 FROM dual;")
+expect_ok("leading -- comment",  "-- comment\nSELECT 1")
+expect_ok("leading /* comment",  "/* hi */ SELECT 1")
+expect_ok("INSERT ... SELECT",   "INSERT INTO t SELECT 1")
+
+print()
+print("=== EXPECT_RAISE ===")
+expect_raise("DROP",              "DROP TABLE loan_order")
+expect_raise("UPDATE",            "UPDATE loan_order SET status = 1")
+expect_raise("DELETE",            "DELETE FROM loan_order")
+expect_raise("TRUNCATE",          "TRUNCATE TABLE loan_order")
+expect_raise("ALTER",             "ALTER TABLE loan_order ADD COLUMN x INT")
+expect_raise("MERGE",             "MERGE INTO t USING s ON t.id = s.id")
+expect_raise("MSCK",              "MSCK REPAIR TABLE loan_order")
+expect_raise("REFRESH",           "REFRESH TABLE loan_order")
+expect_raise("VACUUM",            "VACUUM TABLE loan_order")
+expect_raise("stacked ;",         "SELECT 1; DROP TABLE loan_order")
+expect_raise("empty",             "   ")
+expect_raise("comment-hidden",    "-- ok\nDROP TABLE loan_order")
+
+print()
+print("ALL TESTS PASSED")