| OLD | NEW |
| (Empty) |
| 1 # 2010 April 13 | |
| 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. The | |
| 12 # focus of this file is testing the operation of "blocking-checkpoint" | |
| 13 # operations. | |
| 14 # | |
| 15 | |
| 16 set testdir [file dirname $argv0] | |
| 17 source $testdir/tester.tcl | |
| 18 source $testdir/lock_common.tcl | |
| 19 source $testdir/wal_common.tcl | |
| 20 ifcapable !wal {finish_test ; return } | |
| 21 | |
| 22 set testprefix wal5 | |
| 23 | |
| 24 proc db_page_count {{file test.db}} { expr [file size $file] / 1024 } | |
| 25 proc wal_page_count {{file test.db}} { wal_frame_count ${file}-wal 1024 } | |
| 26 | |
| 27 | |
| 28 # A checkpoint may be requested either using the C API or by executing | |
| 29 # an SQL PRAGMA command. To test both methods, all tests in this file are | |
| 30 # run twice - once using each method to request checkpoints. | |
| 31 # | |
| 32 foreach {testprefix do_wal_checkpoint} { | |
| 33 | |
| 34 wal5-pragma { | |
| 35 proc do_wal_checkpoint { dbhandle args } { | |
| 36 array set a $args | |
| 37 foreach key [array names a] { | |
| 38 if {[lsearch {-mode -db} $key]<0} { error "unknown switch: $key" } | |
| 39 } | |
| 40 | |
| 41 set sql "PRAGMA " | |
| 42 if {[info exists a(-db)]} { append sql "$a(-db)." } | |
| 43 append sql "wal_checkpoint" | |
| 44 if {[info exists a(-mode)]} { append sql " = $a(-mode)" } | |
| 45 | |
| 46 uplevel [list $dbhandle eval $sql] | |
| 47 } | |
| 48 } | |
| 49 | |
| 50 wal5-capi { | |
| 51 proc do_wal_checkpoint { dbhandle args } { | |
| 52 set a(-mode) passive | |
| 53 array set a $args | |
| 54 foreach key [array names a] { | |
| 55 if {[lsearch {-mode -db} $key]<0} { error "unknown switch: $key" } | |
| 56 } | |
| 57 | |
| 58 set vals {restart full truncate} | |
| 59 if {[lsearch -exact $vals $a(-mode)]<0} { set a(-mode) passive } | |
| 60 | |
| 61 set cmd [list sqlite3_wal_checkpoint_v2 $dbhandle $a(-mode)] | |
| 62 if {[info exists a(-db)]} { lappend sql $a(-db) } | |
| 63 | |
| 64 uplevel $cmd | |
| 65 } | |
| 66 } | |
| 67 } { | |
| 68 | |
| 69 eval $do_wal_checkpoint | |
| 70 | |
| 71 do_multiclient_test tn { | |
| 72 | |
| 73 set ::nBusyHandler 0 | |
| 74 set ::busy_handler_script "" | |
| 75 proc busyhandler {n} { | |
| 76 incr ::nBusyHandler | |
| 77 eval $::busy_handler_script | |
| 78 return 0 | |
| 79 } | |
| 80 | |
| 81 proc reopen_all {} { | |
| 82 code1 {db close} | |
| 83 code2 {db2 close} | |
| 84 code3 {db3 close} | |
| 85 | |
| 86 code1 {sqlite3 db test.db} | |
| 87 code2 {sqlite3 db2 test.db} | |
| 88 code3 {sqlite3 db3 test.db} | |
| 89 | |
| 90 sql1 { PRAGMA synchronous = NORMAL } | |
| 91 code1 { db busy busyhandler } | |
| 92 } | |
| 93 | |
| 94 do_test 1.$tn.1 { | |
| 95 reopen_all | |
| 96 sql1 { | |
| 97 PRAGMA page_size = 1024; | |
| 98 PRAGMA auto_vacuum = 0; | |
| 99 CREATE TABLE t1(x, y); | |
| 100 PRAGMA journal_mode = WAL; | |
| 101 INSERT INTO t1 VALUES(1, zeroblob(1200)); | |
| 102 INSERT INTO t1 VALUES(2, zeroblob(1200)); | |
| 103 INSERT INTO t1 VALUES(3, zeroblob(1200)); | |
| 104 } | |
| 105 expr [file size test.db] / 1024 | |
| 106 } {2} | |
| 107 | |
| 108 # Have connection 2 grab a read-lock on the current snapshot. | |
| 109 do_test 1.$tn.2 { sql2 { BEGIN; SELECT x FROM t1 } } {1 2 3} | |
| 110 | |
| 111 # Attempt a checkpoint. | |
| 112 do_test 1.$tn.3 { | |
| 113 code1 { do_wal_checkpoint db } | |
| 114 list [db_page_count] [wal_page_count] | |
| 115 } {5 9} | |
| 116 | |
| 117 # Write to the db again. The log cannot wrap because of the lock still | |
| 118 # held by connection 2. The busy-handler has not yet been invoked. | |
| 119 do_test 1.$tn.4 { | |
| 120 sql1 { INSERT INTO t1 VALUES(4, zeroblob(1200)) } | |
| 121 list [db_page_count] [wal_page_count] $::nBusyHandler | |
| 122 } {5 12 0} | |
| 123 | |
| 124 # Now do a blocking-checkpoint. Set the busy-handler up so that connection | |
| 125 # 2 releases its lock on the 6th invocation. The checkpointer should then | |
| 126 # proceed to checkpoint the entire log file. Next write should go to the | |
| 127 # start of the log file. | |
| 128 # | |
| 129 set ::busy_handler_script { if {$n==5} { sql2 COMMIT } } | |
| 130 do_test 1.$tn.5 { | |
| 131 code1 { do_wal_checkpoint db -mode restart } | |
| 132 list [db_page_count] [wal_page_count] $::nBusyHandler | |
| 133 } {6 12 6} | |
| 134 do_test 1.$tn.6 { | |
| 135 set ::nBusyHandler 0 | |
| 136 sql1 { INSERT INTO t1 VALUES(5, zeroblob(1200)) } | |
| 137 list [db_page_count] [wal_page_count] $::nBusyHandler | |
| 138 } {6 12 0} | |
| 139 | |
| 140 do_test 1.$tn.7 { | |
| 141 reopen_all | |
| 142 list [db_page_count] [wal_page_count] $::nBusyHandler | |
| 143 } {7 0 0} | |
| 144 | |
| 145 do_test 1.$tn.8 { sql2 { BEGIN ; SELECT x FROM t1 } } {1 2 3 4 5} | |
| 146 do_test 1.$tn.9 { | |
| 147 sql1 { INSERT INTO t1 VALUES(6, zeroblob(1200)) } | |
| 148 list [db_page_count] [wal_page_count] $::nBusyHandler | |
| 149 } {7 5 0} | |
| 150 do_test 1.$tn.10 { sql3 { BEGIN ; SELECT x FROM t1 } } {1 2 3 4 5 6} | |
| 151 | |
| 152 set ::busy_handler_script { | |
| 153 if {$n==5} { sql2 COMMIT } | |
| 154 if {$n==6} { set ::db_file_size [db_page_count] } | |
| 155 if {$n==7} { sql3 COMMIT } | |
| 156 } | |
| 157 do_test 1.$tn.11 { | |
| 158 code1 { do_wal_checkpoint db -mode restart } | |
| 159 list [db_page_count] [wal_page_count] $::nBusyHandler | |
| 160 } {10 5 8} | |
| 161 do_test 1.$tn.12 { set ::db_file_size } 10 | |
| 162 } | |
| 163 | |
| 164 #------------------------------------------------------------------------- | |
| 165 # This block of tests explores checkpoint operations on more than one | |
| 166 # database file. | |
| 167 # | |
| 168 proc setup_and_attach_aux {} { | |
| 169 sql1 { ATTACH 'test.db2' AS aux } | |
| 170 sql2 { ATTACH 'test.db2' AS aux } | |
| 171 sql3 { ATTACH 'test.db2' AS aux } | |
| 172 sql1 { | |
| 173 PRAGMA aux.auto_vacuum = 0; | |
| 174 PRAGMA main.auto_vacuum = 0; | |
| 175 PRAGMA main.page_size=1024; PRAGMA main.journal_mode=WAL; | |
| 176 PRAGMA aux.page_size=1024; PRAGMA aux.journal_mode=WAL; | |
| 177 } | |
| 178 } | |
| 179 | |
| 180 proc file_page_counts {} { | |
| 181 list [db_page_count test.db ] \ | |
| 182 [wal_page_count test.db ] \ | |
| 183 [db_page_count test.db2] \ | |
| 184 [wal_page_count test.db2] | |
| 185 } | |
| 186 | |
| 187 # Test that executing "PRAGMA wal_checkpoint" checkpoints all attached | |
| 188 # databases, not just the main db. In capi mode, check that this is | |
| 189 # true if a NULL pointer is passed to wal_checkpoint_v2() in place of a | |
| 190 # database name. | |
| 191 do_multiclient_test tn { | |
| 192 setup_and_attach_aux | |
| 193 do_test 2.1.$tn.1 { | |
| 194 sql1 { | |
| 195 CREATE TABLE t1(a, b); | |
| 196 INSERT INTO t1 VALUES(1, 2); | |
| 197 CREATE TABLE aux.t2(a, b); | |
| 198 INSERT INTO t2 VALUES(1, 2); | |
| 199 } | |
| 200 } {} | |
| 201 do_test 2.2.$tn.2 { file_page_counts } {1 3 1 3} | |
| 202 do_test 2.1.$tn.3 { code1 { do_wal_checkpoint db } } {0 3 3} | |
| 203 do_test 2.1.$tn.4 { file_page_counts } {2 3 2 3} | |
| 204 } | |
| 205 | |
| 206 do_multiclient_test tn { | |
| 207 setup_and_attach_aux | |
| 208 do_test 2.2.$tn.1 { | |
| 209 execsql { | |
| 210 CREATE TABLE t1(a, b); | |
| 211 INSERT INTO t1 VALUES(1, 2); | |
| 212 CREATE TABLE aux.t2(a, b); | |
| 213 INSERT INTO t2 VALUES(1, 2); | |
| 214 INSERT INTO t2 VALUES(3, 4); | |
| 215 } | |
| 216 } {} | |
| 217 do_test 2.2.$tn.2 { file_page_counts } {1 3 1 4} | |
| 218 do_test 2.2.$tn.3 { sql2 { BEGIN; SELECT * FROM t1 } } {1 2} | |
| 219 do_test 2.2.$tn.4 { code1 { do_wal_checkpoint db -mode restart } } {1 3 3} | |
| 220 do_test 2.2.$tn.5 { file_page_counts } {2 3 2 4} | |
| 221 } | |
| 222 | |
| 223 do_multiclient_test tn { | |
| 224 setup_and_attach_aux | |
| 225 do_test 2.3.$tn.1 { | |
| 226 execsql { | |
| 227 CREATE TABLE t1(a, b); | |
| 228 INSERT INTO t1 VALUES(1, 2); | |
| 229 CREATE TABLE aux.t2(a, b); | |
| 230 INSERT INTO t2 VALUES(1, 2); | |
| 231 } | |
| 232 } {} | |
| 233 do_test 2.3.$tn.2 { file_page_counts } {1 3 1 3} | |
| 234 do_test 2.3.$tn.3 { sql2 { BEGIN; SELECT * FROM t1 } } {1 2} | |
| 235 do_test 2.3.$tn.4 { sql1 { INSERT INTO t1 VALUES(3, 4) } } {} | |
| 236 do_test 2.3.$tn.5 { sql1 { INSERT INTO t2 VALUES(3, 4) } } {} | |
| 237 do_test 2.3.$tn.6 { file_page_counts } {1 4 1 4} | |
| 238 do_test 2.3.$tn.7 { code1 { do_wal_checkpoint db -mode full } } {1 4 3} | |
| 239 | |
| 240 # The checkpoint above only writes page 1 of the db file. The other | |
| 241 # page (page 2) is locked by the read-transaction opened by the | |
| 242 # [sql2] commmand above. So normally, the db is 1 page in size here. | |
| 243 # However, in mmap() mode, the db is pre-allocated to 2 pages at the | |
| 244 # start of the checkpoint, even though page 2 cannot be written. | |
| 245 set nDb 2 | |
| 246 if {[permutation]!="mmap"} {set nDb 1} | |
| 247 ifcapable !mmap {set nDb 1} | |
| 248 do_test 2.3.$tn.8 { file_page_counts } [list $nDb 4 2 4] | |
| 249 } | |
| 250 | |
| 251 # Check that checkpoints block on the correct locks. And respond correctly | |
| 252 # if they cannot obtain those locks. There are three locks that a checkpoint | |
| 253 # may block on (in the following order): | |
| 254 # | |
| 255 # 1. The writer lock: FULL and RESTART checkpoints block until any writer | |
| 256 # process releases its lock. | |
| 257 # | |
| 258 # 2. Readers using part of the log file. FULL and RESTART checkpoints block | |
| 259 # until readers using part (but not all) of the log file have finished. | |
| 260 # | |
| 261 # 3. Readers using any of the log file. After copying data into the | |
| 262 # database file, RESTART checkpoints block until readers using any part | |
| 263 # of the log file have finished. | |
| 264 # | |
| 265 # This test case involves running a checkpoint while there exist other | |
| 266 # processes holding all three types of locks. | |
| 267 # | |
| 268 foreach {tn1 checkpoint busy_on ckpt_expected expected} { | |
| 269 1 PASSIVE - {0 3 3} - | |
| 270 2 TYPO - {0 3 3} - | |
| 271 | |
| 272 3 FULL - {0 4 4} 2 | |
| 273 4 FULL 1 {1 3 3} 1 | |
| 274 5 FULL 2 {1 4 3} 2 | |
| 275 6 FULL 3 {0 4 4} 2 | |
| 276 | |
| 277 7 RESTART - {0 4 4} 3 | |
| 278 8 RESTART 1 {1 3 3} 1 | |
| 279 9 RESTART 2 {1 4 3} 2 | |
| 280 10 RESTART 3 {1 4 4} 3 | |
| 281 | |
| 282 11 TRUNCATE - {0 0 0} 3 | |
| 283 12 TRUNCATE 1 {1 3 3} 1 | |
| 284 13 TRUNCATE 2 {1 4 3} 2 | |
| 285 14 TRUNCATE 3 {1 4 4} 3 | |
| 286 | |
| 287 } { | |
| 288 do_multiclient_test tn { | |
| 289 setup_and_attach_aux | |
| 290 | |
| 291 proc busyhandler {x} { | |
| 292 set ::max_busyhandler $x | |
| 293 if {$::busy_on!="-" && $x==$::busy_on} { return 1 } | |
| 294 switch -- $x { | |
| 295 1 { sql2 "COMMIT ; BEGIN ; SELECT * FROM t1" } | |
| 296 2 { sql3 "COMMIT" } | |
| 297 3 { sql2 "COMMIT" } | |
| 298 } | |
| 299 return 0 | |
| 300 } | |
| 301 set ::max_busyhandler - | |
| 302 | |
| 303 do_test 2.4.$tn1.$tn.1 { | |
| 304 sql1 { | |
| 305 CREATE TABLE t1(a, b); | |
| 306 INSERT INTO t1 VALUES(1, 2); | |
| 307 } | |
| 308 sql2 { BEGIN; INSERT INTO t1 VALUES(3, 4) } | |
| 309 sql3 { BEGIN; SELECT * FROM t1 } | |
| 310 } {1 2} | |
| 311 | |
| 312 do_test 2.4.$tn1.$tn.2 { | |
| 313 code1 { db busy busyhandler } | |
| 314 code1 { do_wal_checkpoint db -mode [string tolower $checkpoint] } | |
| 315 } $ckpt_expected | |
| 316 do_test 2.4.$tn1.$tn.3 { set ::max_busyhandler } $expected | |
| 317 } | |
| 318 } | |
| 319 | |
| 320 | |
| 321 do_multiclient_test tn { | |
| 322 | |
| 323 code1 $do_wal_checkpoint | |
| 324 code2 $do_wal_checkpoint | |
| 325 code3 $do_wal_checkpoint | |
| 326 | |
| 327 do_test 3.$tn.1 { | |
| 328 sql1 { | |
| 329 PRAGMA auto_vacuum = 0; | |
| 330 PRAGMA journal_mode = WAL; | |
| 331 PRAGMA synchronous = normal; | |
| 332 CREATE TABLE t1(x, y); | |
| 333 } | |
| 334 | |
| 335 sql2 { PRAGMA journal_mode } | |
| 336 sql3 { PRAGMA journal_mode } | |
| 337 } {wal} | |
| 338 | |
| 339 do_test 3.$tn.2 { code2 { do_wal_checkpoint db2 } } {0 2 2} | |
| 340 | |
| 341 do_test 3.$tn.3 { code2 { do_wal_checkpoint db2 } } {0 2 2} | |
| 342 | |
| 343 do_test 3.$tn.4 { code3 { do_wal_checkpoint db3 } } {0 2 2} | |
| 344 | |
| 345 code1 {db close} | |
| 346 code2 {db2 close} | |
| 347 code3 {db3 close} | |
| 348 | |
| 349 code1 {sqlite3 db test.db} | |
| 350 code2 {sqlite3 db2 test.db} | |
| 351 code3 {sqlite3 db3 test.db} | |
| 352 | |
| 353 do_test 3.$tn.5 { sql3 { PRAGMA journal_mode } } {wal} | |
| 354 | |
| 355 do_test 3.$tn.6 { code3 { do_wal_checkpoint db3 } } {0 0 0} | |
| 356 } | |
| 357 | |
| 358 # Test SQLITE_CHECKPOINT_TRUNCATE. | |
| 359 # | |
| 360 do_multiclient_test tn { | |
| 361 | |
| 362 code1 $do_wal_checkpoint | |
| 363 code2 $do_wal_checkpoint | |
| 364 code3 $do_wal_checkpoint | |
| 365 | |
| 366 do_test 4.$tn.1 { | |
| 367 sql1 { | |
| 368 PRAGMA page_size = 1024; | |
| 369 PRAGMA auto_vacuum = 0; | |
| 370 PRAGMA journal_mode = WAL; | |
| 371 PRAGMA synchronous = normal; | |
| 372 CREATE TABLE t1(x, y); | |
| 373 CREATE INDEX i1 ON t1(x, y); | |
| 374 INSERT INTO t1 VALUES(1, 2); | |
| 375 INSERT INTO t1 VALUES(3, 4); | |
| 376 } | |
| 377 file size test.db-wal | |
| 378 } [wal_file_size 8 1024] | |
| 379 | |
| 380 do_test 4.$tn.2 { do_wal_checkpoint db -mode truncate } {0 0 0} | |
| 381 do_test 4.$tn.3 { file size test.db-wal } 0 | |
| 382 | |
| 383 do_test 4.$tn.4 { | |
| 384 sql2 { SELECT * FROM t1 } | |
| 385 } {1 2 3 4} | |
| 386 | |
| 387 do_test 4.$tn.5 { | |
| 388 sql2 { INSERT INTO t1 VALUES('a', 'b') } | |
| 389 file size test.db-wal | |
| 390 } [wal_file_size 2 1024] | |
| 391 | |
| 392 } | |
| 393 | |
| 394 # Test that FULL, RESTART and TRUNCATE callbacks block on other clients | |
| 395 # and truncate the wal file as required even if the entire wal file has | |
| 396 # already been checkpointed when they are invoked. | |
| 397 # | |
| 398 do_multiclient_test tn { | |
| 399 | |
| 400 code1 $do_wal_checkpoint | |
| 401 code2 $do_wal_checkpoint | |
| 402 code3 $do_wal_checkpoint | |
| 403 | |
| 404 do_test 5.$tn.1 { | |
| 405 sql1 { | |
| 406 PRAGMA page_size = 1024; | |
| 407 PRAGMA auto_vacuum = 0; | |
| 408 PRAGMA journal_mode = WAL; | |
| 409 PRAGMA synchronous = normal; | |
| 410 CREATE TABLE t1(x, y); | |
| 411 CREATE INDEX i1 ON t1(x, y); | |
| 412 INSERT INTO t1 VALUES(1, 2); | |
| 413 INSERT INTO t1 VALUES(3, 4); | |
| 414 INSERT INTO t1 VALUES(5, 6); | |
| 415 } | |
| 416 file size test.db-wal | |
| 417 } [wal_file_size 10 1024] | |
| 418 | |
| 419 do_test 5.$tn.2 { | |
| 420 sql2 { BEGIN; SELECT * FROM t1 } | |
| 421 } {1 2 3 4 5 6} | |
| 422 | |
| 423 do_test 5.$tn.3 { do_wal_checkpoint db -mode passive } {0 10 10} | |
| 424 | |
| 425 do_test 5.$tn.4 { | |
| 426 sql3 { BEGIN; INSERT INTO t1 VALUES(7, 8); } | |
| 427 } {} | |
| 428 | |
| 429 do_test 5.$tn.5 { do_wal_checkpoint db -mode passive } {0 10 10} | |
| 430 do_test 5.$tn.6 { do_wal_checkpoint db -mode full } {1 10 10} | |
| 431 | |
| 432 do_test 5.$tn.7 { sql3 { ROLLBACK } } {} | |
| 433 | |
| 434 do_test 5.$tn.8 { do_wal_checkpoint db -mode full } {0 10 10} | |
| 435 do_test 5.$tn.9 { do_wal_checkpoint db -mode truncate } {1 10 10} | |
| 436 | |
| 437 do_test 5.$tn.10 { | |
| 438 file size test.db-wal | |
| 439 } [wal_file_size 10 1024] | |
| 440 | |
| 441 proc xBusyHandler {n} { sql2 { COMMIT } ; return 0 } | |
| 442 db busy xBusyHandler | |
| 443 | |
| 444 do_test 5.$tn.11 { do_wal_checkpoint db -mode truncate } {0 0 0} | |
| 445 do_test 5.$tn.12 { file size test.db-wal } 0 | |
| 446 | |
| 447 do_test 5.$tn.13 { | |
| 448 sql1 { | |
| 449 INSERT INTO t1 VALUES(7, 8); | |
| 450 INSERT INTO t1 VALUES(9, 10); | |
| 451 SELECT * FROM t1; | |
| 452 } | |
| 453 } {1 2 3 4 5 6 7 8 9 10} | |
| 454 | |
| 455 do_test 5.$tn.14 { | |
| 456 sql2 { BEGIN; SELECT * FROM t1 } | |
| 457 } {1 2 3 4 5 6 7 8 9 10} | |
| 458 | |
| 459 proc xBusyHandler {n} { return 1 } | |
| 460 do_test 5.$tn.15 { do_wal_checkpoint db -mode truncate } {1 4 4} | |
| 461 do_test 5.$tn.16 { file size test.db-wal } [wal_file_size 4 1024] | |
| 462 | |
| 463 do_test 5.$tn.17 { do_wal_checkpoint db -mode restart } {1 4 4} | |
| 464 | |
| 465 proc xBusyHandler {n} { sql2 { COMMIT } ; return 0 } | |
| 466 db busy xBusyHandler | |
| 467 do_test 5.$tn.18 { do_wal_checkpoint db -mode restart } {0 4 4} | |
| 468 do_test 5.$tn.19 { file size test.db-wal } [wal_file_size 4 1024] | |
| 469 | |
| 470 do_test 5.$tn.20 { do_wal_checkpoint db -mode truncate } {0 0 0} | |
| 471 do_test 5.$tn.21 { file size test.db-wal } 0 | |
| 472 } | |
| 473 | |
| 474 } | |
| 475 | |
| 476 | |
| 477 finish_test | |
| OLD | NEW |