OLD | NEW |
| (Empty) |
1 # 2006 June 10 | |
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 the authorisation callback and virtual tables. | |
13 # | |
14 # $Id: vtab3.test,v 1.3 2008/07/12 14:52:21 drh Exp $ | |
15 | |
16 set testdir [file dirname $argv0] | |
17 source $testdir/tester.tcl | |
18 | |
19 ifcapable !vtab||!auth { | |
20 finish_test | |
21 return | |
22 } | |
23 | |
24 set ::auth_fail 0 | |
25 set ::auth_log [list] | |
26 set ::auth_filter [list SQLITE_READ SQLITE_UPDATE SQLITE_SELECT SQLITE_PRAGMA] | |
27 | |
28 proc auth {code arg1 arg2 arg3 arg4 args} { | |
29 if {[lsearch $::auth_filter $code]>-1} { | |
30 return SQLITE_OK | |
31 } | |
32 lappend ::auth_log $code $arg1 $arg2 $arg3 $arg4 | |
33 incr ::auth_fail -1 | |
34 if {$::auth_fail == 0} { | |
35 return SQLITE_DENY | |
36 } | |
37 return SQLITE_OK | |
38 } | |
39 | |
40 do_test vtab3-1.1 { | |
41 execsql { | |
42 CREATE TABLE elephant( | |
43 name VARCHAR(32), | |
44 color VARCHAR(16), | |
45 age INTEGER, | |
46 UNIQUE(name, color) | |
47 ); | |
48 } | |
49 } {} | |
50 | |
51 | |
52 do_test vtab3-1.2 { | |
53 register_echo_module [sqlite3_connection_pointer db] | |
54 db authorizer ::auth | |
55 execsql { | |
56 CREATE VIRTUAL TABLE pachyderm USING echo(elephant); | |
57 } | |
58 set ::auth_log | |
59 } [list \ | |
60 SQLITE_INSERT sqlite_master {} main {} \ | |
61 SQLITE_CREATE_VTABLE pachyderm echo main {} \ | |
62 ] | |
63 | |
64 do_test vtab3-1.3 { | |
65 set ::auth_log [list] | |
66 execsql { | |
67 DROP TABLE pachyderm; | |
68 } | |
69 set ::auth_log | |
70 } [list \ | |
71 SQLITE_DELETE sqlite_master {} main {} \ | |
72 SQLITE_DROP_VTABLE pachyderm echo main {} \ | |
73 SQLITE_DELETE pachyderm {} main {} \ | |
74 SQLITE_DELETE sqlite_master {} main {} \ | |
75 ] | |
76 | |
77 do_test vtab3-1.4 { | |
78 set ::auth_fail 1 | |
79 catchsql { | |
80 CREATE VIRTUAL TABLE pachyderm USING echo(elephant); | |
81 } | |
82 } {1 {not authorized}} | |
83 do_test vtab3-1.5 { | |
84 execsql { | |
85 SELECT name FROM sqlite_master WHERE type = 'table'; | |
86 } | |
87 } {elephant} | |
88 | |
89 do_test vtab3-1.5 { | |
90 set ::auth_fail 2 | |
91 catchsql { | |
92 CREATE VIRTUAL TABLE pachyderm USING echo(elephant); | |
93 } | |
94 } {1 {not authorized}} | |
95 do_test vtab3-1.6 { | |
96 execsql { | |
97 SELECT name FROM sqlite_master WHERE type = 'table'; | |
98 } | |
99 } {elephant} | |
100 | |
101 do_test vtab3-1.5 { | |
102 set ::auth_fail 3 | |
103 catchsql { | |
104 CREATE VIRTUAL TABLE pachyderm USING echo(elephant); | |
105 } | |
106 } {0 {}} | |
107 do_test vtab3-1.6 { | |
108 execsql { | |
109 SELECT name FROM sqlite_master WHERE type = 'table'; | |
110 } | |
111 } {elephant pachyderm} | |
112 | |
113 foreach i [list 1 2 3 4] { | |
114 set ::auth_fail $i | |
115 do_test vtab3-1.7.$i.1 { | |
116 set rc [catch { | |
117 execsql {DROP TABLE pachyderm;} | |
118 } msg] | |
119 if {$msg eq "authorization denied"} {set msg "not authorized"} | |
120 list $rc $msg | |
121 } {1 {not authorized}} | |
122 do_test vtab3-1.7.$i.2 { | |
123 execsql { | |
124 SELECT name FROM sqlite_master WHERE type = 'table'; | |
125 } | |
126 } {elephant pachyderm} | |
127 } | |
128 do_test vtab3-1.8.1 { | |
129 set ::auth_fail 0 | |
130 catchsql { | |
131 DROP TABLE pachyderm; | |
132 } | |
133 } {0 {}} | |
134 do_test vtab3-1.8.2 { | |
135 execsql { | |
136 SELECT name FROM sqlite_master WHERE type = 'table'; | |
137 } | |
138 } {elephant} | |
139 | |
140 finish_test | |
OLD | NEW |