xref: /sqlite-3.40.0/test/auth2.test (revision 5169bbc6) 
1# 2006 Aug 24
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 script is testing the sqlite3_set_authorizer() API
13# and related functionality.
14#
15# $Id: auth2.test,v 1.1 2006/08/24 14:59:46 drh Exp $
16#
17
18set testdir [file dirname $argv0]
19source $testdir/tester.tcl
20
21# disable this test if the SQLITE_OMIT_AUTHORIZATION macro is
22# defined during compilation.
23if {[catch {db auth {}} msg]} {
24  finish_test
25  return
26}
27
28do_test auth2-1.1 {
29  execsql {
30    CREATE TABLE t1(a,b,c);
31    INSERT INTO t1 VALUES(1,2,3);
32  }
33  set ::flist {}
34  proc auth {code arg1 arg2 arg3 arg4} {
35    if {$code=="SQLITE_FUNCTION"} {
36      lappend ::flist $arg2
37      if {$arg2=="max"} {
38        return SQLITE_DENY
39      } elseif {$arg2=="min"} {
40        return SQLITE_IGNORE
41      } else {
42        return SQLITE_OK
43      }
44    }
45    return SQLITE_OK
46  }
47  db authorizer ::auth
48  catchsql {SELECT max(a,b,c) FROM t1}
49} {1 {not authorized to use function: max}}
50do_test auth2-1.2 {
51  set ::flist
52} max
53do_test auth2-1.3 {
54  set ::flist {}
55  catchsql {SELECT min(a,b,c) FROM t1}
56} {0 {{}}}
57do_test auth2-1.4 {
58  set ::flist
59} min
60do_test auth2-1.5 {
61  set ::flist {}
62  catchsql {SELECT coalesce(min(a,b,c),999) FROM t1}
63} {0 999}
64do_test auth2-1.6 {
65  set ::flist
66} {coalesce min}
67do_test auth2-1.7 {
68  set ::flist {}
69  catchsql {SELECT coalesce(a,b,c) FROM t1}
70} {0 1}
71do_test auth2-1.8 {
72  set ::flist
73} coalesce
74
75finish_test
76