test/tabfunc01.test

   1 # 2015-08-19
   2 #
   3 # The author disclaims copyright to this source code.  In place of
   4 # a legal notice, here is a blessing:
   5 #
   6 #    May you do good and not evil.
   7 #    May you find forgiveness for yourself and forgive others.
   8 #    May you share freely, never taking more than you give.
   9 #
  10 #***********************************************************************
  11 #
  12 # This file implements tests for table-valued-functions implemented using
  13 # eponymous virtual tables.
  14 #
  15
  16 set testdir [file dirname $argv0]
  17 source $testdir/tester.tcl
  18 set testprefix tabfunc01
  19
  20 ifcapable !vtab {
  21   finish_test
  22   return
  23 }
  24 load_static_extension db series
  25 load_static_extension db carray
  26 load_static_extension db remember
  27
  28 do_execsql_test tabfunc01-1.1 {
  29   SELECT *, '|' FROM generate_series WHERE start=1 AND stop=9 AND step=2;
  30 } {1 | 3 | 5 | 7 | 9 |}
  31 do_execsql_test tabfunc01-1.2 {
  32   SELECT *, '|' FROM generate_series LIMIT 5;
  33 } {0 | 1 | 2 | 3 | 4 |}
  34 do_catchsql_test tabfunc01-1.3 {
  35   CREATE VIRTUAL TABLE t1 USING generate_series;
  36 } {1 {no such module: generate_series}}
  37 do_execsql_test tabfunc01-1.4 {
  38   SELECT * FROM generate_series(1,9,2);
  39 } {1 3 5 7 9}
  40 do_execsql_test tabfunc01-1.5 {
  41   SELECT * FROM generate_series(1,9);
  42 } {1 2 3 4 5 6 7 8 9}
  43 do_execsql_test tabfunc01-1.6 {
  44   SELECT * FROM generate_series(1,10) WHERE step=3;
  45 } {1 4 7 10}
  46 do_catchsql_test tabfunc01-1.7 {
  47   SELECT * FROM generate_series(1,9,2,11);
  48 } {1 {too many arguments on generate_series() - max 3}}
  49
  50 do_execsql_test tabfunc01-1.8 {
  51   SELECT * FROM generate_series(0,32,5) ORDER BY rowid DESC;
  52 } {30 25 20 15 10 5 0}
  53 do_execsql_test tabfunc01-1.9 {
  54   SELECT rowid, * FROM generate_series(0,32,5) ORDER BY value DESC;
  55 } {1 30 2 25 3 20 4 15 5 10 6 5 7 0}
  56 do_execsql_test tabfunc01-1.10 {
  57   SELECT rowid, * FROM generate_series(0,32,5) ORDER BY +value DESC;
  58 } {7 30 6 25 5 20 4 15 3 10 2 5 1 0}
  59
  60 do_execsql_test tabfunc01-1.20 {
  61   CREATE VIEW v1(a,b) AS VALUES(1,2),(3,4);
  62   SELECT * FROM v1;
  63 } {1 2 3 4}
  64 do_catchsql_test tabfunc01-1.21.1 {
  65   SELECT * FROM v1(55);
  66 } {1 {'v1' is not a function}}
  67 do_catchsql_test tabfunc01-1.21.2 {
  68   SELECT * FROM v1();
  69 } {1 {'v1' is not a function}}
  70 do_execsql_test tabfunc01-1.22 {
  71   CREATE VIEW v2(x) AS SELECT value FROM generate_series(1,5);
  72   SELECT * FROM v2;
  73 } {1 2 3 4 5}
  74 do_catchsql_test tabfunc01-1.23.1 {
  75   SELECT * FROM v2(55);
  76 } {1 {'v2' is not a function}}
  77 do_catchsql_test tabfunc01-1.23.2 {
  78   SELECT * FROM v2();
  79 } {1 {'v2' is not a function}}
  80 do_execsql_test tabfunc01-1.24 {
  81   CREATE TABLE t0(x);
  82   INSERT INTO t0(x) VALUES(123),(456),(789);
  83   SELECT * FROM t0 ORDER BY x;
  84 } {123 456 789}
  85 do_catchsql_test tabfunc01-1.25 {
  86   SELECT * FROM t0(55) ORDER BY x;
  87 } {1 {'t0' is not a function}}
  88 do_catchsql_test tabfunc01-1.26 {
  89   WITH w0 AS (SELECT * FROM t0)
  90   INSERT INTO t0(x) SELECT * FROM w0()
  91 } {1 {'w0' is not a function}}
  92
  93 do_execsql_test tabfunc01-2.1 {
  94   CREATE TABLE t1(x);
  95   INSERT INTO t1(x) VALUES(2),(3);
  96   SELECT *, '|' FROM t1, generate_series(1,x) ORDER BY 1, 2
  97 } {2 1 | 2 2 | 3 1 | 3 2 | 3 3 |}
  98 do_execsql_test tabfunc01-2.2 {
  99   SELECT *, '|' FROM (SELECT x FROM t1) AS y, generate_series(1,y.x)
 100   ORDER BY 1, 2;
 101 } {2 1 | 2 2 | 3 1 | 3 2 | 3 3 |}
 102
 103 do_execsql_test tabfunc01-2.50 {
 104   SELECT * FROM generate_series() LIMIT 5;
 105 } {0 1 2 3 4}
 106
 107 do_execsql_test tabfunc01-3.1 {
 108   SELECT DISTINCT value FROM generate_series(1,x), t1 ORDER BY 1;
 109 } {1 2 3}
 110
 111 # Eponymous virtual table exists in the "main" schema only
 112 #
 113 do_execsql_test tabfunc01-4.1 {
 114   SELECT * FROM main.generate_series(1,4)
 115 } {1 2 3 4}
 116 do_catchsql_test tabfunc01-4.2 {
 117   SELECT * FROM temp.generate_series(1,4)
 118 } {1 {no such table: temp.generate_series}}
 119 do_catchsql_test tabfunc01-4.3 {
 120   ATTACH ':memory:' AS aux1;
 121   CREATE TABLE aux1.t1(a,b,c);
 122   SELECT * FROM aux1.generate_series(1,4)
 123 } {1 {no such table: aux1.generate_series}}
 124
 125 # The next series of tests is verifying that virtual table are able
 126 # to optimize the IN operator, even on terms that are not marked "omit".
 127 # When the generate_series virtual table is compiled for the testfixture,
 128 # the special -DSQLITE_SERIES_CONSTRAINT_VERIFY=1 option is used, which
 129 # causes the xBestIndex method of generate_series to leave the
 130 # sqlite3_index_constraint_usage.omit flag set to 0, which should cause
 131 # the SQLite core to verify the start=, stop=, and step= constraints on
 132 # each step of output.  At one point, the IN operator could not be used
 133 # by virtual tables unless omit was set.
 134 #
 135 do_execsql_test tabfunc01-500 {
 136   SELECT * FROM generate_series WHERE start IN (1,7) AND stop=20 AND step=10
 137   ORDER BY +1;
 138 } {1 7 11 17}
 139
 140 # Table-valued functions on the RHS of an IN operator
 141 #
 142 do_execsql_test tabfunc01-600 {
 143   CREATE TABLE t600(a INTEGER PRIMARY KEY, b TEXT);
 144   WITH RECURSIVE c(x) AS (VALUES(1) UNION ALL SELECT x+1 FROM c WHERE x<100)
 145     INSERT INTO t600(a,b) SELECT x, printf('(%03d)',x) FROM c;
 146   SELECT b FROM t600 WHERE a IN generate_series(2,52,10);
 147 } {(002) (012) (022) (032) (042) (052)}
 148
 149
 150 do_test tabfunc01-700 {
 151   set PTR1 [intarray_addr 5 7 13 17 23]
 152   db eval {
 153     SELECT b FROM t600, carray(inttoptr($PTR1),5) WHERE a=value;
 154   }
 155 } {(005) (007) (013) (017) (023)}
 156 do_test tabfunc01-701 {
 157   db eval {
 158     SELECT b FROM t600 WHERE a IN carray(inttoptr($PTR1),5,'int32');
 159   }
 160 } {(005) (007) (013) (017) (023)}
 161 do_test tabfunc01-702 {
 162   db eval {
 163     SELECT b FROM t600 WHERE a IN carray(inttoptr($PTR1),4,'int32');
 164   }
 165 } {(005) (007) (013) (017)}
 166 do_catchsql_test tabfunc01-710 {
 167   SELECT b FROM t600 WHERE a IN carray(inttoptr($PTR1),5,'int33');
 168 } {1 {unknown datatype: 'int33'}}
 169
 170 do_test tabfunc01-720 {
 171   set PTR2 [int64array_addr 5 7 13 17 23]
 172   db eval {
 173     SELECT b FROM t600, carray(inttoptr($PTR2),5,'int64') WHERE a=value;
 174   }
 175 } {(005) (007) (013) (017) (023)}
 176 do_test tabfunc01-721 {
 177   db eval {
 178     SELECT remember(123,inttoptr($PTR2));
 179     SELECT value FROM carray(inttoptr($PTR2),5,'int64');
 180   }
 181 } {123 123 7 13 17 23}
 182 do_test tabfunc01-722 {
 183   set PTR3 [expr {$PTR2+16}]
 184   db eval {
 185     SELECT remember(987,inttoptr($PTR3));
 186     SELECT value FROM carray(inttoptr($PTR2),5,'int64');
 187   }
 188 } {987 123 7 987 17 23}
 189
 190 do_test tabfunc01-730 {
 191   set PTR4 [doublearray_addr 5.0 7.0 13.0 17.0 23.0]
 192   db eval {
 193     SELECT b FROM t600, carray(inttoptr($PTR4),5,'double') WHERE a=value;
 194   }
 195 } {(005) (007) (013) (017) (023)}
 196
 197 do_test tabfunc01-740 {
 198   set PTR5 [textarray_addr x5 x7 x13 x17 x23]
 199   db eval {
 200     SELECT b FROM t600, carray(inttoptr($PTR5),5,'char*')
 201      WHERE a=trim(value,'x');
 202   }
 203 } {(005) (007) (013) (017) (023)}
 204
 205 do_test tabfunc01-750 {
 206   db eval {
 207     SELECT aa.value, bb.value, '|'
 208       FROM carray(inttoptr($PTR4),5,'double') AS aa
 209       JOIN carray(inttoptr($PTR5),5,'char*') AS bb ON aa.rowid=bb.rowid;
 210   }
 211 } {5.0 x5 | 7.0 x7 | 13.0 x13 | 17.0 x17 | 23.0 x23 |}
 212
 213 # Free up memory allocations
 214 intarray_addr
 215 int64array_addr
 216 doublearray_addr
 217 textarray_addr
 218
 219 finish_test