88 lines
		
	
	
		
			2.3 KiB
		
	
	
	
		
			Plaintext
		
	
	
	
	
	
			
		
		
	
	
			88 lines
		
	
	
		
			2.3 KiB
		
	
	
	
		
			Plaintext
		
	
	
	
	
	
# 2009 January 30
 | 
						|
#
 | 
						|
# 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 the handling of OOM errors by the
 | 
						|
# sqlite3_backup_XXX APIs.
 | 
						|
#
 | 
						|
# $Id: backup_malloc.test,v 1.2 2009/02/04 22:46:47 drh Exp $
 | 
						|
 | 
						|
set testdir [file dirname $argv0]
 | 
						|
source $testdir/tester.tcl
 | 
						|
 | 
						|
source $testdir/malloc_common.tcl
 | 
						|
 | 
						|
do_malloc_test backup_malloc-1 -tclprep {
 | 
						|
  execsql {
 | 
						|
    PRAGMA cache_size = 10;
 | 
						|
    BEGIN;
 | 
						|
    CREATE TABLE t1(a, b);
 | 
						|
    INSERT INTO t1 VALUES(1, randstr(1000,1000));
 | 
						|
    INSERT INTO t1 SELECT a+ 1, randstr(1000,1000) FROM t1;
 | 
						|
    INSERT INTO t1 SELECT a+ 2, randstr(1000,1000) FROM t1;
 | 
						|
    INSERT INTO t1 SELECT a+ 4, randstr(1000,1000) FROM t1;
 | 
						|
    INSERT INTO t1 SELECT a+ 8, randstr(1000,1000) FROM t1;
 | 
						|
    INSERT INTO t1 SELECT a+16, randstr(1000,1000) FROM t1;
 | 
						|
    INSERT INTO t1 SELECT a+32, randstr(1000,1000) FROM t1;
 | 
						|
    INSERT INTO t1 SELECT a+64, randstr(1000,1000) FROM t1;
 | 
						|
    CREATE INDEX i1 ON t1(b);
 | 
						|
    COMMIT;
 | 
						|
  }
 | 
						|
  sqlite3 db2 test2.db
 | 
						|
  execsql { PRAGMA cache_size = 10 } db2
 | 
						|
} -tclbody {
 | 
						|
 | 
						|
  # Create a backup object.
 | 
						|
  #
 | 
						|
  set rc [catch {sqlite3_backup B db2 main db main}]
 | 
						|
  if {$rc && [sqlite3_errcode db2] == "SQLITE_NOMEM"} {
 | 
						|
    error "out of memory"
 | 
						|
  }
 | 
						|
 | 
						|
  # Run the backup process some.
 | 
						|
  #
 | 
						|
  set rc [B step 50]
 | 
						|
  if {$rc == "SQLITE_NOMEM" || $rc == "SQLITE_IOERR_NOMEM"} {
 | 
						|
    error "out of memory"
 | 
						|
  }
 | 
						|
  
 | 
						|
  # Update the database.
 | 
						|
  #
 | 
						|
  execsql { UPDATE t1 SET a = a + 1 }
 | 
						|
  
 | 
						|
  # Finish doing the backup.
 | 
						|
  #
 | 
						|
  set rc [B step 5000]
 | 
						|
  if {$rc == "SQLITE_NOMEM" || $rc == "SQLITE_IOERR_NOMEM"} {
 | 
						|
    error "out of memory"
 | 
						|
  }
 | 
						|
 
 | 
						|
  # Finalize the backup.
 | 
						|
  B finish
 | 
						|
} -cleanup {
 | 
						|
  catch { B finish }
 | 
						|
  catch { db2 close }
 | 
						|
}
 | 
						|
 | 
						|
do_malloc_test backup_malloc-2 -tclprep {
 | 
						|
  sqlite3 db2 test2.db
 | 
						|
} -tclbody {
 | 
						|
  set rc [catch {sqlite3_backup B db2 temp db main}]
 | 
						|
  set errcode [sqlite3_errcode db2]
 | 
						|
  if {$rc && ($errcode == "SQLITE_NOMEM" || $errcode == "SQLITE_IOERR_NOMEM")} {
 | 
						|
    error "out of memory"
 | 
						|
  }
 | 
						|
} -cleanup {
 | 
						|
  catch { B finish }
 | 
						|
  db2 close
 | 
						|
}
 | 
						|
 | 
						|
finish_test
 |