# 2007 Aug 13

 #

 # 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 tests aspects of the malloc failure while parsing

 # CREATE TABLE statements in auto_vacuum mode.

 #

 # $Id: mallocC.test,v 1.9 2008/02/18 22:24:58 drh Exp $

 set testdir [file dirname $argv0]

 source $testdir/tester.tcl

 source $testdir/malloc_common.tcl

 # Only run these tests if memory debugging is turned on.

 #

 if {!$MEMDEBUG} {

    puts "Skipping mallocC tests: not compiled with -DSQLITE_MEMDEBUG..."

    finish_test

    return

 }

 proc do_mallocC_test {tn args} {

   array set ::mallocopts $args

   #set sum [allcksum db]

   for {set ::n 1} {true} {incr ::n} {

     # Run the SQL. Malloc number $::n is set to fail. A malloc() failure

     # may or may not be reported.

     sqlite3_memdebug_fail $::n -repeat 1

     do_test mallocC-$tn.$::n.1 {

       set res [catchsql [string trim $::mallocopts(-sql)]]

       set rc [expr { 

         0==[string compare $res {1 {out of memory}}] ||

         [db errorcode] == 3082 ||

         0==[lindex $res 0]

       }]

       if {$rc!=1} {

         puts "Error: $res"

       }

       set rc

     } {1}

     # If $::n is greater than the number of malloc() calls required to

     # execute the SQL, then this test is finished. Break out of the loop.

     set nFail [sqlite3_memdebug_fail -1]

     if {$nFail==0} {

       break

     }

     # Recover from the malloc failure.

     #

     # Update: The new malloc() failure handling means that a transaction may

     # still be active even if a malloc() has failed. But when these tests were

     # written this was not the case. So do a manual ROLLBACK here so that the

     # tests pass.

     do_test mallocC-$tn.$::n.2 {

       catch {

         execsql {

           ROLLBACK;

         }

       }

       expr 0

     } {0}

     # Checksum the database.

     #do_test mallocC-$tn.$::n.3 {

     #  allcksum db

     #} $sum

     #integrity_check mallocC-$tn.$::n.4

   if {$::nErr>1} return

   }

   unset ::mallocopts

 }

 sqlite3_extended_result_codes db 1

 execsql {

   PRAGMA auto_vacuum=1;

   CREATE TABLE t0(a, b, c);

 }

 do_mallocC_test 1 -sql {

   BEGIN;

   -- Allocate 32 new root pages. This will exercise the 'extract specific 

   -- page from the freelist' code when in auto-vacuum mode (see the

   -- allocatePage() routine in btree.c).

   CREATE TABLE t1(a UNIQUE, b UNIQUE, c UNIQUE);

   CREATE TABLE t2(a UNIQUE, b UNIQUE, c UNIQUE);

   CREATE TABLE t3(a UNIQUE, b UNIQUE, c UNIQUE);

   CREATE TABLE t4(a UNIQUE, b UNIQUE, c UNIQUE);

   CREATE TABLE t5(a UNIQUE, b UNIQUE, c UNIQUE);

   CREATE TABLE t6(a UNIQUE, b UNIQUE, c UNIQUE);

   CREATE TABLE t7(a UNIQUE, b UNIQUE, c UNIQUE);

   CREATE TABLE t8(a UNIQUE, b UNIQUE, c UNIQUE);

   ROLLBACK;

 }

 finish_test