Added extra objects in prepare to have consistent plans between CI and local

Signed-off-by: Rucha Kulkarni <ruchask@amazon.com>

Author: Rucha Kulkarni

test/JDBC/expected/BABEL-6814-vu-cleanup.out

@@ -1,6 +1,28 @@
 DROP FUNCTION babel_6814_func1;
 GO
 
+-- Drop extra tables created for consistent EXPLAIN plan behavior
+DECLARE @i INT = 1;
+DECLARE @sql NVARCHAR(200);
+WHILE @i <= 100
+BEGIN
+    SET @sql = 'DROP TABLE babel_6814_extra_' + CONVERT(VARCHAR, @i);
+    EXEC(@sql);
+    SET @i = @i + 1;
+END
+GO
+
+-- Drop extra schemas
+DECLARE @j INT = 1;
+DECLARE @sql2 NVARCHAR(200);
+WHILE @j <= 500
+BEGIN
+    SET @sql2 = 'DROP SCHEMA babel_6814_schema_' + CONVERT(VARCHAR, @j);
+    EXEC(@sql2);
+    SET @j = @j + 1;
+END
+GO
+
 DROP VIEW babel_6814_view1;
 GO
 

test/JDBC/expected/BABEL-6814-vu-prepare.out

@@ -26,3 +26,28 @@ BEGIN
 RETURN 1;
 END
 GO
+
+-- Create extra tables and indexes to ensure pg_index/pg_class have enough rows
+-- for consistent planner behavior in EXPLAIN tests
+DECLARE @i INT = 1;
+DECLARE @sql NVARCHAR(200);
+WHILE @i <= 100
+BEGIN
+    SET @sql = 'CREATE TABLE babel_6814_extra_' + CONVERT(VARCHAR, @i) + ' (id INT)';
+    EXEC(@sql);
+    SET @sql = 'CREATE INDEX babel_6814_extra_idx_' + CONVERT(VARCHAR, @i) + ' ON babel_6814_extra_' + CONVERT(VARCHAR, @i) + '(id)';
+    EXEC(@sql);
+    SET @i = @i + 1;
+END
+GO
+
+-- Create extra schemas to ensure pg_namespace has enough rows for consistent plans
+DECLARE @j INT = 1;
+DECLARE @sql2 NVARCHAR(200);
+WHILE @j <= 500
+BEGIN
+    SET @sql2 = 'CREATE SCHEMA babel_6814_schema_' + CONVERT(VARCHAR, @j);
+    EXEC(@sql2);
+    SET @j = @j + 1;
+END
+GO

test/JDBC/expected/BABEL-6814-vu-verify.out

@@ -147,7 +147,7 @@ babel_6814_t2#!#USER_TABLE
 
 -- Test 13: Different object types all resolve correctly
 SELECT name, type, type_desc FROM sys.objects
-WHERE name LIKE 'babel_6814%'
+WHERE name LIKE 'babel_6814%' AND name NOT LIKE 'babel_6814_extra%'
 ORDER BY name;
 GO
 ~~START~~
@@ -225,11 +225,12 @@ EXPLAIN (COSTS OFF) SELECT c.relname, n.nspname FROM pg_class c JOIN pg_namespac
 GO
 ~~START~~
 text
-Nested Loop
-  Join Filter: ((c.relnamespace)::integer = (n.oid)::integer)
-  ->  Index Only Scan using pg_class_relname_nsp_index on pg_class c
-        Index Cond: (relname = 'babel_6814_t1'::name)
+Hash Join
+  Hash Cond: ((n.oid)::integer = (c.relnamespace)::integer)
   ->  Seq Scan on pg_namespace n
+  ->  Hash
+        ->  Index Only Scan using pg_class_relname_nsp_index on pg_class c
+              Index Cond: (relname = 'babel_6814_t1'::name)
 ~~END~~
 
 
@@ -241,8 +242,8 @@ text
 Index Only Scan using pg_class_relname_nsp_index on pg_class c
   Index Cond: (relname = 'babel_6814_t1'::name)
   SubPlan 1
-    ->  Seq Scan on pg_namespace n
-          Filter: ((oid)::integer = (c.relnamespace)::integer)
+    ->  Index Scan using pg_namespace_oid_index on pg_namespace n
+          Index Cond: (oid = ((c.relnamespace)::integer)::oid)
 ~~END~~
 
 
@@ -251,8 +252,8 @@ EXPLAIN (COSTS OFF) SELECT * FROM pg_index WHERE (indrelid)::int4 = 12345;
 GO
 ~~START~~
 text
-Seq Scan on pg_index
-  Filter: ((indrelid)::integer = 12345)
+Index Scan using pg_index_indrelid_index on pg_index
+  Index Cond: (indrelid = (12345)::oid)
 ~~END~~
 
 

