# 2007 April 2

#

# 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 implements regression tests for SQLite library.  The

# focus of this file is testing for correct handling of I/O errors

# such as writes failing because the disk is full.

# 

# The tests in this file use special facilities that are only

# available in the SQLite test fixture.

#

# $Id: ioerr2.test,v 1.9 2008/08/20 14:49:25 danielk1977 Exp $

set testdir [file dirname $argv0]

source $testdir/tester.tcl

ifcapable !integrityck {

  finish_test

  return

}

do_test ioerr2-1.1 {

  execsql {

    PRAGMA cache_size = 10;

    PRAGMA default_cache_size = 10;

    CREATE TABLE t1(a, b, PRIMARY KEY(a, b));

    INSERT INTO t1 VALUES(randstr(400,400),randstr(400,400));

    INSERT INTO t1 SELECT randstr(400,400), randstr(400,400) FROM t1; -- 2

    INSERT INTO t1 SELECT randstr(400,400), randstr(400,400) FROM t1; -- 4

    INSERT INTO t1 SELECT randstr(400,400), randstr(400,400) FROM t1; -- 8

    INSERT INTO t1 SELECT randstr(400,400), randstr(400,400) FROM t1; -- 16

    INSERT INTO t1 SELECT randstr(400,400), randstr(400,400) FROM t1; -- 32

  }

} {}

set ::cksum [execsql {SELECT md5sum(a, b) FROM t1}]

proc check_db {testname} {

  # Make sure no I/O errors are simulated in this proc.

  set ::sqlite_io_error_hit 0

  set ::sqlite_io_error_persist 0

  set ::sqlite_io_error_pending 0

  # Run an integrity-check. If "disk I/O error" is returned, the

  # pager must be in error state. In this case open a new database

  # connection. Otherwise, try a ROLLBACK, in case a transaction 

  # is still active.

  set rc [catch {execsql {PRAGMA integrity_check}} msg]

  if {$rc && ($msg eq "disk I/O error" || $msg eq "database is locked")} {

    db close

    sqlite3 db test.db

    set refcnt 0

  } else {

    if {$rc || $msg ne "ok"} {

      error $msg

    }

    catch {execsql ROLLBACK}

  }

  # Check that the database checksum is still $::cksum, and that

  # the integrity-check passes.

  set ck [execsql {SELECT md5sum(a, b) FROM t1}]

  do_test ${testname}.cksum [list set ck $ck] $::cksum

  integrity_check ${testname}.integrity

  do_test ${testname}.refcnt {

    lindex [sqlite3_pager_refcounts db] 0

  } 0

}

check_db ioerr2-2

set sql {

  PRAGMA cache_size = 10;

  PRAGMA default_cache_size = 10;

  BEGIN;

  DELETE FROM t1 WHERE (oid%7)==0;

  INSERT INTO t1 SELECT randstr(400,400), randstr(400,400) 

    WHERE (random()%7)==0;

  UPDATE t1 SET a = randstr(400,400), b = randstr(400,400) 

    WHERE (random()%7)==0;

  ROLLBACK;

}

foreach bPersist [list 0 1] {

  set ::go 1

  for {set ::N 1} {$::go} {incr ::N} {

    db close

    sqlite3 db test.db

    set ::sqlite_io_error_hit 0

    set ::sqlite_io_error_persist $bPersist

    set ::sqlite_io_error_pending $::N

    foreach {::go res} [catchsql $sql] {}

    check_db ioerr2-3.$bPersist.$::N

  }

}

foreach bPersist [list 0 1] {

  set ::go 1

  for {set ::N 1} {$::go} {incr ::N} {

    set ::sqlite_io_error_hit 0

    set ::sqlite_io_error_persist $bPersist

    set ::sqlite_io_error_pending $::N

    foreach {::go res} [catchsql $sql] {}

    check_db ioerr2-4.[expr {$bPersist+2}].$::N

  }

}

do_test ioerr2-5 {

  execsql {

    CREATE TABLE t2 AS SELECT * FROM t1;

    PRAGMA temp_store = memory;

  }

  set ::sqlite_io_error_persist 0

  set ::go 1

  set rc [catch {

    for {set ::N 2} {$::N<200} {incr ::N} {

      db eval {SELECT * FROM t1 WHERE rowid IN (1, 5, 10, 15, 20)} {

        set ::sqlite_io_error_hit 0

        set ::sqlite_io_error_pending $::N

        set sql {UPDATE t2 SET b = randstr(400,400)}

        foreach {::go res} [catchsql $sql] {}

      }

    }

  } msg]

  list $rc $msg

} {1 {callback requested query abort}}

finish_test