vendor/sqlite3/sqlite-3370100

1 files changed, 255 insertions, 0 deletions

diff --git a/test/with6.test b/test/with6.test
new file mode 100644
index 000000000000..333147d8be69
--- /dev/null
+++ b/test/with6.test
@@ -0,0 +1,255 @@
+# 2021-02-22
+#
+# The author disclaims copyright to this source code.  In place of
+# a legal notice, here is a blessing:
+#
+#    May you do good and not evil.
+#    May you find forgiveness for yourself and forgive others.
+#    May you share freely, never taking more than you give.
+#
+#***********************************************************************
+# This file implements regression tests for SQLite library.  The
+# focus of this file is the MATERIALIZED hint to common table expressions
+#
+
+set testdir [file dirname $argv0]
+source $testdir/tester.tcl
+set ::testprefix with6
+
+ifcapable {!cte} {
+  finish_test
+  return
+}
+
+do_execsql_test 100 {
+  WITH c(x) AS (VALUES(0),(1))
+  SELECT c1.x||c2.x||c3.x FROM c c1, c c2, c c3;
+} {000 001 010 011 100 101 110 111}
+do_eqp_test 101 {
+  WITH c(x) AS (VALUES(0),(1))
+  SELECT c1.x||c2.x||c3.x FROM c c1, c c2, c c3;
+} {
+  QUERY PLAN
+  |--MATERIALIZE c
+  |  `--SCAN 2 CONSTANT ROWS
+  |--SCAN c1
+  |--SCAN c2
+  `--SCAN c3
+}
+
+do_execsql_test 110 {
+  WITH c(x) AS MATERIALIZED (VALUES(0),(1))
+  SELECT c1.x||c2.x||c3.x FROM c c1, c c2, c c3;
+} {000 001 010 011 100 101 110 111}
+do_eqp_test 111 {
+  WITH c(x) AS MATERIALIZED (VALUES(0),(1))
+  SELECT c1.x||c2.x||c3.x FROM c c1, c c2, c c3;
+} {
+  QUERY PLAN
+  |--MATERIALIZE c
+  |  `--SCAN 2 CONSTANT ROWS
+  |--SCAN c1
+  |--SCAN c2
+  `--SCAN c3
+}
+
+# Even though the CTE is not materialized, the self-join optimization
+# kicks in and does the materialization for us.
+#
+do_execsql_test 120 {
+  WITH c(x) AS NOT MATERIALIZED (VALUES(0),(1))
+  SELECT c1.x||c2.x||c3.x FROM c c1, c c2, c c3;
+} {000 001 010 011 100 101 110 111}
+do_eqp_test 121 {
+  WITH c(x) AS NOT MATERIALIZED (VALUES(0),(1))
+  SELECT c1.x||c2.x||c3.x FROM c c1, c c2, c c3;
+} {
+  QUERY PLAN
+  |--MATERIALIZE c
+  |  `--SCAN 2 CONSTANT ROWS
+  |--SCAN c1
+  |--SCAN c2
+  `--SCAN c3
+}
+
+do_execsql_test 130 {
+  WITH c(x) AS NOT MATERIALIZED (VALUES(0),(1))
+  SELECT c1.x||c2.x||c3.x
+    FROM (SELECT x FROM c LIMIT 5) AS c1,
+         (SELECT x FROM c LIMIT 5) AS c2,
+         (SELECT x FROM c LIMIT 5) AS c3;
+} {000 001 010 011 100 101 110 111}
+do_eqp_test 131 {
+  WITH c(x) AS NOT MATERIALIZED (VALUES(0),(1))
+  SELECT c1.x||c2.x||c3.x
+    FROM (SELECT x FROM c LIMIT 5) AS c1,
+         (SELECT x FROM c LIMIT 5) AS c2,
+         (SELECT x FROM c LIMIT 5) AS c3;
+} {
+  QUERY PLAN
+  |--MATERIALIZE c1
+  |  |--CO-ROUTINE c
+  |  |  `--SCAN 2 CONSTANT ROWS
+  |  `--SCAN c
+  |--MATERIALIZE c2
+  |  |--CO-ROUTINE c
+  |  |  `--SCAN 2 CONSTANT ROWS
+  |  `--SCAN c
+  |--MATERIALIZE c3
+  |  |--CO-ROUTINE c
+  |  |  `--SCAN 2 CONSTANT ROWS
+  |  `--SCAN c
+  |--SCAN c1
+  |--SCAN c2
+  `--SCAN c3
+}
+
+# The (SELECT x FROM c LIMIT N) subqueries get materialized once each.
+# Show multiple materializations are shown.  But there is only one
+# materialization for c, shown by the "SCAN 2 CONSTANT ROWS" line.
+#
+do_execsql_test 140 {
+  WITH c(x) AS MATERIALIZED (VALUES(0),(1))
+  SELECT c1.x||c2.x||c3.x
+    FROM (SELECT x FROM c LIMIT 5) AS c1,
+         (SELECT x FROM c LIMIT 6) AS c2,
+         (SELECT x FROM c LIMIT 7) AS c3;
+} {000 001 010 011 100 101 110 111}
+do_eqp_test 141 {
+  WITH c(x) AS MATERIALIZED (VALUES(0),(1))
+  SELECT c1.x||c2.x||c3.x
+    FROM (SELECT x FROM c LIMIT 5) AS c1,
+         (SELECT x FROM c LIMIT 6) AS c2,
+         (SELECT x FROM c LIMIT 7) AS c3;
+} {
+  QUERY PLAN
+  |--MATERIALIZE c1
+  |  |--MATERIALIZE c
+  |  |  `--SCAN 2 CONSTANT ROWS
+  |  `--SCAN c
+  |--MATERIALIZE c2
+  |  `--SCAN c
+  |--MATERIALIZE c3
+  |  `--SCAN c
+  |--SCAN c1
+  |--SCAN c2
+  `--SCAN c3
+}
+
+do_execsql_test 150 {
+  WITH c(x) AS (VALUES(0),(1))
+  SELECT c1.x||c2.x||c3.x
+    FROM (SELECT x FROM c LIMIT 5) AS c1,
+         (SELECT x FROM c LIMIT 6) AS c2,
+         (SELECT x FROM c LIMIT 7) AS c3;
+} {000 001 010 011 100 101 110 111}
+do_eqp_test 151 {
+  WITH c(x) AS (VALUES(0),(1))
+  SELECT c1.x||c2.x||c3.x
+    FROM (SELECT x FROM c LIMIT 5) AS c1,
+         (SELECT x FROM c LIMIT 6) AS c2,
+         (SELECT x FROM c LIMIT 7) AS c3;
+} {
+  QUERY PLAN
+  |--MATERIALIZE c1
+  |  |--MATERIALIZE c
+  |  |  `--SCAN 2 CONSTANT ROWS
+  |  `--SCAN c
+  |--MATERIALIZE c2
+  |  `--SCAN c
+  |--MATERIALIZE c3
+  |  `--SCAN c
+  |--SCAN c1
+  |--SCAN c2
+  `--SCAN c3
+}
+
+do_execsql_test 160 {
+  WITH c(x) AS (VALUES(0),(1))
+  SELECT c2.x + 100*(SELECT sum(x+1) FROM c WHERE c.x<=c2.x)
+    FROM c AS c2 WHERE c2.x<10;
+} {100 301}
+do_eqp_test 161 {
+  WITH c(x) AS (VALUES(0),(1))
+  SELECT c2.x + 100*(SELECT sum(x+1) FROM c WHERE c.x<=c2.x)
+    FROM c AS c2 WHERE c2.x<10;
+} {
+  QUERY PLAN
+  |--MATERIALIZE c
+  |  `--SCAN 2 CONSTANT ROWS
+  |--SCAN c2
+  `--CORRELATED SCALAR SUBQUERY xxxxxx
+     `--SCAN c
+}
+
+do_execsql_test 170 {
+  WITH c(x) AS NOT MATERIALIZED (VALUES(0),(1))
+  SELECT c2.x + 100*(SELECT sum(x+1) FROM c WHERE c.x<=c2.x)
+    FROM c AS c2 WHERE c2.x<10;
+} {100 301}
+do_eqp_test 171 {
+  WITH c(x) AS NOT MATERIALIZED (VALUES(0),(1))
+  SELECT c2.x + 100*(SELECT sum(x+1) FROM c WHERE c.x<=c2.x)
+    FROM c AS c2 WHERE c2.x<10;
+} {
+  QUERY PLAN
+  |--CO-ROUTINE c
+  |  `--SCAN 2 CONSTANT ROWS
+  |--SCAN c2
+  `--CORRELATED SCALAR SUBQUERY xxxxxx
+     |--CO-ROUTINE c
+     |  `--SCAN 2 CONSTANT ROWS
+     `--SCAN c
+}
+
+
+do_execsql_test 200 {
+  CREATE TABLE t1(x);
+  INSERT INTO t1(x) VALUES(4);
+  CREATE VIEW t2(y) AS
+    WITH c(z) AS (VALUES(4),(5),(6))
+    SELECT c1.z+c2.z*100+t1.x*10000
+      FROM t1,
+           (SELECT z FROM c LIMIT 5) AS c1,
+           (SELECT z FROM c LIMIT 5) AS c2;
+  SELECT y FROM t2 ORDER BY y;
+} {40404 40405 40406 40504 40505 40506 40604 40605 40606}
+do_execsql_test 210 {
+  DROP VIEW t2;
+  CREATE VIEW t2(y) AS
+    WITH c(z) AS NOT MATERIALIZED (VALUES(4),(5),(6))
+    SELECT c1.z+c2.z*100+t1.x*10000
+      FROM t1,
+           (SELECT z FROM c LIMIT 5) AS c1,
+           (SELECT z FROM c LIMIT 5) AS c2;
+  SELECT y FROM t2 ORDER BY y;
+} {40404 40405 40406 40504 40505 40506 40604 40605 40606}
+do_eqp_test 211 {
+  SELECT y FROM t2 ORDER BY y;
+} {
+  QUERY PLAN
+  |--MATERIALIZE c1
+  |  |--MATERIALIZE c
+  |  |  `--SCAN 3 CONSTANT ROWS
+  |  `--SCAN c
+  |--MATERIALIZE c2
+  |  `--SCAN c
+  |--SCAN c1
+  |--SCAN c2
+  |--SCAN t1
+  `--USE TEMP B-TREE FOR ORDER BY
+}
+do_execsql_test 220 {
+  DROP VIEW t2;
+  CREATE VIEW t2(y) AS
+    WITH c(z) AS MATERIALIZED (VALUES(4),(5),(6))
+    SELECT c1.z+c2.z*100+t1.x*10000
+      FROM t1,
+           (SELECT z FROM c LIMIT 5) AS c1,
+           (SELECT z FROM c LIMIT 5) AS c2;
+  SELECT y FROM t2 ORDER BY y;
+} {40404 40405 40406 40504 40505 40506 40604 40605 40606}
+
+
+
+finish_test