Home | History | Annotate | Download | only in test
      1 # 2011 February 21
      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 # This file implements regression tests for SQLite library.
     12 #
     13 # This file implements tests to verify that ticket [b72787b1a7] has been
     14 # fixed.  From the ticket:
     15 #
     16 #     The sqlite3ExpirePreparedStatements routine marks all statements
     17 #     as expired. This includes statements that are not expired.
     18 #
     19 #     Steps to reproduce:
     20 #
     21 #        * Prepare a statement (A)
     22 #        * Alter the schema to invalidate cookie in A
     23 #        * Prepare a statement (B)
     24 #        * Run B and have A run as part of B
     25 #        * A will find a bad cookie and cause *all* statements
     26 #          to be expired including the currently running B by calling
     27 #          sqlite3ExpirePreparedStatements
     28 #        * When control returns to B it will then abort 
     29 #
     30 #     The bug is that sqlite3ExpirePreparedStatements expires all statements.
     31 #     Note that B was prepared after the schema change and hence is perfectly
     32 #     valid and then is marked as expired while running.
     33 #
     34 
     35 set testdir [file dirname $argv0]
     36 source $testdir/tester.tcl
     37 
     38 unset -nocomplain ::STMT
     39 proc runsql {} {
     40   db eval {CREATE TABLE IF NOT EXISTS t4(q)}
     41   sqlite3_step $::STMT
     42   set rc [sqlite3_column_int $::STMT 0]
     43   sqlite3_reset $::STMT
     44   return $rc
     45 }
     46 
     47 do_test tkt-b72787b1.1 {
     48   db eval {
     49     CREATE TABLE t1(x);
     50     INSERT INTO t1 VALUES(1);
     51     INSERT INTO t1 VALUES(2);
     52     CREATE TABLE t2(y);
     53     INSERT INTO t2 SELECT x+2 FROM t1;
     54     INSERT INTO t2 SELECT x+4 FROM t1;
     55   }
     56   db func runsql ::runsql
     57   set DB [sqlite3_connection_pointer db]
     58   set sql {SELECT max(x) FROM t1}
     59   set ::STMT [sqlite3_prepare_v2 $DB $sql -1 TAIL]
     60 
     61   # The runsql() call on the second row of the first query will
     62   # cause all $::STMT to hit an expired cookie.  Prior to the fix
     63   # for [b72787b1a7, the bad cookie would expire all statements, including
     64   # the following compound SELECT, which would cause a fault when the
     65   # second SELECT was reached.  After the fix, the current statement
     66   # continues to completion.
     67   db eval {
     68     SELECT CASE WHEN y=3 THEN y+100 WHEN y==4 THEN runsql()+200
     69                 ELSE 300+y END FROM t2
     70     UNION ALL
     71     SELECT * FROM t1;
     72   }
     73 } {103 202 305 306 1 2}
     74 
     75 sqlite3_finalize $::STMT
     76 
     77 finish_test
     78