test/JDBC/expected/parallel_query/BABEL-6814-vu-verify.out

@@ -147,7 +147,7 @@ babel_6814_t2#!#USER_TABLE
 
 -- Test 13: Different object types all resolve correctly
 SELECT name, type, type_desc FROM sys.objects
-WHERE name LIKE 'babel_6814%'
+WHERE name LIKE 'babel_6814%' AND name NOT LIKE 'babel_6814_extra%'
 ORDER BY name;
 GO
 ~~START~~
@@ -232,7 +232,7 @@ GO
 ~~START~~
 text
 Gather
-  Workers Planned: 1
+  Workers Planned: 3
   ->  Parallel Hash Join
         Hash Cond: ((n.oid)::integer = (c.relnamespace)::integer)
         ->  Parallel Seq Scan on pg_namespace n
@@ -250,8 +250,8 @@ text
 Index Only Scan using pg_class_relname_nsp_index on pg_class c
   Index Cond: (relname = 'babel_6814_t1'::name)
   SubPlan 1
-    ->  Seq Scan on pg_namespace n
-          Filter: ((oid)::integer = (c.relnamespace)::integer)
+    ->  Index Scan using pg_namespace_oid_index on pg_namespace n
+          Index Cond: (oid = ((c.relnamespace)::integer)::oid)
 ~~END~~
 
 
@@ -261,9 +261,10 @@ GO
 ~~START~~
 text
 Gather
-  Workers Planned: 2
-  ->  Parallel Seq Scan on pg_index
-        Filter: ((indrelid)::integer = 12345)
+  Workers Planned: 1
+  Single Copy: true
+  ->  Index Scan using pg_index_indrelid_index on pg_index
+        Index Cond: (indrelid = (12345)::oid)
 ~~END~~
 
 

test/JDBC/input/BABEL-6814-vu-cleanup.sql

@@ -1,6 +1,28 @@
 DROP FUNCTION babel_6814_func1;
 GO
 
+-- Drop extra tables created for consistent EXPLAIN plan behavior
+DECLARE @i INT = 1;
+DECLARE @sql NVARCHAR(200);
+WHILE @i <= 100
+BEGIN
+    SET @sql = 'DROP TABLE babel_6814_extra_' + CONVERT(VARCHAR, @i);
+    EXEC(@sql);
+    SET @i = @i + 1;
+END
+GO
+
+-- Drop extra schemas
+DECLARE @j INT = 1;
+DECLARE @sql2 NVARCHAR(200);
+WHILE @j <= 500
+BEGIN
+    SET @sql2 = 'DROP SCHEMA babel_6814_schema_' + CONVERT(VARCHAR, @j);
+    EXEC(@sql2);
+    SET @j = @j + 1;
+END
+GO
+
 DROP VIEW babel_6814_view1;
 GO
 

test/JDBC/input/BABEL-6814-vu-prepare.sql

@@ -26,3 +26,28 @@ BEGIN
 RETURN 1;
 END
 GO
+
+-- Create extra tables and indexes to ensure pg_index/pg_class have enough rows
+-- for consistent planner behavior in EXPLAIN tests
+DECLARE @i INT = 1;
+DECLARE @sql NVARCHAR(200);
+WHILE @i <= 100
+BEGIN
+    SET @sql = 'CREATE TABLE babel_6814_extra_' + CONVERT(VARCHAR, @i) + ' (id INT)';
+    EXEC(@sql);
+    SET @sql = 'CREATE INDEX babel_6814_extra_idx_' + CONVERT(VARCHAR, @i) + ' ON babel_6814_extra_' + CONVERT(VARCHAR, @i) + '(id)';
+    EXEC(@sql);
+    SET @i = @i + 1;
+END
+GO
+
+-- Create extra schemas to ensure pg_namespace has enough rows for consistent plans
+DECLARE @j INT = 1;
+DECLARE @sql2 NVARCHAR(200);
+WHILE @j <= 500
+BEGIN
+    SET @sql2 = 'CREATE SCHEMA babel_6814_schema_' + CONVERT(VARCHAR, @j);
+    EXEC(@sql2);
+    SET @j = @j + 1;
+END
+GO

test/JDBC/input/BABEL-6814-vu-verify.mix

@@ -78,7 +78,7 @@ GO
 
 -- Test 13: Different object types all resolve correctly
 SELECT name, type, type_desc FROM sys.objects
-WHERE name LIKE 'babel_6814%'
+WHERE name LIKE 'babel_6814%' AND name NOT LIKE 'babel_6814_extra%'
 ORDER BY name;
 GO