| 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} { | |
| 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 |