OLD | NEW |
| (Empty) |
1 # 2010 September 1 | |
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 | |
13 set testdir [file dirname $argv0] | |
14 source $testdir/tester.tcl | |
15 | |
16 # If SQLITE_CURDIR is not defined, omit this file. | |
17 ifcapable !curdir { | |
18 finish_test | |
19 return | |
20 } | |
21 | |
22 source $testdir/malloc_common.tcl | |
23 | |
24 unset -nocomplain defaultVfs | |
25 set defaultVfs [file_control_vfsname db] | |
26 db close | |
27 | |
28 do_test quota-1.1 { sqlite3_quota_initialize nosuchvfs 1 } {SQLITE_ERROR} | |
29 do_test quota-1.2 { sqlite3_quota_initialize "" 1 } {SQLITE_OK} | |
30 do_test quota-1.3 { sqlite3_quota_initialize "" 1 } {SQLITE_MISUSE} | |
31 do_test quota-1.4 { sqlite3_quota_shutdown } {SQLITE_OK} | |
32 | |
33 do_test quota-1.5 { sqlite3_quota_initialize "" 0 } {SQLITE_OK} | |
34 do_test quota-1.6 { sqlite3_quota_shutdown } {SQLITE_OK} | |
35 do_test quota-1.7 { sqlite3_quota_initialize "" 1 } {SQLITE_OK} | |
36 do_test quota-1.8 { sqlite3_quota_shutdown } {SQLITE_OK} | |
37 | |
38 | |
39 #------------------------------------------------------------------------- | |
40 # Some simple warm-body tests with a single database file in rollback | |
41 # mode: | |
42 # | |
43 # quota-2.1.*: Test that SQLITE_FULL is returned if the database would | |
44 # exceed the configured quota. | |
45 # | |
46 # quota-2.2.*: Test that SQLITE_FULL is not returned and the database | |
47 # grows if the callback extends the quota when the database | |
48 # attempts to grow beyond the configured quota. | |
49 # | |
50 # quota-2.3.*: Open and close a db that is not part of any quota group. At | |
51 # one point this was causing mutex refs to be leaked. | |
52 # | |
53 # quota-2.4.*: Try to shutdown the quota system before closing the db | |
54 # file. Check that this fails and the quota system still works | |
55 # afterwards. Then close the database and successfully shut | |
56 # down the quota system. | |
57 # | |
58 sqlite3_quota_initialize "" 1 | |
59 | |
60 unset -nocomplain quota_request_ok | |
61 proc quota_check {filename limitvar size} { | |
62 upvar $limitvar limit | |
63 | |
64 lappend ::quota [set limit] $size | |
65 if {[info exists ::quota_request_ok]} { set limit $size } | |
66 } | |
67 | |
68 do_test quota-2.1.1 { | |
69 sqlite3_quota_set *test.db 4096 quota_check | |
70 } {SQLITE_OK} | |
71 do_test quota-2.1.2 { | |
72 sqlite3 db test.db | |
73 execsql { | |
74 PRAGMA page_size=1024; | |
75 PRAGMA auto_vacuum=OFF; | |
76 PRAGMA journal_mode=DELETE; | |
77 } | |
78 set ::quota [list] | |
79 execsql { | |
80 CREATE TABLE t1(a, b); | |
81 INSERT INTO t1 VALUES(1, randomblob(1100)); | |
82 INSERT INTO t1 VALUES(2, randomblob(1100)); | |
83 } | |
84 set ::quota | |
85 } {} | |
86 do_test quota-2.1.2.1 { | |
87 file_control_vfsname db | |
88 } quota/$defaultVfs | |
89 do_test quota-2.1.3 { file size test.db } {4096} | |
90 do_test quota-2.1.4 { | |
91 catchsql { INSERT INTO t1 VALUES(3, randomblob(1100)) } | |
92 } {1 {database or disk is full}} | |
93 do_test quota-2.1.5 { set ::quota } {4096 5120} | |
94 | |
95 set ::quota_request_ok 1 | |
96 set ::quota [list] | |
97 do_test quota-2.2.1 { | |
98 execsql { INSERT INTO t1 VALUES(3, randomblob(1100)) } | |
99 } {} | |
100 do_test quota-2.2.2 { set ::quota } {4096 5120} | |
101 do_test quota-2.2.3 { file size test.db } {5120} | |
102 unset ::quota_request_ok | |
103 | |
104 do_test quota-2.3.1 { | |
105 sqlite3 db2 bak.db | |
106 db2 close | |
107 } {} | |
108 | |
109 do_test quota-2.4.1 { | |
110 sqlite3_quota_shutdown | |
111 } {SQLITE_MISUSE} | |
112 set ::quota [list] | |
113 do_test quota-2.4.2 { | |
114 catchsql { INSERT INTO t1 VALUES(3, randomblob(1100)) } | |
115 } {1 {database or disk is full}} | |
116 do_test quota-2.4.3 { set ::quota } {5120 6144} | |
117 do_test quota-2.4.4 { file size test.db } {5120} | |
118 do_test quota-2.4.99 { | |
119 db close | |
120 sqlite3_quota_shutdown | |
121 } {SQLITE_OK} | |
122 | |
123 #------------------------------------------------------------------------- | |
124 # Try some tests with more than one connection to a database file. Still | |
125 # in rollback mode. | |
126 # | |
127 # quota-3.1.*: Two connections to a single database file. | |
128 # | |
129 # quota-3.2.*: Two connections to each of several database files (that | |
130 # are in the same quota group). | |
131 # | |
132 proc quota_check {filename limitvar size} { | |
133 upvar $limitvar limit | |
134 lappend ::quota [set limit] $size | |
135 if {[info exists ::quota_request_ok]} { set limit $size } | |
136 } | |
137 | |
138 do_test quota-3.1.1 { | |
139 forcedelete test.db | |
140 sqlite3_quota_initialize "" 1 | |
141 sqlite3_quota_set *test.db 4096 quota_check | |
142 } {SQLITE_OK} | |
143 do_test quota-3.1.2 { | |
144 sqlite3 db test.db | |
145 execsql { | |
146 PRAGMA page_size = 1024; | |
147 PRAGMA journal_mode = delete; | |
148 PRAGMA auto_vacuum = off; | |
149 CREATE TABLE t1(a PRIMARY KEY, b); | |
150 INSERT INTO t1 VALUES(1, 'one'); | |
151 } | |
152 file size test.db | |
153 } {3072} | |
154 do_test quota-3.1.3 { | |
155 sqlite3 db2 test.db | |
156 set ::quota [list] | |
157 execsql { CREATE TABLE t2(a, b) } db2 | |
158 set ::quota | |
159 } {} | |
160 do_test quota-3.1.4 { | |
161 catchsql { CREATE TABLE t3(a, b) } | |
162 } {1 {database or disk is full}} | |
163 do_test quota-3.1.5 { | |
164 set ::quota_request_ok 1 | |
165 execsql { CREATE TABLE t3(a, b) } | |
166 } {} | |
167 do_test quota-3.1.6 { | |
168 db close | |
169 db2 close | |
170 sqlite3_quota_set *test.db 0 {} | |
171 } {SQLITE_OK} | |
172 | |
173 do_test quota-3.2.1 { | |
174 delete_file force test.db test2.db | |
175 | |
176 sqlite3_quota_set * 4096 {} | |
177 sqlite3 db1a test.db | |
178 sqlite3 db2a test2.db | |
179 | |
180 foreach db {db1a db2a} { | |
181 execsql { | |
182 PRAGMA page_size = 1024; | |
183 PRAGMA journal_mode = delete; | |
184 PRAGMA auto_vacuum = off; | |
185 CREATE TABLE t1(a, b); | |
186 } $db | |
187 } | |
188 | |
189 sqlite3 db1b test.db | |
190 sqlite3 db2b test2.db | |
191 | |
192 list [file size test.db] [file size test2.db] | |
193 } {2048 2048} | |
194 | |
195 catch { unset ::quota_request_ok } | |
196 | |
197 do_test quota-3.2.2 { execsql { INSERT INTO t1 VALUES('x', 'y') } db1a } {} | |
198 do_test quota-3.2.3 { execsql { INSERT INTO t1 VALUES('v', 'w') } db1b } {} | |
199 do_test quota-3.2.4 { execsql { INSERT INTO t1 VALUES('t', 'u') } db2a } {} | |
200 do_test quota-3.2.5 { execsql { INSERT INTO t1 VALUES('r', 's') } db2b } {} | |
201 | |
202 do_test quota-3.2.6 { | |
203 catchsql { INSERT INTO t1 VALUES(randomblob(500), randomblob(500)) } db1a | |
204 } {1 {database or disk is full}} | |
205 do_test quota-3.2.7 { | |
206 catchsql { INSERT INTO t1 VALUES(randomblob(500), randomblob(500)) } db1b | |
207 } {1 {database or disk is full}} | |
208 do_test quota-3.2.8 { | |
209 catchsql { INSERT INTO t1 VALUES(randomblob(500), randomblob(500)) } db2a | |
210 } {1 {database or disk is full}} | |
211 do_test quota-3.2.9 { | |
212 catchsql { INSERT INTO t1 VALUES(randomblob(500), randomblob(500)) } db2b | |
213 } {1 {database or disk is full}} | |
214 | |
215 set ::quota [list] | |
216 proc quota_callback {file limitvar size} { | |
217 upvar $limitvar limit | |
218 if {$::tcl_platform(platform)=="windows"} { | |
219 set file [ lindex [string map {\\ \/} $file] 0 ] | |
220 } | |
221 lappend ::quota $file $size | |
222 set limit 0 | |
223 } | |
224 sqlite3_quota_set * 4096 quota_callback | |
225 do_test quota-3.3.1 { | |
226 execsql { INSERT INTO t1 VALUES(randomblob(500), randomblob(500)) } db1a | |
227 execsql { INSERT INTO t1 VALUES(randomblob(500), randomblob(500)) } db1b | |
228 execsql { INSERT INTO t1 VALUES(randomblob(500), randomblob(500)) } db2a | |
229 execsql { INSERT INTO t1 VALUES(randomblob(500), randomblob(500)) } db2b | |
230 set ::quota | |
231 } [list [file join [get_pwd] test.db] 5120] | |
232 | |
233 do_test quota-3.2.X { | |
234 foreach db {db1a db2a db2b db1b} { catch { $db close } } | |
235 sqlite3_quota_set * 0 {} | |
236 } {SQLITE_OK} | |
237 | |
238 #------------------------------------------------------------------------- | |
239 # Quotas are deleted when unused and when their limit is set to zero | |
240 # | |
241 | |
242 # Return a list of all currently defined quotas. Each quota is identified | |
243 # by its pattern. | |
244 proc quota_list {} { | |
245 set allq {} | |
246 foreach q [sqlite3_quota_dump] { | |
247 lappend allq [lindex $q 0] | |
248 } | |
249 return [lsort $allq] | |
250 } | |
251 proc quota_size {name} { | |
252 set allq {} | |
253 foreach q [sqlite3_quota_dump] { | |
254 if {[lindex $q 0]==$name} {return [lindex $q 2]} | |
255 } | |
256 return 0 | |
257 } | |
258 | |
259 do_test quota-4.1.1 { | |
260 sqlite3_quota_set *test.db 0 {} | |
261 quota_list | |
262 } {} | |
263 do_test quota-4.1.2 { | |
264 sqlite3_quota_set *test.db 4096 {} | |
265 quota_list | |
266 } {*test.db} | |
267 do_test quota-4.1.3 { | |
268 sqlite3_quota_set *test2.db 0 {} | |
269 quota_list | |
270 } {*test.db} | |
271 do_test quota-4.1.4 { | |
272 sqlite3_quota_set *test2.db 100000 {} | |
273 quota_list | |
274 } {*test.db *test2.db} | |
275 do_test quota-4.1.5 { | |
276 sqlite3_quota_set *test.db 0 {} | |
277 quota_list | |
278 } {*test2.db} | |
279 do_test quota-4.1.6 { | |
280 forcedelete test2.db test2.db-journal test2.db-wal | |
281 sqlite3 db test2.db | |
282 db eval {CREATE TABLE t2(x); INSERT INTO t2 VALUES('tab-t2');} | |
283 quota_list | |
284 } {*test2.db} | |
285 do_test quota-4.1.7 { | |
286 catchsql {INSERT INTO t2 VALUES(zeroblob(200000))} | |
287 } {1 {database or disk is full}} | |
288 do_test quota-4.1.8 { | |
289 sqlite3 db2 test2.db | |
290 db2 eval {SELECT * FROM t2} | |
291 } {tab-t2} | |
292 do_test quota-4.1.9 { | |
293 sqlite3_quota_set *test2.db 0 {} | |
294 catchsql {INSERT INTO t2 VALUES(zeroblob(200000))} | |
295 } {0 {}} | |
296 do_test quota-4.1.10 { | |
297 quota_list | |
298 } {*test2.db} | |
299 do_test quota-4.1.11 { | |
300 db2 close | |
301 quota_list | |
302 } {*test2.db} | |
303 do_test quota-4.1.12 { | |
304 db close | |
305 quota_list | |
306 } {} | |
307 | |
308 do_test quota-4.2.1 { | |
309 sqlite3_quota_set A 1000 {} | |
310 sqlite3_quota_set B 1000 {} | |
311 sqlite3_quota_set C 1000 {} | |
312 sqlite3_quota_set D 1000 {} | |
313 quota_list | |
314 } {A B C D} | |
315 do_test quota-4.2.2 { | |
316 sqlite3_quota_set C 0 {} | |
317 sqlite3_quota_set B 0 {} | |
318 quota_list | |
319 } {A D} | |
320 do_test quota-4.2.3 { | |
321 sqlite3_quota_set A 0 {} | |
322 sqlite3_quota_set D 0 {} | |
323 quota_list | |
324 } {} | |
325 do_test quota-4.2.4 { | |
326 sqlite3_quota_set A 1000 {} | |
327 sqlite3_quota_set B 1000 {} | |
328 sqlite3_quota_set C 1000 {} | |
329 sqlite3_quota_set A 0 {} | |
330 sqlite3_quota_set B 0 {} | |
331 sqlite3_quota_set C 0 {} | |
332 quota_list | |
333 } {} | |
334 do_test quota-4.2.5 { | |
335 sqlite3_quota_set A 1000 {} | |
336 sqlite3_quota_set B 1000 {} | |
337 sqlite3_quota_set C 1000 {} | |
338 sqlite3_quota_set C 0 {} | |
339 sqlite3_quota_set B 0 {} | |
340 sqlite3_quota_set A 0 {} | |
341 quota_list | |
342 } {} | |
343 | |
344 do_test quota-4.3.1 { | |
345 sqlite3_quota_set A 1000 quota_callback | |
346 sqlite3 db A | |
347 sqlite3_quota_set A 0 quota_callback | |
348 db close | |
349 quota_list | |
350 } {} | |
351 | |
352 unset -nocomplain quotagroup | |
353 if {$tcl_platform(platform)=="windows"} { | |
354 set quotagroup *\\quota-test-A?.db | |
355 } else { | |
356 set quotagroup */quota-test-A?.db | |
357 } | |
358 foreach file [glob -nocomplain quota-test-A*] { | |
359 forcedelete $file | |
360 } | |
361 do_test quota-4.4.1 { | |
362 set ::quota {} | |
363 sqlite3_quota_set $::quotagroup 10000 quota_callback | |
364 forcedelete ./quota-test-A1.db ./quota-test-A2.db | |
365 sqlite3 db ./quota-test-A1.db | |
366 db eval { | |
367 CREATE TABLE t1(x); | |
368 INSERT INTO t1 VALUES(randomblob(5000)); | |
369 } | |
370 quota_list | |
371 } [list $quotagroup] | |
372 do_test quota-4.4.2 { | |
373 expr {$::quota==""} | |
374 } {1} | |
375 do_test quota-4.4.3 { | |
376 db close | |
377 sqlite3 db ./quota-test-A2.db | |
378 db eval { | |
379 CREATE TABLE t1(x); | |
380 INSERT INTO t1 VALUES(randomblob(5000)); | |
381 } | |
382 quota_list | |
383 } [list $quotagroup] | |
384 do_test quota-4.4.4 { | |
385 expr {$::quota!=""} | |
386 } {1} | |
387 do_test quota-4.4.5 { | |
388 db close | |
389 sqlite3_quota_set $::quotagroup 0 {} | |
390 sqlite3_quota_dump | |
391 } {} | |
392 do_test quota-4.4.6 { | |
393 sqlite3_quota_set $quotagroup 10000 quota_callback | |
394 sqlite3 db quota-test-A1.db | |
395 db eval {SELECT count(*) FROM sqlite_master} | |
396 quota_size $quotagroup | |
397 } [file size quota-test-A1.db] | |
398 do_test quota-4.4.7 { | |
399 sqlite3_quota_file quota-test-A2.db | |
400 quota_size $::quotagroup | |
401 } [expr {[file size quota-test-A1.db]+[file size quota-test-A2.db]}] | |
402 | |
403 unset -nocomplain quotagroup | |
404 if {$tcl_platform(platform)=="windows"} { | |
405 set quotagroup *\\quota-test-B* | |
406 } else { | |
407 set quotagroup */quota-test-B* | |
408 } | |
409 foreach file [glob -nocomplain quota-test-B*] { | |
410 forcedelete $file | |
411 } | |
412 do_test quota-4.5.1 { | |
413 sqlite3_quota_set $::quotagroup 100000 quota_callback | |
414 quota_size $::quotagroup | |
415 } {0} | |
416 do_test quota-4.5.2 { | |
417 sqlite3_quota_file quota-test-B1.txt | |
418 quota_size $::quotagroup | |
419 } {0} | |
420 proc add_to_file {name n} { | |
421 set out [open $name a] | |
422 fconfigure $out -translation binary | |
423 puts -nonewline $out [string repeat x $n] | |
424 close $out | |
425 } | |
426 do_test quota-4.5.3 { | |
427 add_to_file quota-test-B1.txt 123 | |
428 sqlite3_quota_file quota-test-B1.txt | |
429 quota_size $::quotagroup | |
430 } {123} | |
431 do_test quota-4.5.4 { | |
432 add_to_file quota-test-B2.txt 234 | |
433 sqlite3_quota_file quota-test-B2.txt | |
434 quota_size $::quotagroup | |
435 } {357} | |
436 do_test quota-4.5.5 { | |
437 add_to_file quota-test-B1.txt 2000 | |
438 sqlite3_quota_file quota-test-B1.txt | |
439 quota_size $::quotagroup | |
440 } {2357} | |
441 do_test quota-4.5.6 { | |
442 forcedelete quota-test-B1.txt | |
443 sqlite3_quota_file quota-test-B1.txt | |
444 quota_size $::quotagroup | |
445 } {234} | |
446 do_test quota-4.5.7 { | |
447 forcedelete quota-test-B2.txt | |
448 sqlite3_quota_file quota-test-B2.txt | |
449 quota_size $::quotagroup | |
450 } {0} | |
451 do_test quota-4.5.8 { | |
452 add_to_file quota-test-B3.txt 1234 | |
453 sqlite3_quota_file quota-test-B3.txt | |
454 quota_size $::quotagroup | |
455 } {1234} | |
456 do_test quota-4.5.9 { | |
457 sqlite3_quota_set $quotagroup 0 {} | |
458 quota_size $::quotagroup | |
459 } {0} | |
460 | |
461 do_test quota-4.9.1 { | |
462 db close | |
463 sqlite3_quota_set A 1000 quota_callback | |
464 sqlite3_quota_shutdown | |
465 } {SQLITE_OK} | |
466 do_test quota-4.9.2 { | |
467 quota_list | |
468 } {} | |
469 | |
470 #------------------------------------------------------------------------- | |
471 # The following tests test that the quota VFS handles malloc and IO | |
472 # errors. | |
473 # | |
474 | |
475 sqlite3_quota_initialize "" 1 | |
476 sqlite3_quota_set *test.db 4096 {} | |
477 | |
478 do_faultsim_test quota-5.1 -prep { | |
479 catch {db close} | |
480 } -body { | |
481 sqlite3 db test2.db | |
482 } | |
483 do_faultsim_test quota-5.2 -prep { | |
484 catch {db close} | |
485 } -body { | |
486 sqlite3 db test.db | |
487 } | |
488 | |
489 catch { db close } | |
490 forcedelete test.db | |
491 | |
492 do_test quota-5.3.prep { | |
493 sqlite3 db test.db | |
494 execsql { | |
495 PRAGMA auto_vacuum = 1; | |
496 PRAGMA page_size = 1024; | |
497 CREATE TABLE t1(a, b); | |
498 INSERT INTO t1 VALUES(10, zeroblob(1200)); | |
499 } | |
500 faultsim_save_and_close | |
501 } {} | |
502 do_faultsim_test quota-5.3 -prep { | |
503 faultsim_restore_and_reopen | |
504 } -body { | |
505 execsql { DELETE FROM t1 } | |
506 } | |
507 | |
508 do_test quota-5.4.1 { | |
509 catch { db close } | |
510 forcedelete test.db | |
511 file mkdir test.db | |
512 list [catch { sqlite3 db test.db } msg] $msg | |
513 } {1 {unable to open database file}} | |
514 | |
515 do_faultsim_test quota-5.5 -prep { | |
516 catch { sqlite3_quota_shutdown } | |
517 } -body { | |
518 sqlite3_quota_initialize "" 1 | |
519 } | |
520 | |
521 do_faultsim_test quota-5.6 -prep { | |
522 catch { sqlite3_quota_shutdown } | |
523 sqlite3_quota_initialize "" 1 | |
524 } -body { | |
525 sqlite3_quota_set * 4096 {} | |
526 } | |
527 | |
528 catch { sqlite3_quota_shutdown } | |
529 finish_test | |
OLD | NEW |