Add cross-db testcases for Table Valued function (#3802) (#3803)

Fixed behavior of cross-database table valued functions

Earlier, tabled valued function was not working for cross-db scenarios, with this commit we have fixed it's behaviour.

Engine PR : babelfish-for-postgresql/postgresql_modified_for_babelfish#583

Task: BABEL-5792

Signed-off-by: ANJU BHARTI <[email protected]>

Author: anju15bharti

test/JDBC/expected/BABEL-CROSS-DB-vu-cleanup.out

@@ -48,6 +48,9 @@ GO
 DROP FUNCTION babel_cross_db_vu_prepare_s1.babel_cross_db_vu_prepare_f1;
 GO
 
+DROP FUNCTION babel_cross_db_vu_prepare_s1.babel_cross_db_vu_prepare_tvf1;
+GO
+
 DROP PROCEDURE babel_cross_db_vu_prepare_s1.babel_cross_db_vu_prepare_p1_nested;
 GO
 
@@ -84,12 +87,18 @@ GO
 DROP FUNCTION dbo.babel_cross_db_vu_prepare_f2;
 GO
 
+DROP FUNCTION dbo.babel_cross_db_vu_prepare_tvf2;
+GO
+
 DROP VIEW babel_cross_db_vu_prepare_s2.babel_cross_db_vu_prepare_v2;
 GO
 
 DROP FUNCTION babel_cross_db_vu_prepare_s2.babel_cross_db_vu_prepare_f2;
 GO
 
+DROP FUNCTION babel_cross_db_vu_prepare_s2.babel_cross_db_vu_prepare_tvf2;
+GO
+
 DROP TABLE babel_cross_db_vu_prepare_s2.babel_cross_db_vu_prepare_t2;
 GO
 

test/JDBC/expected/BABEL-CROSS-DB-vu-prepare.out

@@ -102,6 +102,11 @@ CREATE FUNCTION dbo.babel_cross_db_vu_prepare_f2 (@a int)
     END;
 GO
 
+CREATE FUNCTION dbo.babel_cross_db_vu_prepare_tvf2()
+    RETURNS TABLE AS
+    RETURN select 'IT IS TVF' as col;
+GO
+
 -- tsql user=babel_cross_db_vu_prepare_l1 password=12345678
 USE my_babel_cross_db_vu_prepare_db1;
 GO
@@ -121,8 +126,14 @@ CREATE FUNCTION babel_cross_db_vu_prepare_s2.babel_cross_db_vu_prepare_f2 (@a in
     END;
 GO
 
+CREATE FUNCTION babel_cross_db_vu_prepare_s2.babel_cross_db_vu_prepare_tvf2()
+    RETURNS TABLE AS
+    RETURN select 'IT IS TVF' as col;
+GO
+
 CREATE VIEW babel_cross_db_vu_prepare_s2.babel_cross_db_vu_prepare_v2 AS
-    SELECT id, babel_cross_db_vu_prepare_s2.babel_cross_db_vu_prepare_f2(id), name
+    SELECT id, babel_cross_db_vu_prepare_s2.babel_cross_db_vu_prepare_f2(id),
+    babel_cross_db_vu_prepare_s2.babel_cross_db_vu_prepare_tvf2(), name
     FROM babel_cross_db_vu_prepare_s2.babel_cross_db_vu_prepare_t2
     ORDER BY id;
 GO
@@ -148,6 +159,7 @@ GO
 CREATE VIEW babel_cross_db_vu_prepare_s1.babel_cross_db_vu_prepare_v1 AS
     SELECT x.id,
         my_babel_cross_db_vu_prepare_db1.babel_cross_db_vu_prepare_s2.babel_cross_db_vu_prepare_f2(x.id),
+        my_babel_cross_db_vu_prepare_db1.babel_cross_db_vu_prepare_s2.babel_cross_db_vu_prepare_tvf2(),
         x.name AS t1_name,
         y.name AS t2_name
     FROM babel_cross_db_vu_prepare_s1.babel_cross_db_vu_prepare_t1 x INNER JOIN
@@ -165,7 +177,8 @@ GO
 
 -- View containing cross-db function on which login babel_cross_db_vu_prepare_l1 does not have privilege
 CREATE VIEW babel_cross_db_vu_prepare_s1.babel_cross_db_vu_prepare_v3 AS
-    SELECT my_babel_cross_db_vu_prepare_db1.dbo.babel_cross_db_vu_prepare_f2(id) AS id, name
+    SELECT my_babel_cross_db_vu_prepare_db1.dbo.babel_cross_db_vu_prepare_f2(id) AS id,
+    my_babel_cross_db_vu_prepare_db1.dbo.babel_cross_db_vu_prepare_tvf2(), name
     FROM babel_cross_db_vu_prepare_s1.babel_cross_db_vu_prepare_t1
     ORDER BY id;
 GO
@@ -200,6 +213,11 @@ CREATE FUNCTION babel_cross_db_vu_prepare_s1.babel_cross_db_vu_prepare_f1 (@a in
     END;
 GO
 
+CREATE FUNCTION babel_cross_db_vu_prepare_s1.babel_cross_db_vu_prepare_tvf1()
+    RETURNS TABLE AS
+    RETURN SELECT my_babel_cross_db_vu_prepare_db1.babel_cross_db_vu_prepare_s2.babel_cross_db_vu_prepare_tvf2();
+GO
+
 CREATE VIEW babel_cross_db_vu_prepare_s1.babel_cross_db_vu_prepare_v1_nested AS
     SELECT * FROM babel_cross_db_vu_prepare_s1.babel_cross_db_vu_prepare_v1;
 GO
@@ -217,7 +235,8 @@ CREATE VIEW babel_cross_db_vu_prepare_s1.babel_cross_db_vu_prepare_v4_nested AS
 GO
 
 CREATE VIEW babel_cross_db_vu_prepare_s1.babel_cross_db_vu_prepare_v5_nested AS
-    SELECT babel_cross_db_vu_prepare_s1.babel_cross_db_vu_prepare_f1(1) AS id;
+    SELECT babel_cross_db_vu_prepare_s1.babel_cross_db_vu_prepare_f1(1) AS id,
+    babel_cross_db_vu_prepare_s1.babel_cross_db_vu_prepare_tvf1();
 GO
 
 CREATE PROCEDURE babel_cross_db_vu_prepare_s1.babel_cross_db_vu_prepare_p1_nested

test/JDBC/expected/BABEL-CROSS-DB-vu-verify.out

@@ -436,8 +436,8 @@ int#!#varchar
 SELECT * FROM my_babel_cross_db_vu_prepare_db1.babel_cross_db_vu_prepare_s2.babel_cross_db_vu_prepare_v2;
 GO
 ~~START~~
-int#!#int#!#varchar
-1#!#1#!#abc
+int#!#int#!#varchar#!#varchar
+1#!#1#!#IT IS TVF#!#abc
 ~~END~~
 
 
@@ -450,6 +450,14 @@ int
 ~~END~~
 
 
+SELECT my_babel_cross_db_vu_prepare_db1.babel_cross_db_vu_prepare_s2.babel_cross_db_vu_prepare_tvf2();
+GO
+~~START~~
+varchar
+IT IS TVF
+~~END~~
+
+
 -- bare SELECT table from db1, which login babel_cross_db_vu_prepare_l1 does not have privilege, should throw error
 SELECT * FROM my_babel_cross_db_vu_prepare_db1.dbo.babel_cross_db_vu_prepare_t2 ORDER BY id;
 GO
@@ -466,28 +474,35 @@ GO
 ~~ERROR (Message: permission denied for function babel_cross_db_vu_prepare_f2)~~
 
 
+SELECT my_babel_cross_db_vu_prepare_db1.dbo.babel_cross_db_vu_prepare_tvf2();
+GO
+~~ERROR (Code: 33557097)~~
+
+~~ERROR (Message: permission denied for function babel_cross_db_vu_prepare_tvf2)~~
+
+
 -- SELECT from a view in db1 which references objects in db1, which login babel_cross_db_vu_prepare_l1 has privilege, should be successful
 SELECT * FROM babel_cross_db_vu_prepare_s1.babel_cross_db_vu_prepare_v1;
 GO
 ~~START~~
-int#!#int#!#varchar#!#varchar
-1#!#1#!#def#!#abc
+int#!#int#!#varchar#!#varchar#!#varchar
+1#!#1#!#IT IS TVF#!#def#!#abc
 ~~END~~
 
 
 SELECT * FROM babel_cross_db_vu_prepare_s1.babel_cross_db_vu_prepare_v1_nested;
 GO
 ~~START~~
-int#!#int#!#varchar#!#varchar
-1#!#1#!#def#!#abc
+int#!#int#!#varchar#!#varchar#!#varchar
+1#!#1#!#IT IS TVF#!#def#!#abc
 ~~END~~
 
 
 SELECT * FROM babel_cross_db_vu_prepare_s1.babel_cross_db_vu_prepare_v5_nested;
 GO
 ~~START~~
-int
-1
+int#!#varchar
+1#!#IT IS TVF
 ~~END~~
 
 
@@ -545,8 +560,8 @@ int#!#varchar
 ~~END~~
 
 ~~START~~
-int#!#int#!#varchar
-1#!#1#!#abc
+int#!#int#!#varchar#!#varchar
+1#!#1#!#IT IS TVF#!#abc
 ~~END~~
 
 
@@ -564,8 +579,8 @@ int#!#varchar
 ~~END~~
 
 ~~START~~
-int#!#int#!#varchar#!#varchar
-1#!#1#!#def#!#abc
+int#!#int#!#varchar#!#varchar#!#varchar
+1#!#1#!#IT IS TVF#!#def#!#abc
 ~~END~~
 
 ~~ERROR (Code: 33557097)~~
@@ -586,8 +601,8 @@ int#!#varchar
 ~~END~~
 
 ~~START~~
-int#!#int#!#varchar#!#varchar
-1#!#1#!#def#!#abc
+int#!#int#!#varchar#!#varchar#!#varchar
+1#!#1#!#IT IS TVF#!#def#!#abc
 ~~END~~
 
 ~~ERROR (Code: 33557097)~~
@@ -612,8 +627,8 @@ int#!#varchar
 ~~END~~
 
 ~~START~~
-int#!#int#!#varchar#!#varchar
-1#!#1#!#def#!#abc
+int#!#int#!#varchar#!#varchar#!#varchar
+1#!#1#!#IT IS TVF#!#def#!#abc
 ~~END~~
 
 ~~ERROR (Code: 33557097)~~
@@ -638,8 +653,8 @@ int#!#varchar
 ~~END~~
 
 ~~START~~
-int#!#int#!#varchar#!#varchar
-1#!#1#!#def#!#abc
+int#!#int#!#varchar#!#varchar#!#varchar
+1#!#1#!#IT IS TVF#!#def#!#abc
 ~~END~~
 
 ~~ERROR (Code: 33557097)~~