1# 2011 Mar 16 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# 12# The focus of this file is testing the session module. 13# 14 15if {![info exists testdir]} { 16 set testdir [file join [file dirname [info script]] .. .. test] 17} 18source [file join [file dirname [info script]] session_common.tcl] 19source $testdir/tester.tcl 20 21set testprefix session2 22 23proc test_reset {} { 24 catch { db close } 25 catch { db2 close } 26 forcedelete test.db test.db2 27 sqlite3 db test.db 28 sqlite3 db2 test.db2 29} 30 31########################################################################## 32# End of proc definitions. Start of tests. 33########################################################################## 34 35test_reset 36do_execsql_test 1.0 { 37 CREATE TABLE t1(a PRIMARY KEY, b); 38 INSERT INTO t1 VALUES('i', 'one'); 39} 40do_iterator_test 1.1 t1 { 41 DELETE FROM t1 WHERE a = 'i'; 42 INSERT INTO t1 VALUES('ii', 'two'); 43} { 44 {DELETE t1 0 X. {t i t one} {}} 45 {INSERT t1 0 X. {} {t ii t two}} 46} 47 48do_iterator_test 1.2 t1 { 49 INSERT INTO t1 VALUES(1.5, 99.9) 50} { 51 {INSERT t1 0 X. {} {f 1.5 f 99.9}} 52} 53 54do_iterator_test 1.3 t1 { 55 UPDATE t1 SET b = 100.1 WHERE a = 1.5; 56 UPDATE t1 SET b = 99.9 WHERE a = 1.5; 57} { } 58 59do_iterator_test 1.4 t1 { 60 UPDATE t1 SET b = 100.1 WHERE a = 1.5; 61} { 62 {UPDATE t1 0 X. {f 1.5 f 99.9} {{} {} f 100.1}} 63} 64 65 66# Execute each of the following blocks of SQL on database [db1]. Collect 67# changes using a session object. Apply the resulting changeset to 68# database [db2]. Then check that the contents of the two databases are 69# identical. 70# 71 72set set_of_tests { 73 1 { INSERT INTO %T1% VALUES(1, 2) } 74 75 2 { 76 INSERT INTO %T2% VALUES(1, NULL); 77 INSERT INTO %T2% VALUES(2, NULL); 78 INSERT INTO %T2% VALUES(3, NULL); 79 DELETE FROM %T2% WHERE a = 2; 80 INSERT INTO %T2% VALUES(4, NULL); 81 UPDATE %T2% SET b=0 WHERE b=1; 82 } 83 84 3 { INSERT INTO %T3% SELECT *, NULL FROM %T2% } 85 86 4 { 87 INSERT INTO %T3% SELECT a||a, b||b, NULL FROM %T3%; 88 DELETE FROM %T3% WHERE rowid%2; 89 } 90 91 5 { UPDATE %T3% SET c = a||b } 92 93 6 { UPDATE %T1% SET a = 32 } 94 95 7 { 96 INSERT INTO %T1% SELECT randomblob(32), randomblob(32) FROM %T1%; 97 INSERT INTO %T1% SELECT randomblob(32), randomblob(32) FROM %T1%; 98 INSERT INTO %T1% SELECT randomblob(32), randomblob(32) FROM %T1%; 99 INSERT INTO %T1% SELECT randomblob(32), randomblob(32) FROM %T1%; 100 INSERT INTO %T1% SELECT randomblob(32), randomblob(32) FROM %T1%; 101 INSERT INTO %T1% SELECT randomblob(32), randomblob(32) FROM %T1%; 102 INSERT INTO %T1% SELECT randomblob(32), randomblob(32) FROM %T1%; 103 INSERT INTO %T1% SELECT randomblob(32), randomblob(32) FROM %T1%; 104 INSERT INTO %T1% SELECT randomblob(32), randomblob(32) FROM %T1%; 105 INSERT INTO %T1% SELECT randomblob(32), randomblob(32) FROM %T1%; 106 INSERT INTO %T1% SELECT randomblob(32), randomblob(32) FROM %T1%; 107 DELETE FROM %T1% WHERE (rowid%3)==0; 108 } 109 110 8 { 111 BEGIN; 112 INSERT INTO %T1% SELECT randomblob(32), randomblob(32) FROM %T1%; 113 ROLLBACK; 114 } 115 9 { 116 BEGIN; 117 UPDATE %T1% SET b = 'xxx'; 118 ROLLBACK; 119 } 120 10 { 121 BEGIN; 122 DELETE FROM %T1% WHERE 1; 123 ROLLBACK; 124 } 125 11 { 126 INSERT INTO %T1% VALUES(randomblob(21000), randomblob(0)); 127 INSERT INTO %T1% VALUES(1.5, 1.5); 128 INSERT INTO %T1% VALUES(4.56, -99.999999999999999999999); 129 } 130 12 { 131 INSERT INTO %T2% VALUES(NULL, NULL); 132 } 133 134 13 { 135 DELETE FROM %T1% WHERE 1; 136 137 -- Insert many rows with real primary keys. Enough to force the session 138 -- objects hash table to resize. 139 INSERT INTO %T1% VALUES(0.1, 0.1); 140 INSERT INTO %T1% SELECT a+0.1, b+0.1 FROM %T1%; 141 INSERT INTO %T1% SELECT a+0.2, b+0.2 FROM %T1%; 142 INSERT INTO %T1% SELECT a+0.4, b+0.4 FROM %T1%; 143 INSERT INTO %T1% SELECT a+0.8, b+0.8 FROM %T1%; 144 INSERT INTO %T1% SELECT a+1.6, b+1.6 FROM %T1%; 145 INSERT INTO %T1% SELECT a+3.2, b+3.2 FROM %T1%; 146 INSERT INTO %T1% SELECT a+6.4, b+6.4 FROM %T1%; 147 INSERT INTO %T1% SELECT a+12.8, b+12.8 FROM %T1%; 148 INSERT INTO %T1% SELECT a+25.6, b+25.6 FROM %T1%; 149 INSERT INTO %T1% SELECT a+51.2, b+51.2 FROM %T1%; 150 INSERT INTO %T1% SELECT a+102.4, b+102.4 FROM %T1%; 151 INSERT INTO %T1% SELECT a+204.8, b+204.8 FROM %T1%; 152 } 153 154 14 { 155 DELETE FROM %T1% WHERE 1; 156 } 157 158 15 { 159 INSERT INTO %T1% VALUES(1, 1); 160 INSERT INTO %T1% SELECT a+2, b+2 FROM %T1%; 161 INSERT INTO %T1% SELECT a+4, b+4 FROM %T1%; 162 INSERT INTO %T1% SELECT a+8, b+8 FROM %T1%; 163 INSERT INTO %T1% SELECT a+256, b+256 FROM %T1%; 164 } 165 166 16 { 167 INSERT INTO %T4% VALUES('abc', 'def'); 168 INSERT INTO %T4% VALUES('def', 'abc'); 169 } 170 17 { UPDATE %T4% SET b = 1 } 171 18 { DELETE FROM %T4% WHERE 1 } 172} 173 174test_reset 175do_common_sql { 176 CREATE TABLE t1(a PRIMARY KEY, b); 177 CREATE TABLE t2(a, b INTEGER PRIMARY KEY); 178 CREATE TABLE t3(a, b, c, PRIMARY KEY(a, b)); 179 CREATE TABLE t4(a, b, PRIMARY KEY(b, a)); 180} 181 182foreach {tn sql} [string map {%T1% t1 %T2% t2 %T3% t3 %T4% t4} $set_of_tests] { 183 do_then_apply_sql $sql 184 do_test 2.$tn { compare_db db db2 } {} 185} 186 187# The following block of tests is similar to the last, except that the 188# session object is recording changes made to an attached database. The 189# main database contains a table of the same name as the table being 190# modified within the attached db. 191# 192test_reset 193forcedelete test.db3 194sqlite3 db3 test.db3 195do_test 3.0 { 196 execsql { 197 ATTACH 'test.db3' AS 'aux'; 198 CREATE TABLE t1(a, b PRIMARY KEY); 199 CREATE TABLE t2(x, y, z); 200 CREATE TABLE t3(a); 201 202 CREATE TABLE aux.t1(a PRIMARY KEY, b); 203 CREATE TABLE aux.t2(a, b INTEGER PRIMARY KEY); 204 CREATE TABLE aux.t3(a, b, c, PRIMARY KEY(a, b)); 205 CREATE TABLE aux.t4(a, b, PRIMARY KEY(b, a)); 206 } 207 execsql { 208 CREATE TABLE t1(a PRIMARY KEY, b); 209 CREATE TABLE t2(a, b INTEGER PRIMARY KEY); 210 CREATE TABLE t3(a, b, c, PRIMARY KEY(a, b)); 211 CREATE TABLE t4(a, b, PRIMARY KEY(b, a)); 212 } db2 213} {} 214 215proc xTrace {args} { puts $args } 216 217foreach {tn sql} [ 218 string map {%T1% aux.t1 %T2% aux.t2 %T3% aux.t3 %T4% aux.t4} $set_of_tests 219] { 220 do_then_apply_sql $sql aux 221 do_test 3.$tn { compare_db db2 db3 } {} 222} 223catch {db3 close} 224 225 226#------------------------------------------------------------------------- 227# The following tests verify that NULL values in primary key columns are 228# handled correctly by the session module. 229# 230test_reset 231do_execsql_test 4.0 { 232 CREATE TABLE t1(a PRIMARY KEY); 233 CREATE TABLE t2(a, b, c, PRIMARY KEY(c, b)); 234 CREATE TABLE t3(a, b INTEGER PRIMARY KEY); 235} 236 237foreach {tn sql changeset} { 238 1 { 239 INSERT INTO t1 VALUES(123); 240 INSERT INTO t1 VALUES(NULL); 241 INSERT INTO t1 VALUES(456); 242 } { 243 {INSERT t1 0 X {} {i 456}} 244 {INSERT t1 0 X {} {i 123}} 245 } 246 247 2 { 248 UPDATE t1 SET a = NULL; 249 } { 250 {DELETE t1 0 X {i 456} {}} 251 {DELETE t1 0 X {i 123} {}} 252 } 253 254 3 { DELETE FROM t1 } { } 255 256 4 { 257 INSERT INTO t3 VALUES(NULL, NULL) 258 } { 259 {INSERT t3 0 .X {} {n {} i 1}} 260 } 261 262 5 { INSERT INTO t2 VALUES(1, 2, NULL) } { } 263 6 { INSERT INTO t2 VALUES(1, NULL, 3) } { } 264 7 { INSERT INTO t2 VALUES(1, NULL, NULL) } { } 265 8 { INSERT INTO t2 VALUES(1, 2, 3) } { {INSERT t2 0 .XX {} {i 1 i 2 i 3}} } 266 9 { DELETE FROM t2 WHERE 1 } { {DELETE t2 0 .XX {i 1 i 2 i 3} {}} } 267 268} { 269 do_iterator_test 4.$tn {t1 t2 t3} $sql $changeset 270} 271 272 273#------------------------------------------------------------------------- 274# Test that if NULL is passed to sqlite3session_attach(), all database 275# tables are attached to the session object. 276# 277test_reset 278do_execsql_test 5.0 { 279 CREATE TABLE t1(a PRIMARY KEY); 280 CREATE TABLE t2(x, y PRIMARY KEY); 281} 282 283foreach {tn sql changeset} { 284 1 { INSERT INTO t1 VALUES(35) } { {INSERT t1 0 X {} {i 35}} } 285 2 { INSERT INTO t2 VALUES(36, 37) } { {INSERT t2 0 .X {} {i 36 i 37}} } 286 3 { 287 DELETE FROM t1 WHERE 1; 288 UPDATE t2 SET x = 34; 289 } { 290 {UPDATE t2 0 .X {i 36 i 37} {i 34 {} {}}} 291 {DELETE t1 0 X {i 35} {}} 292 } 293} { 294 do_iterator_test 5.$tn * $sql $changeset 295} 296 297#------------------------------------------------------------------------- 298# The next block of tests verify that the "indirect" flag is set 299# correctly within changesets. The indirect flag is set for a change 300# if either of the following are true: 301# 302# * The sqlite3session_indirect() API has been used to set the session 303# indirect flag to true, or 304# * The change was made by a trigger. 305# 306# If the same row is updated more than once during a session, then the 307# change is considered indirect only if all changes meet the criteria 308# above. 309# 310test_reset 311db function indirect [list S indirect] 312 313do_execsql_test 6.0 { 314 CREATE TABLE t1(a PRIMARY KEY, b, c); 315 316 CREATE TABLE t2(x PRIMARY KEY, y); 317 CREATE TRIGGER AFTER INSERT ON t2 WHEN new.x%2 BEGIN 318 INSERT INTO t2 VALUES(new.x+1, NULL); 319 END; 320} 321 322do_iterator_test 6.1.1 * { 323 INSERT INTO t1 VALUES(1, 'one', 'i'); 324 SELECT indirect(1); 325 INSERT INTO t1 VALUES(2, 'two', 'ii'); 326 SELECT indirect(0); 327 INSERT INTO t1 VALUES(3, 'three', 'iii'); 328} { 329 {INSERT t1 0 X.. {} {i 1 t one t i}} 330 {INSERT t1 1 X.. {} {i 2 t two t ii}} 331 {INSERT t1 0 X.. {} {i 3 t three t iii}} 332} 333 334do_iterator_test 6.1.2 * { 335 SELECT indirect(1); 336 UPDATE t1 SET c = 'I' WHERE a = 1; 337 SELECT indirect(0); 338} { 339 {UPDATE t1 1 X.. {i 1 {} {} t i} {{} {} {} {} t I}} 340} 341do_iterator_test 6.1.3 * { 342 SELECT indirect(1); 343 UPDATE t1 SET c = '.' WHERE a = 1; 344 SELECT indirect(0); 345 UPDATE t1 SET c = 'o' WHERE a = 1; 346} { 347 {UPDATE t1 0 X.. {i 1 {} {} t I} {{} {} {} {} t o}} 348} 349do_iterator_test 6.1.4 * { 350 SELECT indirect(0); 351 UPDATE t1 SET c = 'x' WHERE a = 1; 352 SELECT indirect(1); 353 UPDATE t1 SET c = 'i' WHERE a = 1; 354} { 355 {UPDATE t1 0 X.. {i 1 {} {} t o} {{} {} {} {} t i}} 356} 357do_iterator_test 6.1.4 * { 358 SELECT indirect(1); 359 UPDATE t1 SET c = 'y' WHERE a = 1; 360 SELECT indirect(1); 361 UPDATE t1 SET c = 'I' WHERE a = 1; 362} { 363 {UPDATE t1 1 X.. {i 1 {} {} t i} {{} {} {} {} t I}} 364} 365 366do_iterator_test 6.1.5 * { 367 INSERT INTO t2 VALUES(1, 'x'); 368} { 369 {INSERT t2 0 X. {} {i 1 t x}} 370 {INSERT t2 1 X. {} {i 2 n {}}} 371} 372 373do_iterator_test 6.1.6 * { 374 SELECT indirect(1); 375 INSERT INTO t2 VALUES(3, 'x'); 376 SELECT indirect(0); 377 UPDATE t2 SET y = 'y' WHERE x>2; 378} { 379 {INSERT t2 0 X. {} {i 3 t y}} 380 {INSERT t2 0 X. {} {i 4 t y}} 381} 382 383do_iterator_test 6.1.7 * { 384 SELECT indirect(1); 385 DELETE FROM t2 WHERE x = 4; 386 SELECT indirect(0); 387 INSERT INTO t2 VALUES(4, 'new'); 388} { 389 {UPDATE t2 0 X. {i 4 t y} {{} {} t new}} 390} 391 392sqlite3session S db main 393do_execsql_test 6.2.1 { 394 SELECT indirect(0); 395 SELECT indirect(-1); 396 SELECT indirect(45); 397 SELECT indirect(-100); 398} {0 0 1 1} 399S delete 400 401#------------------------------------------------------------------------- 402# Test that if a conflict-handler that has been passed either NOTFOUND or 403# CONSTRAINT returns REPLACE - the sqlite3changeset_apply() call returns 404# MISUSE and rolls back any changes made so far. 405# 406# 7.1.*: NOTFOUND conflict-callback. 407# 7.2.*: CONSTRAINT conflict-callback. 408# 409proc xConflict {args} {return REPLACE} 410test_reset 411 412do_execsql_test 7.1.1 { 413 CREATE TABLE t1(a PRIMARY KEY, b); 414 INSERT INTO t1 VALUES(1, 'one'); 415 INSERT INTO t1 VALUES(2, 'two'); 416} 417do_test 7.1.2 { 418 execsql { 419 CREATE TABLE t1(a PRIMARY KEY, b NOT NULL); 420 INSERT INTO t1 VALUES(1, 'one'); 421 } db2 422} {} 423do_test 7.1.3 { 424 set changeset [changeset_from_sql { 425 UPDATE t1 SET b = 'five' WHERE a = 1; 426 UPDATE t1 SET b = 'six' WHERE a = 2; 427 }] 428 set x [list] 429 sqlite3session_foreach c $changeset { lappend x $c } 430 set x 431} [list \ 432 {UPDATE t1 0 X. {i 1 t one} {{} {} t five}} \ 433 {UPDATE t1 0 X. {i 2 t two} {{} {} t six}} \ 434] 435do_test 7.1.4 { 436 list [catch {sqlite3changeset_apply db2 $changeset xConflict} msg] $msg 437} {1 SQLITE_MISUSE} 438do_test 7.1.5 { execsql { SELECT * FROM t1 } db2 } {1 one} 439 440do_test 7.2.1 { 441 set changeset [changeset_from_sql { UPDATE t1 SET b = NULL WHERE a = 1 }] 442 443 set x [list] 444 sqlite3session_foreach c $changeset { lappend x $c } 445 set x 446} [list \ 447 {UPDATE t1 0 X. {i 1 t five} {{} {} n {}}} \ 448] 449do_test 7.2.2 { 450 list [catch {sqlite3changeset_apply db2 $changeset xConflict} msg] $msg 451} {1 SQLITE_MISUSE} 452do_test 7.2.3 { execsql { SELECT * FROM t1 } db2 } {1 one} 453 454#------------------------------------------------------------------------- 455# Test that if a conflict-handler returns ABORT, application of the 456# changeset is rolled back and the sqlite3changeset_apply() method returns 457# SQLITE_ABORT. 458# 459# Also test that the same thing happens if a conflict handler returns an 460# unrecognized integer value. Except, in this case SQLITE_MISUSE is returned 461# instead of SQLITE_ABORT. 462# 463foreach {tn conflict_return apply_return} { 464 1 ABORT SQLITE_ABORT 465 2 567 SQLITE_MISUSE 466} { 467 test_reset 468 proc xConflict {args} [list return $conflict_return] 469 470 do_test 8.$tn.0 { 471 do_common_sql { 472 CREATE TABLE t1(x, y, PRIMARY KEY(x, y)); 473 INSERT INTO t1 VALUES('x', 'y'); 474 } 475 execsql { INSERT INTO t1 VALUES('w', 'w') } 476 477 set changeset [changeset_from_sql { DELETE FROM t1 WHERE 1 }] 478 479 set x [list] 480 sqlite3session_foreach c $changeset { lappend x $c } 481 set x 482 } [list \ 483 {DELETE t1 0 XX {t w t w} {}} \ 484 {DELETE t1 0 XX {t x t y} {}} \ 485 ] 486 487 do_test 8.$tn.1 { 488 list [catch {sqlite3changeset_apply db2 $changeset xConflict} msg] $msg 489 } [list 1 $apply_return] 490 491 do_test 8.$tn.2 { 492 execsql {SELECT * FROM t1} db2 493 } {x y} 494} 495 496 497#------------------------------------------------------------------------- 498# Try to cause an infinite loop as follows: 499# 500# 1. Have a changeset insert a row that causes a CONFLICT callback, 501# 2. Have the conflict handler return REPLACE, 502# 3. After the session module deletes the conflicting row, have a trigger 503# re-insert it. 504# 4. Goto step 1... 505# 506# This doesn't work, as the second invocation of the conflict handler is a 507# CONSTRAINT, not a CONFLICT. There is at most one CONFLICT callback for 508# each change in the changeset. 509# 510test_reset 511proc xConflict {type args} { 512 if {$type == "CONFLICT"} { return REPLACE } 513 return OMIT 514} 515do_test 9.1 { 516 execsql { 517 CREATE TABLE t1(a PRIMARY KEY, b); 518 } 519 execsql { 520 CREATE TABLE t1(a PRIMARY KEY, b); 521 INSERT INTO t1 VALUES('x', 2); 522 CREATE TRIGGER tr1 AFTER DELETE ON t1 BEGIN 523 INSERT INTO t1 VALUES(old.a, old.b); 524 END; 525 } db2 526} {} 527do_test 9.2 { 528 set changeset [changeset_from_sql { INSERT INTO t1 VALUES('x', 1) }] 529 sqlite3changeset_apply db2 $changeset xConflict 530} {} 531do_test 9.3 { 532 execsql { SELECT * FROM t1 } db2 533} {x 2} 534 535#------------------------------------------------------------------------- 536# 537test_reset 538db function enable [list S enable] 539 540do_common_sql { 541 CREATE TABLE t1(a PRIMARY KEY, b); 542 INSERT INTO t1 VALUES('x', 'X'); 543} 544 545do_iterator_test 10.1 t1 { 546 INSERT INTO t1 VALUES('y', 'Y'); 547 SELECT enable(0); 548 INSERT INTO t1 VALUES('z', 'Z'); 549 SELECT enable(1); 550} { 551 {INSERT t1 0 X. {} {t y t Y}} 552} 553 554sqlite3session S db main 555do_execsql_test 10.2 { 556 SELECT enable(0); 557 SELECT enable(-1); 558 SELECT enable(1); 559 SELECT enable(-1); 560} {0 0 1 1} 561S delete 562 563finish_test 564