CoolFace
Modelpublic

AryaWu/sqlite

sourceHugging Faceupdated 9mo agoView on Hugging Face
0likes
indexexpr2.test429 linesDownload Raw Back to test
1# 2017 April 112#3# The author disclaims copyright to this source code.  In place of4# 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# This file implements regression tests for SQLite library.12#13 14set testdir [file dirname $argv0]15source $testdir/tester.tcl16set testprefix indexexpr217 18do_execsql_test 1 {19  CREATE TABLE t1(a, b);20  INSERT INTO t1 VALUES(1, 'one');21  INSERT INTO t1 VALUES(2, 'two');22  INSERT INTO t1 VALUES(3, 'three');23 24  CREATE INDEX i1 ON t1(b || 'x');25}26 27do_execsql_test 1.1 {28  SELECT 'TWOX' == (b || 'x') FROM t1 WHERE (b || 'x')>'onex'29} {0 0}30 31do_execsql_test 1.2 {32  SELECT 'TWOX' == (b || 'x') COLLATE nocase  FROM t1 WHERE (b || 'x')>'onex'33} {0 1}34 35do_execsql_test 2.0 {36  CREATE INDEX i2 ON t1(a+1);37}38 39do_execsql_test 2.1 {40  SELECT a+1, quote(a+1) FROM t1 ORDER BY 1;41} {2 2 3 3 4 4}42 43#-------------------------------------------------------------------------44# At one point SQLite was incorrectly using indexes on expressions to45# optimize ORDER BY and GROUP BY clauses even when the collation46# sequences of the query and index did not match (ticket [e20dd54ab0e4]).47# The following tests - 3.* - attempt to verify that this has been fixed.48#49 50reset_db51do_execsql_test 3.1.0 {52  CREATE TABLE t1(a, b);53  CREATE INDEX i1 ON t1(a, b);54} {}55 56do_eqp_test 3.1.1 {57  SELECT b FROM t1 WHERE b IS NOT NULL AND a IS NULL 58  GROUP BY b COLLATE nocase59  ORDER BY b COLLATE nocase;60} {/USE TEMP B-TREE FOR GROUP BY/}61 62do_execsql_test 3.2.0 {63  CREATE TABLE t2(x);64 65  INSERT INTO t2 VALUES('.ABC');66  INSERT INTO t2 VALUES('.abcd');67  INSERT INTO t2 VALUES('.defg');68  INSERT INTO t2 VALUES('.DEF');69} {}70 71do_execsql_test 3.2.1 {72  SELECT x FROM t2 ORDER BY substr(x, 2) COLLATE nocase;73} {74  .ABC .abcd .DEF .defg75}76 77do_execsql_test 3.2.2 {78  CREATE INDEX i2 ON t2( substr(x, 2) );79  SELECT x FROM t2 ORDER BY substr(x, 2) COLLATE nocase;80} {81  .ABC .abcd .DEF .defg82}83 84do_execsql_test 3.3.0 {85  CREATE TABLE t3(x);86}87 88ifcapable json1 {89  do_eqp_test 3.3.1 {90    SELECT json_extract(x, '$.b') FROM t2 91    WHERE json_extract(x, '$.b') IS NOT NULL AND json_extract(x, '$.a') IS NULL 92    GROUP BY json_extract(x, '$.b') COLLATE nocase93    ORDER BY json_extract(x, '$.b') COLLATE nocase;94  } [string map {"\n  " \n} {95    QUERY PLAN96    |--SCAN t297    `--USE TEMP B-TREE FOR GROUP BY98  }]99  100  do_execsql_test 3.3.2 {101    CREATE INDEX i3 ON t3(json_extract(x, '$.a'), json_extract(x, '$.b'));102  } {}103  104  do_eqp_test 3.3.3 {105    SELECT json_extract(x, '$.b') FROM t3 106    WHERE json_extract(x, '$.b') IS NOT NULL AND json_extract(x, '$.a') IS NULL 107    GROUP BY json_extract(x, '$.b') COLLATE nocase108    ORDER BY json_extract(x, '$.b') COLLATE nocase;109  } [string map {"\n  " \n} {110    QUERY PLAN111    |--SEARCH t3 USING INDEX i3 (<expr>=?)112    `--USE TEMP B-TREE FOR GROUP BY113  }]114}115 116do_execsql_test 3.4.0 {117  CREATE TABLE t4(a, b);118  INSERT INTO t4 VALUES('.ABC', 1);119  INSERT INTO t4 VALUES('.abc', 2);120  INSERT INTO t4 VALUES('.ABC', 3);121  INSERT INTO t4 VALUES('.abc', 4);122}123 124do_execsql_test 3.4.1 {125  SELECT * FROM t4 126  WHERE substr(a, 2) = 'abc' COLLATE NOCASE127  ORDER BY substr(a, 2), b;128} {129  .ABC 1   .ABC 3   .abc 2   .abc 4130}131 132do_execsql_test 3.4.2 {133  CREATE INDEX i4 ON t4( substr(a, 2) COLLATE NOCASE, b );134  SELECT * FROM t4 135  WHERE substr(a, 2) = 'abc' COLLATE NOCASE136  ORDER BY substr(a, 2), b;137} {138  .ABC 1   .ABC 3   .abc 2   .abc 4139}140 141do_execsql_test 3.4.3 {142  DROP INDEX i4;143  UPDATE t4 SET a = printf('%s%d',a,b);144  SELECT * FROM t4 ORDER BY Substr(a,-2) COLLATE nocase;145} {.ABC1 1 .abc2 2 .ABC3 3 .abc4 4}146do_execsql_test 3.4.4 {147  SELECT * FROM t4 ORDER BY Substr(a,-2) COLLATE binary;148} {.ABC1 1 .ABC3 3 .abc2 2 .abc4 4}149 150do_execsql_test 3.4.5 {151  CREATE INDEX i4 ON t4( Substr(a,-2) COLLATE nocase );152  SELECT * FROM t4 ORDER BY Substr(a,-2) COLLATE nocase;153} {.ABC1 1 .abc2 2 .ABC3 3 .abc4 4}154do_execsql_test 3.4.5eqp {155  EXPLAIN QUERY PLAN156  SELECT * FROM t4 ORDER BY Substr(a,-2) COLLATE nocase;157} {/SCAN t4 USING INDEX i4/}158do_execsql_test 3.4.6 {159  SELECT * FROM t4 ORDER BY Substr(a,-2) COLLATE binary;160} {.ABC1 1 .ABC3 3 .abc2 2 .abc4 4}161 162# 2014-09-15:  Verify that UPDATEs of columns not referenced by a163# index on expression do not modify the index.164#165unset -nocomplain cnt166set cnt 0167proc refcnt {x} {168  global cnt169  incr cnt170  return $x171}172db close173sqlite3 db :memory:174db function refcnt -deterministic refcnt175do_test 4.100 {176  db eval {177    CREATE TABLE t1(a,b,c,d,e,f);178    CREATE INDEX t1abc ON t1(refcnt(a+b+c));179  }180  set ::cnt181} {0}182do_test 4.110 {183  db eval {INSERT INTO t1 VALUES(1,2,3,4,5,6);}184  set ::cnt185  # The refcnt() function is invoked once to compute the index value 186} {1}187do_test 4.120 {188  set ::cnt 0189  db eval {UPDATE t1 SET b=b+1;}190  set ::cnt191  # The refcnt() function is invoked twice, once to remove the old index192  # entry and a second time to insert the new one.193} {2}194do_test 4.130 {195  set ::cnt 0196  db eval {UPDATE t1 SET d=d+1;}197  set ::cnt198  # Refcnt() should not be invoked because that index does not change.199} {0}200 201# Additional test cases to show that UPDATE does not modify indexes that202# do not involve unchanged columns.203#204ifcapable vtab {205  load_static_extension db explain206  do_execsql_test 4.200 {207    CREATE TABLE t2(a,b,c,d,e,f);208    INSERT INTO t2 VALUES(2,3,4,5,6,7);209    CREATE INDEX t2abc ON t2(a+b+c);210    CREATE INDEX t2cd ON t2(c*d);211    CREATE INDEX t2def ON t2(d,e+25*f);212    SELECT sqlite_master.name 213      FROM sqlite_master, explain('UPDATE t2 SET b=b+1')214     WHERE explain.opcode LIKE 'Open%'215       AND sqlite_master.rootpage=explain.p2216     ORDER BY 1;217  } {t2 t2abc}218  do_execsql_test 4.210 {219    SELECT sqlite_master.name 220      FROM sqlite_master, explain('UPDATE t2 SET c=c+1')221     WHERE explain.opcode LIKE 'Open%'222       AND sqlite_master.rootpage=explain.p2223     ORDER BY 1;224  } {t2 t2abc t2cd}225  do_execsql_test 4.220 {226    SELECT sqlite_master.name 227      FROM sqlite_master, explain('UPDATE t2 SET c=c+1, f=NULL')228     WHERE explain.opcode LIKE 'Open%'229       AND sqlite_master.rootpage=explain.p2230     ORDER BY 1;231  } {t2 t2abc t2cd t2def}232}233 234#-------------------------------------------------------------------------235# Test that ticket [d96eba87] has been fixed.236#237do_execsql_test 5.0 {238  CREATE TABLE t5(a INTEGER, b INTEGER);239  INSERT INTO t5 VALUES(2, 4), (3, 9);240}241do_execsql_test 5.1 {242  SELECT * FROM t5 WHERE abs(a)=2 or abs(b)=9;243} {2 4 3 9}244do_execsql_test 5.2 {245  CREATE INDEX t5a ON t5( abs(a) );246  CREATE INDEX t5b ON t5( abs(b) );247}248do_execsql_test 5.4 {249  SELECT * FROM t5 WHERE abs(a)=2 or abs(b)=9;250} {2 4 3 9}251 252#-------------------------------------------------------------------------253do_execsql_test 6.0 {254  CREATE TABLE x1(a INTEGER PRIMARY KEY, b);255  INSERT INTO x1 VALUES256      (1, 123), (2, '123'), (3, '123abc'), (4, 123.0), (5, 1234);257}258 259do_execsql_test 6.1.1 {260  SELECT a, b FROM x1 WHERE CAST(b AS INTEGER) = 123;261} {1 123   2 123   3 123abc  4 123.0}262do_execsql_test 6.1.2 {263  CREATE INDEX x1i ON x1( CAST(b AS INTEGER) );264  SELECT a, b FROM x1 WHERE CAST(b AS INTEGER) = 123;265} {1 123   2 123   3 123abc  4 123.0}266do_eqp_test 6.1.3 {267  SELECT a, b FROM x1 WHERE CAST(b AS INTEGER) = 123;268} {SEARCH x1 USING INDEX x1i (<expr>=?)}269 270do_execsql_test 6.2.1 {271  SELECT a, b FROM x1 WHERE CAST(b AS TEXT) = 123;272} {1 123   2 123}273do_execsql_test 6.2.2 {274  CREATE INDEX x1i2 ON x1( CAST(b AS TEXT) );275  SELECT a, b FROM x1 WHERE CAST(b AS TEXT) = 123;276} {1 123   2 123}277do_eqp_test 6.2.3 {278  SELECT a, b FROM x1 WHERE CAST(b AS TEXT) = 123;279} {SEARCH x1 USING INDEX x1i2 (<expr>=?)}280 281do_execsql_test 7.0 {282  CREATE TABLE IF NOT EXISTS t0(c0);283  INSERT INTO t0(c0) VALUES (-9223372036854775808);284  BEGIN;285}286do_catchsql_test 7.1 {287  CREATE INDEX i0 ON t0(ABS(c0));288} {1 {integer overflow}}289do_execsql_test 7.2 {290  COMMIT;291  SELECT sql FROM sqlite_master WHERE tbl_name = 't0';292  CREATE INDEX i0 ON t0(c0);293} {{CREATE TABLE t0(c0)}}294do_execsql_test 7.3 {295  REINDEX;296} {}297 298#-------------------------------------------------------------------------299reset_db300do_execsql_test 8.0 {301  CREATE TABLE t0(c0);302  CREATE INDEX i0 ON t0(c0) WHERE c0 NOT NULL;303  INSERT INTO t0(c0) VALUES (NULL);304}305 306do_execsql_test 8.1.1 {307  SELECT * FROM t0 WHERE ~('' BETWEEN t0.c0 AND TRUE);308} {{}}309do_execsql_test 8.1.2 {310  SELECT ~('' BETWEEN t0.c0 AND TRUE) FROM t0;311} {-1}312 313foreach {tn expr} {314  1 " 0  ==  (34 BETWEEN c0 AND 33)"315  2 " 1  !=  (34 BETWEEN c0 AND 33)"316  3 "-1   <  (34 BETWEEN c0 AND 33)"317  4 "-1  <=  (34 BETWEEN c0 AND 33)"318  5 " 1   >  (34 BETWEEN c0 AND 33)"319  6 " 1  >=  (34 BETWEEN c0 AND 33)"320  7 " 1   -  (34 BETWEEN c0 AND 33)"321  8 "-1   +  (34 BETWEEN c0 AND 33)"322  9 " 1   |  (34 BETWEEN c0 AND 33)"323 10 " 1  <<  (34 BETWEEN c0 AND 33)"324 11 " 1  >>  (34 BETWEEN c0 AND 33)"325 12 " 1  ||  (34 BETWEEN c0 AND 33)"326} {327  do_execsql_test 8.3.$tn.1 "SELECT * FROM t0 WHERE $expr ORDER BY c0" { {} }328  do_execsql_test 8.3.$tn.2 "SELECT ($expr) IS TRUE FROM t0"           { 1 }329}330 331do_execsql_test 8.4 {332  CREATE TABLE t1(a, b);333  INSERT INTO t1 VALUES(1, 2), (3, 4);334  CREATE TABLE t2(x, y);335}336 337foreach {tn expr} {338  1 " 0  ==  (a=0 AND y=1)"339  2 " 1  !=  (a=0 AND y=1)"340  3 "-1  <   (a=0 AND y=1)"341  4 "-1  <=  (a=0 AND y=1)"342  5 " 1   >  (a=0 AND y=1)"343  6 " 1  >=  (a=0 AND y=1)"344  7 " 1   -  (a=0 AND y=1)"345  8 "-1   +  (a=0 AND y=1)"346  9 " 1   |  (a=0 AND y=1)"347  10 "1  <<  (a=0 AND y=1)"348  11 "1  >>  (a=0 AND y=1)"349  12 "1  ||  (a=0 AND y=1)"350 351  13 " 0  ==  (10 BETWEEN y AND b)"352  14 " 1  !=  (10 BETWEEN y AND b)"353  15 "-1  <   (10 BETWEEN y AND b)"354  16 "-1  <=  (10 BETWEEN y AND b)"355  17 " 1   >  (10 BETWEEN y AND b)"356  18 " 1  >=  (10 BETWEEN y AND b)"357  19 " 1   -  (10 BETWEEN y AND b)"358  20 "-1   +  (10 BETWEEN y AND b)"359  21 " 1   |  (10 BETWEEN y AND b)"360  22 " 1  <<  (10 BETWEEN y AND b)"361  23 " 1  >>  (10 BETWEEN y AND b)"362  24 " 1  ||  (10 BETWEEN y AND b)"363 364  25 " 1  ||  (10 BETWEEN y AND b)"365} {366  do_execsql_test 8.5.$tn.1 "367    SELECT * FROM t1 LEFT JOIN t2 WHERE $expr368  " {1 2 {} {} 3 4 {} {}}369 370  do_execsql_test 8.5.$tn.2 "371    SELECT ($expr) IS TRUE FROM t1 LEFT JOIN t2372  " {1 1}373}374 375# 2023-03-24 https://sqlite.org/forum/forumpost/79cf371080376#377reset_db378do_execsql_test 9.0 {379  CREATE TABLE t1(a INT, b INT);380  CREATE INDEX t1x ON t1(a, abs(b));381  CREATE TABLE t2(c INT, d INT);382  INSERT INTO t1(a,b) VALUES(4,4),(5,-5),(5,20),(6,6);383  INSERT INTO t2(c,d) VALUES(100,1),(200,1),(300,2);384  SELECT *,385    (SELECT max(c+abs(b)) FROM t2 GROUP BY d ORDER BY d LIMIT 1) AS subq386   FROM t1 WHERE a=5;387} {5 -5 205 5 20 220}388 389# 2023-04-03 https://sqlite.org/forum/forumpost/44270909bb390# and https://sqlite.org/forum/forumpost/e45108732c which are the391# same problem, namely the failure to omit the EP_Collate property392# from an expression node when changing it from TK_COLLATE into393# TK_AGG_COLUMN because it resolves to an indexed expression.394#395reset_db396do_execsql_test 10.0 {397  CREATE TABLE t1(a INTEGER PRIMARY KEY, b TEXT);398  CREATE INDEX t1x ON t1 (b, +b COLLATE NOCASE);399  INSERT INTO t1(a,b) VALUES(1,'abcde');400  SELECT * FROM t1 AS a0401   WHERE (SELECT count(a0.b=+a0.b COLLATE NOCASE IN (b)) FROM t1 GROUP BY 2.5)402   ORDER BY a0.b;403} {1 abcde}404do_execsql_test 10.1 {405  CREATE TABLE t2(a TEXT);406  INSERT INTO t2 VALUES('alice'),('bob'),('cindy'),('david');407  CREATE INDEX t2x ON t2 (+a COLLATE NOCASE);408  SELECT count(+a COLLATE NOCASE IN (SELECT 1)) AS x409    FROM t2410   GROUP BY SUBSTR(0,0);411} 4412 413# 2023-04-03 https://sqlite.org/forum/forumpost/409ebc7368414# When a generated column appears in both an outer and an inner loop415# (that is to say, the same table is used in both loops) and the416# generated column is indexed and it is used inside an aggregate function,417# make sure that the terms resolve to the correct aggregate.418#419do_execsql_test 11.0 {420  CREATE TABLE t3 (a INT, b AS (-a));421  CREATE INDEX t3x ON t3(b, a);422  INSERT INTO t3(a) VALUES(44);423  SELECT * FROM t3 AS a0424   WHERE (SELECT sum(-a0.a=b) FROM t3 GROUP BY b)425   GROUP BY b;426} {44 -44}427 428finish_test429