# 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