chore: adjust policy table reference test case (#228)

Author: tianzhou

testdata/diff/create_policy/add_policy/diff.sql

@@ -1,3 +1,5 @@
+ALTER TABLE orders ENABLE ROW LEVEL SECURITY;
+CREATE POLICY orders_user_access ON orders FOR SELECT TO PUBLIC USING (user_id IN ( SELECT users.id FROM users));
 CREATE POLICY "UserPolicy" ON users TO PUBLIC USING (tenant_id = current_setting('app.current_tenant')::integer);
 CREATE POLICY "my-policy" ON users FOR INSERT TO PUBLIC WITH CHECK ((role)::text = 'user');
 CREATE POLICY "select" ON users FOR SELECT TO PUBLIC USING (true);

testdata/diff/create_policy/add_policy/new.sql

@@ -5,9 +5,18 @@ CREATE TABLE users (
     role VARCHAR(50) NOT NULL
 );
 
+CREATE TABLE orders (
+    id SERIAL PRIMARY KEY,
+    user_id INTEGER REFERENCES users(id),
+    total NUMERIC(10,2)
+);
+
 -- RLS is enabled with multiple policies demonstrating quoting scenarios
 ALTER TABLE users ENABLE ROW LEVEL SECURITY;
 
+-- RLS on orders with policy referencing users table (Issue #224)
+ALTER TABLE orders ENABLE ROW LEVEL SECURITY;
+
 -- Policy with reserved word name (requires quoting)
 CREATE POLICY "select" ON users
     FOR SELECT
@@ -31,3 +40,10 @@ CREATE POLICY user_tenant_isolation ON users
     FOR UPDATE
     TO PUBLIC
     USING (tenant_id = current_setting('app.current_tenant')::INTEGER);
+
+-- Policy with subquery referencing another table (Issue #224)
+-- Tests that same-schema table qualifiers are normalized
+CREATE POLICY orders_user_access ON orders
+    FOR SELECT
+    TO PUBLIC
+    USING (user_id IN (SELECT id FROM users));

testdata/diff/create_policy/add_policy/old.sql

@@ -5,5 +5,11 @@ CREATE TABLE users (
     role VARCHAR(50) NOT NULL
 );
 
+CREATE TABLE orders (
+    id SERIAL PRIMARY KEY,
+    user_id INTEGER REFERENCES users(id),
+    total NUMERIC(10,2)
+);
+
 -- RLS is enabled but no policies exist yet
 ALTER TABLE users ENABLE ROW LEVEL SECURITY;

testdata/diff/create_policy/add_policy/plan.json

@@ -3,11 +3,23 @@
   "pgschema_version": "1.5.1",
   "created_at": "1970-01-01T00:00:00Z",
   "source_fingerprint": {
-    "hash": "1c45159a6e2607b1c83b933afa76aff63b63792704098a3046ec922e8f32649f"
+    "hash": "9323772d9678bd1630383ff088214914f1c01c427086930540c96be45e4be387"
   },
   "groups": [
     {
       "steps": [
+        {
+          "sql": "ALTER TABLE orders ENABLE ROW LEVEL SECURITY;",
+          "type": "table.rls",
+          "operation": "create",
+          "path": "public.orders"
+        },
+        {
+          "sql": "CREATE POLICY orders_user_access ON orders FOR SELECT TO PUBLIC USING (user_id IN ( SELECT users.id FROM users));",
+          "type": "table.policy",
+          "operation": "create",
+          "path": "public.orders.orders_user_access"
+        },
         {
           "sql": "CREATE POLICY \"UserPolicy\" ON users TO PUBLIC USING (tenant_id = current_setting('app.current_tenant')::integer);",
           "type": "table.policy",

testdata/diff/create_policy/add_policy/plan.sql

@@ -1,3 +1,7 @@
+ALTER TABLE orders ENABLE ROW LEVEL SECURITY;
+
+CREATE POLICY orders_user_access ON orders FOR SELECT TO PUBLIC USING (user_id IN ( SELECT users.id FROM users));
+
 CREATE POLICY "UserPolicy" ON users TO PUBLIC USING (tenant_id = current_setting('app.current_tenant')::integer);
 
 CREATE POLICY "my-policy" ON users FOR INSERT TO PUBLIC WITH CHECK ((role)::text = 'user');

testdata/diff/create_policy/add_policy/plan.txt

@@ -1,9 +1,12 @@
-Plan: 1 to modify.
+Plan: 2 to modify.
 
 Summary by type:
-  tables: 1 to modify
+  tables: 2 to modify
 
 Tables:
+  ~ orders
+    + orders_user_access (policy)
+    + orders (rls)
   ~ users
     + UserPolicy (policy)
     + my-policy (policy)
@@ -13,6 +16,10 @@ Tables:
 DDL to be executed:
 --------------------------------------------------
 
+ALTER TABLE orders ENABLE ROW LEVEL SECURITY;
+
+CREATE POLICY orders_user_access ON orders FOR SELECT TO PUBLIC USING (user_id IN ( SELECT users.id FROM users));
+
 CREATE POLICY "UserPolicy" ON users TO PUBLIC USING (tenant_id = current_setting('app.current_tenant')::integer);
 
 CREATE POLICY "my-policy" ON users FOR INSERT TO PUBLIC WITH CHECK ((role)::text = 'user');