test inlined SQL functions

funbringer · funbringer · commit 010bf7ed0bb3 · 2016-11-21T18:31:18.000+03:00
diff --git a/expected/pathman_basic.out b/expected/pathman_basic.out
@@ -1080,6 +1080,40 @@ WHERE j1.dt < '2015-03-01' AND j2.dt >= '2015-02-01' ORDER BY j2.dt;
                                  ->  Index Scan using range_rel_2_pkey on range_rel_2 j1_1
 (20 rows)
 
+/*
+ * Test inlined SQL functions
+ */
+CREATE TABLE test.sql_inline (id INT NOT NULL);
+SELECT pathman.create_hash_partitions('test.sql_inline', 'id', 3);
+ create_hash_partitions 
+------------------------
+                      3
+(1 row)
+
+CREATE OR REPLACE FUNCTION test.sql_inline_func(i_id int) RETURNS SETOF INT AS $$
+	select * from test.sql_inline where id = i_id limit 1;
+$$ LANGUAGE sql STABLE;
+EXPLAIN (COSTS OFF) SELECT * FROM test.sql_inline_func(5);
+              QUERY PLAN              
+--------------------------------------
+ Limit
+   ->  Append
+         ->  Seq Scan on sql_inline_0
+               Filter: (id = 5)
+(4 rows)
+
+EXPLAIN (COSTS OFF) SELECT * FROM test.sql_inline_func(1);
+              QUERY PLAN              
+--------------------------------------
+ Limit
+   ->  Append
+         ->  Seq Scan on sql_inline_2
+               Filter: (id = 1)
+(4 rows)
+
+DROP FUNCTION test.sql_inline_func(int);
+DROP TABLE test.sql_inline CASCADE;
+NOTICE:  drop cascades to 3 other objects
 /*
  * Test CTE query
  */
diff --git a/sql/pathman_basic.sql b/sql/pathman_basic.sql
@@ -264,6 +264,22 @@ JOIN test.range_rel j2 on j2.id = j1.id
 JOIN test.num_range_rel j3 on j3.id = j1.id
 WHERE j1.dt < '2015-03-01' AND j2.dt >= '2015-02-01' ORDER BY j2.dt;
 
+/*
+ * Test inlined SQL functions
+ */
+CREATE TABLE test.sql_inline (id INT NOT NULL);
+SELECT pathman.create_hash_partitions('test.sql_inline', 'id', 3);
+
+CREATE OR REPLACE FUNCTION test.sql_inline_func(i_id int) RETURNS SETOF INT AS $$
+	select * from test.sql_inline where id = i_id limit 1;
+$$ LANGUAGE sql STABLE;
+
+EXPLAIN (COSTS OFF) SELECT * FROM test.sql_inline_func(5);
+EXPLAIN (COSTS OFF) SELECT * FROM test.sql_inline_func(1);
+
+DROP FUNCTION test.sql_inline_func(int);
+DROP TABLE test.sql_inline CASCADE;
+
 /*
  * Test CTE query
  */
