sqlite/test/vtab3.test

141 lines
3.1 KiB
Plaintext
Raw Normal View History

# 2006 June 10
#
# The author disclaims copyright to this source code. In place of
# a legal notice, here is a blessing:
#
# May you do good and not evil.
# May you find forgiveness for yourself and forgive others.
# May you share freely, never taking more than you give.
#
#***********************************************************************
# This file implements regression tests for SQLite library. The
# focus of this file is the authorisation callback and virtual tables.
#
# $Id: vtab3.test,v 1.3 2008/07/12 14:52:21 drh Exp $
set testdir [file dirname $argv0]
source $testdir/tester.tcl
ifcapable !vtab||!auth {
finish_test
return
}
set ::auth_fail 0
set ::auth_log [list]
set ::auth_filter [list SQLITE_READ SQLITE_UPDATE SQLITE_SELECT SQLITE_PRAGMA]
proc auth {code arg1 arg2 arg3 arg4} {
if {[lsearch $::auth_filter $code]>-1} {
return SQLITE_OK
}
lappend ::auth_log $code $arg1 $arg2 $arg3 $arg4
incr ::auth_fail -1
if {$::auth_fail == 0} {
return SQLITE_DENY
}
return SQLITE_OK
}
do_test vtab3-1.1 {
execsql {
CREATE TABLE elephant(
name VARCHAR(32),
color VARCHAR(16),
age INTEGER,
UNIQUE(name, color)
);
}
} {}
do_test vtab3-1.2 {
register_echo_module [sqlite3_connection_pointer db]
db authorizer ::auth
execsql {
CREATE VIRTUAL TABLE pachyderm USING echo(elephant);
}
set ::auth_log
} [list \
SQLITE_INSERT sqlite_master {} main {} \
SQLITE_CREATE_VTABLE pachyderm echo main {} \
]
do_test vtab3-1.3 {
set ::auth_log [list]
execsql {
DROP TABLE pachyderm;
}
set ::auth_log
} [list \
SQLITE_DELETE sqlite_master {} main {} \
SQLITE_DROP_VTABLE pachyderm echo main {} \
SQLITE_DELETE pachyderm {} main {} \
SQLITE_DELETE sqlite_master {} main {} \
]
do_test vtab3-1.4 {
set ::auth_fail 1
catchsql {
CREATE VIRTUAL TABLE pachyderm USING echo(elephant);
}
} {1 {not authorized}}
do_test vtab3-1.5 {
execsql {
SELECT name FROM sqlite_master WHERE type = 'table';
}
} {elephant}
do_test vtab3-1.5 {
set ::auth_fail 2
catchsql {
CREATE VIRTUAL TABLE pachyderm USING echo(elephant);
}
} {1 {not authorized}}
do_test vtab3-1.6 {
execsql {
SELECT name FROM sqlite_master WHERE type = 'table';
}
} {elephant}
do_test vtab3-1.5 {
set ::auth_fail 3
catchsql {
CREATE VIRTUAL TABLE pachyderm USING echo(elephant);
}
} {0 {}}
do_test vtab3-1.6 {
execsql {
SELECT name FROM sqlite_master WHERE type = 'table';
}
} {elephant pachyderm}
foreach i [list 1 2 3 4] {
set ::auth_fail $i
do_test vtab3-1.7.$i.1 {
set rc [catch {
execsql {DROP TABLE pachyderm;}
} msg]
if {$msg eq "authorization denied"} {set msg "not authorized"}
list $rc $msg
} {1 {not authorized}}
do_test vtab3-1.7.$i.2 {
execsql {
SELECT name FROM sqlite_master WHERE type = 'table';
}
} {elephant pachyderm}
}
do_test vtab3-1.8.1 {
set ::auth_fail 0
catchsql {
DROP TABLE pachyderm;
}
} {0 {}}
do_test vtab3-1.8.2 {
execsql {
SELECT name FROM sqlite_master WHERE type = 'table';
}
} {elephant}
finish_test