CoolFace
Modelpublic

AryaWu/sqlite

sourceHugging Faceupdated 9mo agoView on Hugging Face
0likes
mallocK.test176 linesDownload Raw Back to test
1# 2008 August 012#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#12# This test script checks malloc failures in WHERE clause analysis.13# 14# $Id: mallocK.test,v 1.3 2009/01/08 21:00:03 drh Exp $15 16set testdir [file dirname $argv0]17source $testdir/tester.tcl18source $testdir/malloc_common.tcl19set testprefix mallocK20 21set sql {SELECT * FROM t1, t2 WHERE (a=1 OR a=2)}22for {set x 1} {$x<5} {incr x} {23  append sql " AND b=y"24  do_malloc_test mallocK-1.$x -sqlbody $sql -sqlprep {25    CREATE TABLE t1(a,b);26    CREATE TABLE t2(x,y);27  }28}29 30set sql {SELECT * FROM t1 WHERE a GLOB 'xyz*' AND (a=1 OR a=2)}31for {set x 1} {$x<5} {incr x} {32  append sql " AND b!=$x"33  do_malloc_test mallocK-2.$x -sqlbody $sql -sqlprep {34    CREATE TABLE t1(a,b);35  }36}37 38set sql {SELECT * FROM t1 WHERE a BETWEEN 5 AND 10}39for {set x 1} {$x<5} {incr x} {40  append sql " AND b=$x"41  do_malloc_test mallocK-3.$x -sqlbody $sql -sqlprep {42    CREATE TABLE t1(a,b);43  }44}45 46set sql {SELECT * FROM t1 WHERE b=0}47for {set x 1} {$x<5} {incr x} {48  set term "(b=$x"49  for {set y 0} {$y<$x} {incr y} {50    append term " AND a!=$y"51  }52  append sql " OR $term)"53  do_malloc_test mallocK-4.$x -sqlbody $sql -sqlprep {54    CREATE TABLE t1(a,b);55  }56}57 58ifcapable vtab {59  set sql {SELECT * FROM t2 WHERE a MATCH 'xyz'}60  for {set x 1} {$x<5} {incr x} {61    append sql " AND b!=$x"62    do_malloc_test mallocK-5.$x -sqlbody $sql -tclprep {63      register_echo_module [sqlite3_connection_pointer db]64      db eval {65        CREATE TABLE t1(a,b);66        CREATE VIRTUAL TABLE t2 USING echo(t1);67      }68    }69  }70}71 72#-------------------------------------------------------------------------73# Test that OOM errors are correctly handled by the code that uses stat474# data to estimate the number of rows visited by a skip-scan range query.75#76add_alignment_test_collations db77do_execsql_test 6.0 {78  CREATE TABLE t3(a TEXT, b TEXT COLLATE utf16_aligned, c);79  INSERT INTO t3 VALUES('one', '.....', 0);80  INSERT INTO t3 VALUES('one', '....x', 1);81  INSERT INTO t3 VALUES('one', '...x.', 2);82  INSERT INTO t3 VALUES('one', '...xx', 3);83  INSERT INTO t3 VALUES('one', '..x..', 4);84  INSERT INTO t3 VALUES('one', '..x.x', 5);85  INSERT INTO t3 VALUES('one', '..xx.', 6);86  INSERT INTO t3 VALUES('one', '..xxx', 7);87  INSERT INTO t3 VALUES('one', '.x...', 8);88  INSERT INTO t3 VALUES('one', '.x..x', 9);89  INSERT INTO t3 VALUES('one', '.x.x.', 10);90  INSERT INTO t3 VALUES('one', '.x.xx', 11);91  INSERT INTO t3 VALUES('one', '.xx..', 12);92  INSERT INTO t3 VALUES('one', '.xx.x', 13);93  INSERT INTO t3 VALUES('one', '.xxx.', 14);94  INSERT INTO t3 VALUES('one', '.xxxx', 15);95 96  INSERT INTO t3 VALUES('two', 'x....', 16);97  INSERT INTO t3 VALUES('two', 'x...x', 17);98  INSERT INTO t3 VALUES('two', 'x..x.', 18);99  INSERT INTO t3 VALUES('two', 'x..xx', 19);100  INSERT INTO t3 VALUES('two', 'x.x..', 20);101  INSERT INTO t3 VALUES('two', 'x.x.x', 21);102  INSERT INTO t3 VALUES('two', 'x.xx.', 22);103  INSERT INTO t3 VALUES('two', 'x.xxx', 23);104  INSERT INTO t3 VALUES('two', 'xx...', 24);105  INSERT INTO t3 VALUES('two', 'xx..x', 25);106  INSERT INTO t3 VALUES('two', 'xx.x.', 26);107  INSERT INTO t3 VALUES('two', 'xx.xx', 27);108  INSERT INTO t3 VALUES('two', 'xxx..', 28);109  INSERT INTO t3 VALUES('two', 'xxx.x', 29);110  INSERT INTO t3 VALUES('two', 'xxxx.', 30);111  INSERT INTO t3 VALUES('two', 'xxxxx', 31);112 113  INSERT INTO t3 SELECT * FROM t3;114 115  CREATE INDEX i3 ON t3(a, b);116  ANALYZE;117 118  SELECT 'x' > '.';119} {1}120 121ifcapable stat4 {122  do_eqp_test 6.1 {123    SELECT DISTINCT c FROM t3 WHERE b BETWEEN '.xx..' AND '.xxxx';124  } [string map {"\n  " \n} {125    QUERY PLAN126    |--SEARCH t3 USING INDEX i3 (ANY(a) AND b>? AND b<?)127    `--USE TEMP B-TREE FOR DISTINCT128  }]129}130 131do_faultsim_test 6 -faults oom* -body {132  db cache flush133  db eval { SELECT DISTINCT c FROM t3 WHERE b BETWEEN '.xx..' AND '.xxxx' }134} -test {135  faultsim_test_result {0 {12 13 14 15}} 136}137 138do_execsql_test 7.1 {139  CREATE TABLE x1(a INTEGER PRIMARY KEY, b);140}141do_faultsim_test 7.2 -faults oom* -body {142  execsql { SELECT * FROM x1 WHERE a = (SELECT 1) }143} -test {144  faultsim_test_result [list 0 {}]145}146 147reset_db148 149proc isqrt {i} { expr { int(sqrt($i)) } }150db func isqrt isqrt151 152do_execsql_test 8.0 {153  PRAGMA encoding = 'utf-16';154  CREATE TABLE x2(x TEXT, y TEXT);155  WITH data(i) AS (156    SELECT 1 UNION ALL SELECT i+1 FROM data157  )158  INSERT INTO x2 SELECT isqrt(i), isqrt(i) FROM data LIMIT 400;159  CREATE INDEX x2x ON x2(x);160  CREATE INDEX x2y ON x2(y);161  ANALYZE;162  DELETE FROM x2;163}164 165proc str {a} { return $a }166db func str -deterministic str167 168do_faultsim_test 8 -faults oom* -body {169  execsql { SELECT * FROM x2 WHERE x = str('19') AND y = str('4') }170} -test {171  faultsim_test_result [list 0 {}]172}173 174 175finish_test176