1045 lines
32 KiB
Ruby
1045 lines
32 KiB
Ruby
$:.unshift "#{File.dirname(__FILE__)}/../lib"
|
|
$:.unshift "#{File.dirname(__FILE__)}/../ext/sqlite3_api"
|
|
|
|
require 'test/unit'
|
|
require 'benchmark'
|
|
require 'sqlite3/database'
|
|
require 'thread'
|
|
|
|
class String
|
|
def to_utf16(terminate=false)
|
|
self.split(//).map { |c| c[0] }.pack("v*") +
|
|
(terminate ? "\0\0" : "")
|
|
end
|
|
|
|
def from_utf16
|
|
result = ""
|
|
length.times do |i|
|
|
result << self[i,1] if i % 2 == 0 && self[i] != 0
|
|
end
|
|
result
|
|
end
|
|
end
|
|
|
|
module Integration
|
|
|
|
drivers_to_test = ( ENV["SQLITE3_DRIVERS"] || "Native" ).split(',')
|
|
drivers_to_test.each do |driver|
|
|
|
|
# == TC_OpenClose =========================================================
|
|
|
|
test_case = Class.new( Test::Unit::TestCase ) do
|
|
define_method( "test_create_close" ) do
|
|
begin
|
|
db = SQLite3::Database.new( "test-create.db",
|
|
:driver => driver )
|
|
assert File.exist?( "test-create.db" )
|
|
assert_nothing_raised { db.close }
|
|
ensure
|
|
File.delete( "test-create.db" ) rescue nil
|
|
end
|
|
end
|
|
|
|
define_method( "test_open_close" ) do
|
|
begin
|
|
File.open( "test-open.db", "w" ) { |f| }
|
|
assert File.exist?( "test-open.db" )
|
|
db = SQLite3::Database.new( "test-open.db",
|
|
:driver => driver )
|
|
assert_nothing_raised { db.close }
|
|
ensure
|
|
File.delete( "test-open.db" ) rescue nil
|
|
end
|
|
end
|
|
|
|
define_method( "test_bad_open" ) do
|
|
assert_raise( SQLite3::CantOpenException ) do
|
|
SQLite3::Database.new( ".", :driver => driver )
|
|
end
|
|
end
|
|
end
|
|
const_set( "TC_OpenClose_#{driver}", test_case )
|
|
|
|
# == TC_Database ==========================================================
|
|
|
|
test_case = Class.new( Test::Unit::TestCase ) do
|
|
define_method( "setup" ) do
|
|
@db = SQLite3::Database.new( "test.db", :driver=>driver )
|
|
@db.transaction do
|
|
@db.execute "create table foo ( a integer primary key, b text )"
|
|
@db.execute "insert into foo ( b ) values ( 'foo' )"
|
|
@db.execute "insert into foo ( b ) values ( 'bar' )"
|
|
@db.execute "insert into foo ( b ) values ( 'baz' )"
|
|
end
|
|
end
|
|
|
|
define_method( "teardown" ) do
|
|
@db.close
|
|
File.delete( "test.db" )
|
|
end
|
|
|
|
define_method( "test_table_info_with_type_translation_active" ) do
|
|
@db.type_translation = true
|
|
assert_nothing_raised { @db.table_info("foo") }
|
|
end
|
|
|
|
define_method( "test_table_info_with_defaults_for_version_3_3_8_and_higher" ) do
|
|
@db.transaction do
|
|
@db.execute "create table defaults_test ( a string default NULL, b string default 'Hello' )"
|
|
data = @db.table_info( "defaults_test" )
|
|
assert_equal({"name" => "a", "type" => "string", "dflt_value" => nil, "notnull" => "0", "cid" => "0", "pk" => "0"},
|
|
data[0])
|
|
assert_equal({"name" => "b", "type" => "string", "dflt_value" => "Hello", "notnull" => "0", "cid" => "1", "pk" => "0"},
|
|
data[1])
|
|
end
|
|
end
|
|
|
|
define_method( "test_table_info_without_defaults_for_version_3_3_8_and_higher" ) do
|
|
@db.transaction do
|
|
@db.execute "create table no_defaults_test ( a integer default 1, b integer )"
|
|
data = @db.table_info( "no_defaults_test" )
|
|
assert_equal({"name" => "a", "type" => "integer", "dflt_value" => "1", "notnull" => "0", "cid" => "0", "pk" => "0"},
|
|
data[0])
|
|
assert_equal({"name" => "b", "type" => "integer", "dflt_value" => nil, "notnull" => "0", "cid" => "1", "pk" => "0"},
|
|
data[1])
|
|
end
|
|
end
|
|
|
|
define_method( "test_complete_fail" ) do
|
|
assert !@db.complete?( "select * from foo" )
|
|
end
|
|
define_method( "test_complete_success" ) do
|
|
assert @db.complete?( "select * from foo;" )
|
|
end
|
|
|
|
define_method( "test_complete_fail_utf16" ) do
|
|
assert !@db.complete?( "select * from foo".to_utf16(false), true )
|
|
end
|
|
|
|
define_method( "test_complete_success_utf16" ) do
|
|
assert @db.complete?( "select * from foo;".to_utf16(true), true )
|
|
end
|
|
|
|
define_method( "test_errmsg" ) do
|
|
assert_equal "not an error", @db.errmsg
|
|
end
|
|
|
|
define_method( "test_errmsg_utf16" ) do
|
|
assert_equal "not an error".to_utf16, @db.errmsg(true)
|
|
end
|
|
|
|
define_method( "test_errcode" ) do
|
|
assert_equal 0, @db.errcode
|
|
end
|
|
|
|
define_method( "test_trace" ) do
|
|
result = nil
|
|
@db.trace( "data" ) { |data,sql| result = [ data, sql ]; 0 }
|
|
@db.execute "select * from foo"
|
|
assert_equal ["data","select * from foo"], result
|
|
end
|
|
|
|
define_method( "test_authorizer_okay" ) do
|
|
@db.authorizer( "data" ) { |data,type,a,b,c,d| 0 }
|
|
rows = @db.execute "select * from foo"
|
|
assert_equal 3, rows.length
|
|
end
|
|
|
|
define_method( "test_authorizer_error" ) do
|
|
@db.authorizer( "data" ) { |data,type,a,b,c,d| 1 }
|
|
assert_raise( SQLite3::AuthorizationException ) do
|
|
@db.execute "select * from foo"
|
|
end
|
|
end
|
|
|
|
# FIXME: this test is failing with sqlite3 3.2.5
|
|
# define_method( "test_authorizer_silent" ) do
|
|
# @db.authorizer( "data" ) { |data,type,a,b,c,d| 2 }
|
|
# rows = @db.execute "select * from foo"
|
|
# assert rows.empty?
|
|
# end
|
|
|
|
define_method( "test_prepare_invalid_syntax" ) do
|
|
assert_raise( SQLite3::SQLException ) do
|
|
@db.prepare "select from foo"
|
|
end
|
|
end
|
|
|
|
define_method( "test_prepare_invalid_column" ) do
|
|
assert_raise( SQLite3::SQLException ) do
|
|
@db.prepare "select k from foo"
|
|
end
|
|
end
|
|
|
|
define_method( "test_prepare_invalid_table" ) do
|
|
assert_raise( SQLite3::SQLException ) do
|
|
@db.prepare "select * from barf"
|
|
end
|
|
end
|
|
|
|
define_method( "test_prepare_no_block" ) do
|
|
stmt = @db.prepare "select * from foo"
|
|
assert stmt.respond_to?(:execute)
|
|
stmt.close
|
|
end
|
|
|
|
define_method( "test_prepare_with_block" ) do
|
|
called = false
|
|
@db.prepare "select * from foo" do |stmt|
|
|
called = true
|
|
assert stmt.respond_to?(:execute)
|
|
end
|
|
assert called
|
|
end
|
|
|
|
define_method( "test_execute_no_block_no_bind_no_match" ) do
|
|
rows = @db.execute( "select * from foo where a > 100" )
|
|
assert rows.empty?
|
|
end
|
|
|
|
define_method( "test_execute_with_block_no_bind_no_match" ) do
|
|
called = false
|
|
@db.execute( "select * from foo where a > 100" ) do |row|
|
|
called = true
|
|
end
|
|
assert !called
|
|
end
|
|
|
|
define_method( "test_execute_no_block_with_bind_no_match" ) do
|
|
rows = @db.execute( "select * from foo where a > ?", 100 )
|
|
assert rows.empty?
|
|
end
|
|
|
|
define_method( "test_execute_with_block_with_bind_no_match" ) do
|
|
called = false
|
|
@db.execute( "select * from foo where a > ?", 100 ) do |row|
|
|
called = true
|
|
end
|
|
assert !called
|
|
end
|
|
|
|
define_method( "test_execute_no_block_no_bind_with_match" ) do
|
|
rows = @db.execute( "select * from foo where a = 1" )
|
|
assert_equal 1, rows.length
|
|
end
|
|
|
|
define_method( "test_execute_with_block_no_bind_with_match" ) do
|
|
called = 0
|
|
@db.execute( "select * from foo where a = 1" ) do |row|
|
|
called += 1
|
|
end
|
|
assert_equal 1, called
|
|
end
|
|
|
|
define_method( "test_execute_no_block_with_bind_with_match" ) do
|
|
rows = @db.execute( "select * from foo where a = ?", 1 )
|
|
assert_equal 1, rows.length
|
|
end
|
|
|
|
define_method( "test_execute_with_block_with_bind_with_match" ) do
|
|
called = 0
|
|
@db.execute( "select * from foo where a = ?", 1 ) do |row|
|
|
called += 1
|
|
end
|
|
assert_equal 1, called
|
|
end
|
|
|
|
define_method( "test_execute2_no_block_no_bind_no_match" ) do
|
|
columns, *rows = @db.execute2( "select * from foo where a > 100" )
|
|
assert rows.empty?
|
|
assert [ "a", "b" ], columns
|
|
end
|
|
|
|
define_method( "test_execute2_with_block_no_bind_no_match" ) do
|
|
called = 0
|
|
@db.execute2( "select * from foo where a > 100" ) do |row|
|
|
assert [ "a", "b" ], row unless called == 0
|
|
called += 1
|
|
end
|
|
assert_equal 1, called
|
|
end
|
|
|
|
define_method( "test_execute2_no_block_with_bind_no_match" ) do
|
|
columns, *rows = @db.execute2( "select * from foo where a > ?", 100 )
|
|
assert rows.empty?
|
|
assert [ "a", "b" ], columns
|
|
end
|
|
|
|
define_method( "test_execute2_with_block_with_bind_no_match" ) do
|
|
called = 0
|
|
@db.execute2( "select * from foo where a > ?", 100 ) do |row|
|
|
assert [ "a", "b" ], row unless called == 0
|
|
called += 1
|
|
end
|
|
assert_equal 1, called
|
|
end
|
|
|
|
define_method( "test_execute2_no_block_no_bind_with_match" ) do
|
|
columns, *rows = @db.execute2( "select * from foo where a = 1" )
|
|
assert_equal 1, rows.length
|
|
assert [ "a", "b" ], columns
|
|
end
|
|
|
|
define_method( "test_execute2_with_block_no_bind_with_match" ) do
|
|
called = 0
|
|
@db.execute2( "select * from foo where a = 1" ) do |row|
|
|
assert [ "a", "b" ], row unless called == 0
|
|
called += 1
|
|
end
|
|
assert_equal 2, called
|
|
end
|
|
|
|
define_method( "test_execute2_no_block_with_bind_with_match" ) do
|
|
columns, *rows = @db.execute2( "select * from foo where a = ?", 1 )
|
|
assert_equal 1, rows.length
|
|
assert [ "a", "b" ], columns
|
|
end
|
|
|
|
define_method( "test_execute2_with_block_with_bind_with_match" ) do
|
|
called = 0
|
|
@db.execute2( "select * from foo where a = ?", 1 ) do |row|
|
|
called += 1
|
|
end
|
|
assert_equal 2, called
|
|
end
|
|
|
|
define_method( "test_execute_batch_empty" ) do
|
|
assert_nothing_raised { @db.execute_batch "" }
|
|
end
|
|
|
|
define_method( "test_execute_batch_no_bind" ) do
|
|
@db.transaction do
|
|
@db.execute_batch <<-SQL
|
|
create table bar ( a, b, c );
|
|
insert into bar values ( 'one', 2, 'three' );
|
|
insert into bar values ( 'four', 5, 'six' );
|
|
insert into bar values ( 'seven', 8, 'nine' );
|
|
SQL
|
|
end
|
|
rows = @db.execute( "select * from bar" )
|
|
assert_equal 3, rows.length
|
|
end
|
|
|
|
define_method( "test_execute_batch_with_bind" ) do
|
|
@db.execute_batch( <<-SQL, 1 )
|
|
create table bar ( a, b, c );
|
|
insert into bar values ( 'one', 2, ? );
|
|
insert into bar values ( 'four', 5, ? );
|
|
insert into bar values ( 'seven', 8, ? );
|
|
SQL
|
|
rows = @db.execute( "select * from bar" ).map { |a,b,c| c }
|
|
assert_equal %w{1 1 1}, rows
|
|
end
|
|
|
|
define_method( "test_query_no_block_no_bind_no_match" ) do
|
|
result = @db.query( "select * from foo where a > 100" )
|
|
assert_nil result.next
|
|
result.close
|
|
end
|
|
|
|
define_method( "test_query_with_block_no_bind_no_match" ) do
|
|
r = nil
|
|
@db.query( "select * from foo where a > 100" ) do |result|
|
|
assert_nil result.next
|
|
r = result
|
|
end
|
|
assert r.closed?
|
|
end
|
|
|
|
define_method( "test_query_no_block_with_bind_no_match" ) do
|
|
result = @db.query( "select * from foo where a > ?", 100 )
|
|
assert_nil result.next
|
|
result.close
|
|
end
|
|
|
|
define_method( "test_query_with_block_with_bind_no_match" ) do
|
|
r = nil
|
|
@db.query( "select * from foo where a > ?", 100 ) do |result|
|
|
assert_nil result.next
|
|
r = result
|
|
end
|
|
assert r.closed?
|
|
end
|
|
|
|
define_method( "test_query_no_block_no_bind_with_match" ) do
|
|
result = @db.query( "select * from foo where a = 1" )
|
|
assert_not_nil result.next
|
|
assert_nil result.next
|
|
result.close
|
|
end
|
|
|
|
define_method( "test_query_with_block_no_bind_with_match" ) do
|
|
r = nil
|
|
@db.query( "select * from foo where a = 1" ) do |result|
|
|
assert_not_nil result.next
|
|
assert_nil result.next
|
|
r = result
|
|
end
|
|
assert r.closed?
|
|
end
|
|
|
|
define_method( "test_query_no_block_with_bind_with_match" ) do
|
|
result = @db.query( "select * from foo where a = ?", 1 )
|
|
assert_not_nil result.next
|
|
assert_nil result.next
|
|
result.close
|
|
end
|
|
|
|
define_method( "test_query_with_block_with_bind_with_match" ) do
|
|
r = nil
|
|
@db.query( "select * from foo where a = ?", 1 ) do |result|
|
|
assert_not_nil result.next
|
|
assert_nil result.next
|
|
r = result
|
|
end
|
|
assert r.closed?
|
|
end
|
|
|
|
define_method( "test_get_first_row_no_bind_no_match" ) do
|
|
result = @db.get_first_row( "select * from foo where a=100" )
|
|
assert_nil result
|
|
end
|
|
|
|
define_method( "test_get_first_row_no_bind_with_match" ) do
|
|
result = @db.get_first_row( "select * from foo where a=1" )
|
|
assert_equal [ "1", "foo" ], result
|
|
end
|
|
|
|
define_method( "test_get_first_row_with_bind_no_match" ) do
|
|
result = @db.get_first_row( "select * from foo where a=?", 100 )
|
|
assert_nil result
|
|
end
|
|
|
|
define_method( "test_get_first_row_with_bind_with_match" ) do
|
|
result = @db.get_first_row( "select * from foo where a=?", 1 )
|
|
assert_equal [ "1", "foo" ], result
|
|
end
|
|
|
|
define_method( "test_get_first_value_no_bind_no_match" ) do
|
|
result = @db.get_first_value( "select b, a from foo where a=100" )
|
|
assert_nil result
|
|
end
|
|
|
|
define_method( "test_get_first_value_no_bind_with_match" ) do
|
|
result = @db.get_first_value( "select b, a from foo where a=1" )
|
|
assert_equal "foo", result
|
|
end
|
|
|
|
define_method( "test_get_first_value_with_bind_no_match" ) do
|
|
result = @db.get_first_value( "select b, a from foo where a=?", 100 )
|
|
assert_nil result
|
|
end
|
|
|
|
define_method( "test_get_first_value_with_bind_with_match" ) do
|
|
result = @db.get_first_value( "select b, a from foo where a=?", 1 )
|
|
assert_equal "foo", result
|
|
end
|
|
|
|
define_method( "test_last_insert_row_id" ) do
|
|
@db.execute "insert into foo ( b ) values ( 'test' )"
|
|
assert_equal 4, @db.last_insert_row_id
|
|
@db.execute "insert into foo ( b ) values ( 'again' )"
|
|
assert_equal 5, @db.last_insert_row_id
|
|
end
|
|
|
|
define_method( "test_changes" ) do
|
|
@db.execute "insert into foo ( b ) values ( 'test' )"
|
|
assert_equal 1, @db.changes
|
|
@db.execute "delete from foo where 1=1"
|
|
assert_equal 4, @db.changes
|
|
end
|
|
|
|
define_method( "test_total_changes" ) do
|
|
assert_equal 3, @db.total_changes
|
|
@db.execute "insert into foo ( b ) values ( 'test' )"
|
|
@db.execute "delete from foo where 1=1"
|
|
assert_equal 8, @db.total_changes
|
|
end
|
|
|
|
define_method( "test_transaction_nest" ) do
|
|
assert_raise( SQLite3::SQLException ) do
|
|
@db.transaction do
|
|
@db.transaction do
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
define_method( "test_transaction_rollback" ) do
|
|
@db.transaction
|
|
@db.execute_batch <<-SQL
|
|
insert into foo (b) values ( 'test1' );
|
|
insert into foo (b) values ( 'test2' );
|
|
insert into foo (b) values ( 'test3' );
|
|
insert into foo (b) values ( 'test4' );
|
|
SQL
|
|
assert_equal 7, @db.get_first_value("select count(*) from foo").to_i
|
|
@db.rollback
|
|
assert_equal 3, @db.get_first_value("select count(*) from foo").to_i
|
|
end
|
|
|
|
define_method( "test_transaction_commit" ) do
|
|
@db.transaction
|
|
@db.execute_batch <<-SQL
|
|
insert into foo (b) values ( 'test1' );
|
|
insert into foo (b) values ( 'test2' );
|
|
insert into foo (b) values ( 'test3' );
|
|
insert into foo (b) values ( 'test4' );
|
|
SQL
|
|
assert_equal 7, @db.get_first_value("select count(*) from foo").to_i
|
|
@db.commit
|
|
assert_equal 7, @db.get_first_value("select count(*) from foo").to_i
|
|
end
|
|
|
|
define_method( "test_transaction_rollback_in_block" ) do
|
|
assert_raise( SQLite3::SQLException ) do
|
|
@db.transaction do
|
|
@db.rollback
|
|
end
|
|
end
|
|
end
|
|
|
|
define_method( "test_transaction_commit_in_block" ) do
|
|
assert_raise( SQLite3::SQLException ) do
|
|
@db.transaction do
|
|
@db.commit
|
|
end
|
|
end
|
|
end
|
|
|
|
define_method( "test_transaction_active" ) do
|
|
assert !@db.transaction_active?
|
|
@db.transaction
|
|
assert @db.transaction_active?
|
|
@db.commit
|
|
assert !@db.transaction_active?
|
|
end
|
|
|
|
define_method( "no_tests_at" ) do |file,line,method|
|
|
warn "[(#{self.class}):#{file}:#{line}] no tests for #{method}"
|
|
end
|
|
|
|
define_method( "test_interrupt" ) do
|
|
@db.create_function( "abort", 1 ) do |func,x|
|
|
@db.interrupt
|
|
func.result = x
|
|
end
|
|
|
|
assert_raise( SQLite3::SQLException ) do
|
|
@db.execute "select abort(a) from foo"
|
|
end
|
|
end
|
|
|
|
define_method( "test_busy_handler_outwait" ) do
|
|
busy = Mutex.new
|
|
busy.lock
|
|
handler_call_count = 0
|
|
|
|
t = Thread.new do
|
|
begin
|
|
db2 = SQLite3::Database.open( "test.db", :driver=>driver )
|
|
db2.transaction( :exclusive ) do
|
|
busy.lock
|
|
end
|
|
ensure
|
|
db2.close if db2
|
|
end
|
|
end
|
|
|
|
@db.busy_handler do |data,count|
|
|
handler_call_count += 1
|
|
busy.unlock
|
|
true
|
|
end
|
|
|
|
assert_nothing_raised do
|
|
@db.execute "insert into foo (b) values ( 'from 2' )"
|
|
end
|
|
|
|
t.join
|
|
|
|
assert_equal 1, handler_call_count
|
|
end
|
|
|
|
define_method( "test_busy_handler_impatient" ) do
|
|
busy = Mutex.new
|
|
busy.lock
|
|
handler_call_count = 0
|
|
|
|
t = Thread.new do
|
|
begin
|
|
db2 = SQLite3::Database.open( "test.db", :driver=>driver )
|
|
db2.transaction( :exclusive ) do
|
|
busy.lock
|
|
end
|
|
ensure
|
|
db2.close if db2
|
|
end
|
|
end
|
|
|
|
@db.busy_handler do |data, count|
|
|
handler_call_count += 1
|
|
false
|
|
end
|
|
|
|
assert_raise( SQLite3::BusyException ) do
|
|
@db.execute "insert into foo (b) values ( 'from 2' )"
|
|
end
|
|
|
|
busy.unlock
|
|
t.join
|
|
|
|
assert_equal 1, handler_call_count
|
|
end
|
|
|
|
define_method( "test_busy_timeout" ) do
|
|
@db.busy_timeout 1000
|
|
busy = Mutex.new
|
|
busy.lock
|
|
|
|
t = Thread.new do
|
|
begin
|
|
db2 = SQLite3::Database.open( "test.db", :driver=>driver )
|
|
db2.transaction( :exclusive ) do
|
|
busy.lock
|
|
end
|
|
ensure
|
|
db2.close if db2
|
|
end
|
|
end
|
|
|
|
time = Benchmark.measure do
|
|
assert_raise( SQLite3::BusyException ) do
|
|
@db.execute "insert into foo (b) values ( 'from 2' )"
|
|
end
|
|
end
|
|
|
|
busy.unlock
|
|
t.join
|
|
|
|
assert time.real*1000 >= 1000
|
|
end
|
|
|
|
define_method( "test_create_function" ) do
|
|
@db.create_function( "munge", 1 ) do |func,x|
|
|
func.result = ">>>#{x}<<<"
|
|
end
|
|
|
|
value = @db.get_first_value( "select munge(b) from foo where a=1" )
|
|
assert_match( />>>.*<<</, value )
|
|
end
|
|
|
|
define_method( "test_create_aggregate_without_block" ) do
|
|
step = proc do |ctx,a|
|
|
ctx[:sum] ||= 0
|
|
ctx[:sum] += a.to_i
|
|
end
|
|
|
|
final = proc { |ctx| ctx.result = ctx[:sum] }
|
|
|
|
@db.create_aggregate( "accumulate", 1, step, final )
|
|
|
|
value = @db.get_first_value( "select accumulate(a) from foo" )
|
|
assert_equal "6", value
|
|
end
|
|
|
|
define_method( "test_create_aggregate_with_block" ) do
|
|
@db.create_aggregate( "accumulate", 1 ) do
|
|
step do |ctx,a|
|
|
ctx[:sum] ||= 0
|
|
ctx[:sum] += a.to_i
|
|
end
|
|
|
|
finalize { |ctx| ctx.result = ctx[:sum] }
|
|
end
|
|
|
|
value = @db.get_first_value( "select accumulate(a) from foo" )
|
|
assert_equal "6", value
|
|
end
|
|
|
|
define_method( "test_create_aggregate_with_no_data" ) do
|
|
@db.create_aggregate( "accumulate", 1 ) do
|
|
step do |ctx,a|
|
|
ctx[:sum] ||= 0
|
|
ctx[:sum] += a.to_i
|
|
end
|
|
|
|
finalize { |ctx| ctx.result = ctx[:sum] || 0 }
|
|
end
|
|
|
|
value = @db.get_first_value(
|
|
"select accumulate(a) from foo where a = 100" )
|
|
assert_equal "0", value
|
|
end
|
|
|
|
define_method( "test_create_aggregate_handler" ) do
|
|
handler = Class.new do
|
|
class << self
|
|
define_method( "arity" ) { 1 }
|
|
define_method( "text_rep" ) { SQLite3::Constants::TextRep::ANY }
|
|
define_method( "name" ) { "multiply" }
|
|
end
|
|
define_method( "step" ) do |ctx,a|
|
|
ctx[:buffer] ||= 1
|
|
ctx[:buffer] *= a.to_i
|
|
end
|
|
define_method( "finalize" ) { |ctx| ctx.result = ctx[:buffer] }
|
|
end
|
|
|
|
@db.create_aggregate_handler( handler )
|
|
value = @db.get_first_value( "select multiply(a) from foo" )
|
|
assert_equal "6", value
|
|
end
|
|
|
|
define_method( "test_bind_array_parameter" ) do
|
|
result = @db.get_first_value( "select b from foo where a=? and b=?",
|
|
[ 1, "foo" ] )
|
|
assert_equal "foo", result
|
|
end
|
|
end
|
|
const_set( "TC_Database_#{driver}", test_case )
|
|
|
|
# == TC_Statement =========================================================
|
|
|
|
test_case = Class.new( Test::Unit::TestCase ) do
|
|
define_method( "setup" ) do
|
|
@db = SQLite3::Database.new( "test.db", :driver=>driver )
|
|
@db.transaction do
|
|
@db.execute "create table foo ( a integer primary key, b text )"
|
|
@db.execute "insert into foo ( b ) values ( 'foo' )"
|
|
@db.execute "insert into foo ( b ) values ( 'bar' )"
|
|
@db.execute "insert into foo ( b ) values ( 'baz' )"
|
|
end
|
|
@stmt = @db.prepare( "select * from foo where a in ( ?, :named )" )
|
|
end
|
|
|
|
define_method( "teardown" ) do
|
|
@stmt.close
|
|
@db.close
|
|
File.delete( "test.db" )
|
|
end
|
|
|
|
define_method( "test_remainder_empty" ) do
|
|
assert_equal "", @stmt.remainder
|
|
end
|
|
|
|
define_method( "test_remainder_nonempty" ) do
|
|
called = false
|
|
@db.prepare( "select * from foo;\n blah" ) do |stmt|
|
|
called = true
|
|
assert_equal "\n blah", stmt.remainder
|
|
end
|
|
assert called
|
|
end
|
|
|
|
define_method( "test_bind_params_empty" ) do
|
|
assert_nothing_raised { @stmt.bind_params }
|
|
assert @stmt.execute!.empty?
|
|
end
|
|
|
|
define_method( "test_bind_params_array" ) do
|
|
@stmt.bind_params 1, 2
|
|
assert_equal 2, @stmt.execute!.length
|
|
end
|
|
|
|
define_method( "test_bind_params_hash" ) do
|
|
@stmt.bind_params ":named" => 2
|
|
assert_equal 1, @stmt.execute!.length
|
|
end
|
|
|
|
define_method( "test_bind_params_hash_without_colon" ) do
|
|
@stmt.bind_params "named" => 2
|
|
assert_equal 1, @stmt.execute!.length
|
|
end
|
|
|
|
define_method( "test_bind_params_hash_as_symbol" ) do
|
|
@stmt.bind_params :named => 2
|
|
assert_equal 1, @stmt.execute!.length
|
|
end
|
|
|
|
define_method( "test_bind_params_mixed" ) do
|
|
@stmt.bind_params( 1, ":named" => 2 )
|
|
assert_equal 2, @stmt.execute!.length
|
|
end
|
|
|
|
define_method( "test_bind_param_by_index" ) do
|
|
@stmt.bind_params( 1, 2 )
|
|
assert_equal 2, @stmt.execute!.length
|
|
end
|
|
|
|
define_method( "test_bind_param_by_name_bad" ) do
|
|
assert_raise( SQLite3::Exception ) { @stmt.bind_param( "@named", 2 ) }
|
|
end
|
|
|
|
define_method( "test_bind_param_by_name_good" ) do
|
|
@stmt.bind_param( ":named", 2 )
|
|
assert_equal 1, @stmt.execute!.length
|
|
end
|
|
|
|
define_method( "test_bind_param_with_various_types" ) do
|
|
@db.transaction do
|
|
@db.execute "create table all_types ( a integer primary key, b float, c string, d integer )"
|
|
@db.execute "insert into all_types ( b, c, d ) values ( 1.4, 'hello', 68719476735 )"
|
|
end
|
|
|
|
assert_equal 1, @db.execute( "select * from all_types where b = ?", 1.4 ).length
|
|
assert_equal 1, @db.execute( "select * from all_types where c = ?", 'hello').length
|
|
assert_equal 1, @db.execute( "select * from all_types where d = ?", 68719476735).length
|
|
end
|
|
|
|
define_method( "test_execute_no_bind_no_block" ) do
|
|
assert_instance_of SQLite3::ResultSet, @stmt.execute
|
|
end
|
|
|
|
define_method( "test_execute_with_bind_no_block" ) do
|
|
assert_instance_of SQLite3::ResultSet, @stmt.execute( 1, 2 )
|
|
end
|
|
|
|
define_method( "test_execute_no_bind_with_block" ) do
|
|
called = false
|
|
@stmt.execute { |row| called = true }
|
|
assert called
|
|
end
|
|
|
|
define_method( "test_execute_with_bind_with_block" ) do
|
|
called = 0
|
|
@stmt.execute( 1, 2 ) { |row| called += 1 }
|
|
assert_equal 1, called
|
|
end
|
|
|
|
define_method( "test_reexecute" ) do
|
|
r = @stmt.execute( 1, 2 )
|
|
assert_equal 2, r.to_a.length
|
|
assert_nothing_raised { r = @stmt.execute( 1, 2 ) }
|
|
assert_equal 2, r.to_a.length
|
|
end
|
|
|
|
define_method( "test_execute_bang_no_bind_no_block" ) do
|
|
assert @stmt.execute!.empty?
|
|
end
|
|
|
|
define_method( "test_execute_bang_with_bind_no_block" ) do
|
|
assert_equal 2, @stmt.execute!( 1, 2 ).length
|
|
end
|
|
|
|
define_method( "test_execute_bang_no_bind_with_block" ) do
|
|
called = 0
|
|
@stmt.execute! { |row| called += 1 }
|
|
assert_equal 0, called
|
|
end
|
|
|
|
define_method( "test_execute_bang_with_bind_with_block" ) do
|
|
called = 0
|
|
@stmt.execute!( 1, 2 ) { |row| called += 1 }
|
|
assert_equal 2, called
|
|
end
|
|
|
|
define_method( "test_columns" ) do
|
|
c1 = @stmt.columns
|
|
c2 = @stmt.columns
|
|
assert_same c1, c2
|
|
assert_equal 2, c1.length
|
|
end
|
|
|
|
define_method( "test_columns_computed" ) do
|
|
called = false
|
|
@db.prepare( "select count(*) from foo" ) do |stmt|
|
|
called = true
|
|
assert_equal [ "count(*)" ], stmt.columns
|
|
end
|
|
assert called
|
|
end
|
|
|
|
define_method( "test_types" ) do
|
|
t1 = @stmt.types
|
|
t2 = @stmt.types
|
|
assert_same t1, t2
|
|
assert_equal 2, t1.length
|
|
end
|
|
|
|
define_method( "test_types_computed" ) do
|
|
called = false
|
|
@db.prepare( "select count(*) from foo" ) do |stmt|
|
|
called = true
|
|
assert_equal [ nil ], stmt.types
|
|
end
|
|
assert called
|
|
end
|
|
|
|
define_method( "test_close" ) do
|
|
stmt = @db.prepare( "select * from foo" )
|
|
assert !stmt.closed?
|
|
stmt.close
|
|
assert stmt.closed?
|
|
assert_raise( SQLite3::Exception ) { stmt.execute }
|
|
assert_raise( SQLite3::Exception ) { stmt.execute! }
|
|
assert_raise( SQLite3::Exception ) { stmt.close }
|
|
assert_raise( SQLite3::Exception ) { stmt.bind_params 5 }
|
|
assert_raise( SQLite3::Exception ) { stmt.bind_param 1, 5 }
|
|
assert_raise( SQLite3::Exception ) { stmt.columns }
|
|
assert_raise( SQLite3::Exception ) { stmt.types }
|
|
end
|
|
|
|
define_method( "test_committing_tx_with_statement_active" ) do
|
|
called = false
|
|
@db.prepare( "select count(*) from foo" ) do |stmt|
|
|
called = true
|
|
count = stmt.execute!.first.first.to_i
|
|
@db.transaction do
|
|
@db.execute "insert into foo ( b ) values ( 'hello' )"
|
|
end
|
|
new_count = stmt.execute!.first.first.to_i
|
|
assert_equal new_count, count+1
|
|
end
|
|
assert called
|
|
end
|
|
end
|
|
const_set( "TC_Statement_#{driver}", test_case )
|
|
|
|
# == TC_ResultSet =========================================================
|
|
|
|
test_case = Class.new( Test::Unit::TestCase ) do
|
|
define_method( "setup" ) do
|
|
@db = SQLite3::Database.new( "test.db", :driver=>driver )
|
|
@db.transaction do
|
|
@db.execute "create table foo ( a integer primary key, b text )"
|
|
@db.execute "insert into foo ( b ) values ( 'foo' )"
|
|
@db.execute "insert into foo ( b ) values ( 'bar' )"
|
|
@db.execute "insert into foo ( b ) values ( 'baz' )"
|
|
end
|
|
@stmt = @db.prepare( "select * from foo where a in ( ?, ? )" )
|
|
@result = @stmt.execute
|
|
end
|
|
|
|
define_method( "teardown" ) do
|
|
@stmt.close
|
|
@db.close
|
|
File.delete( "test.db" )
|
|
end
|
|
|
|
define_method( "test_reset_unused" ) do
|
|
assert_nothing_raised { @result.reset }
|
|
assert @result.to_a.empty?
|
|
end
|
|
|
|
define_method( "test_reset_used" ) do
|
|
@result.to_a
|
|
assert_nothing_raised { @result.reset }
|
|
assert @result.to_a.empty?
|
|
end
|
|
|
|
define_method( "test_reset_with_bind" ) do
|
|
@result.to_a
|
|
assert_nothing_raised { @result.reset( 1, 2 ) }
|
|
assert_equal 2, @result.to_a.length
|
|
end
|
|
|
|
define_method( "test_eof_inner" ) do
|
|
@result.reset( 1 )
|
|
assert !@result.eof?
|
|
end
|
|
|
|
define_method( "test_eof_edge" ) do
|
|
@result.reset( 1 )
|
|
@result.next # to first row
|
|
@result.next # to end of result set
|
|
assert @result.eof?
|
|
end
|
|
|
|
define_method( "test_next_eof" ) do
|
|
@result.reset( 1 )
|
|
assert_not_nil @result.next
|
|
assert_nil @result.next
|
|
end
|
|
|
|
define_method( "test_next_no_type_translation_no_hash" ) do
|
|
@result.reset( 1 )
|
|
assert_equal [ "1", "foo" ], @result.next
|
|
end
|
|
|
|
define_method( "test_next_type_translation" ) do
|
|
@db.type_translation = true
|
|
@result.reset( 1 )
|
|
assert_equal [ 1, "foo" ], @result.next
|
|
end
|
|
|
|
define_method( "test_next_type_translation_with_untyped_column" ) do
|
|
@db.type_translation = true
|
|
@db.query( "select count(*) from foo" ) do |result|
|
|
assert_equal ["3"], result.next
|
|
end
|
|
end
|
|
|
|
define_method( "test_type_translation_with_null_column" ) do
|
|
@db.type_translation = true
|
|
@db.execute "create table bar ( a integer, b time, c string )"
|
|
@db.execute "insert into bar (a, b, c) values (NULL, '1974-07-25 14:39:00', 'hello')"
|
|
@db.execute "insert into bar (a, b, c) values (1, NULL, 'hello')"
|
|
@db.execute "insert into bar (a, b, c) values (2, '1974-07-25 14:39:00', NULL)"
|
|
@db.query( "select * from bar" ) do |result|
|
|
assert_equal [nil, Time.local(1974, 7, 25, 14, 39, 0), 'hello'], result.next
|
|
assert_equal [1, nil, 'hello'], result.next
|
|
assert_equal [2, Time.local(1974, 7, 25, 14, 39, 0), nil], result.next
|
|
end
|
|
end
|
|
|
|
define_method( "test_date_and_time_translation" ) do
|
|
@db.type_translation = true
|
|
@db.execute "create table bar ( a date, b datetime, c time, d timestamp )"
|
|
@db.execute "insert into bar (a, b, c, d) values ('1999-01-08', '1997-12-17 07:37:16', '07:37:16', '2004-10-19 10:23:54')"
|
|
@db.query( "select * from bar" ) do |result|
|
|
result = result.next
|
|
assert result[0].is_a?(Date)
|
|
assert result[1].is_a?(DateTime)
|
|
assert result[2].is_a?(Time)
|
|
assert result[3].is_a?(Time)
|
|
end
|
|
end
|
|
|
|
define_method( "test_next_results_as_hash" ) do
|
|
@db.results_as_hash = true
|
|
@result.reset( 1 )
|
|
assert_equal( { "a" => "1", "b" => "foo", 0 => "1", 1 => "foo" },
|
|
@result.next )
|
|
end
|
|
|
|
define_method( "test_each" ) do
|
|
called = 0
|
|
@result.reset( 1, 2 )
|
|
@result.each { |row| called += 1 }
|
|
assert_equal 2, called
|
|
end
|
|
|
|
define_method( "test_enumerable" ) do
|
|
@result.reset( 1, 2 )
|
|
assert_equal 2, @result.to_a.length
|
|
end
|
|
|
|
define_method( "test_types" ) do
|
|
assert_equal [ "integer", "text" ], @result.types
|
|
end
|
|
|
|
define_method( "test_columns" ) do
|
|
assert_equal [ "a", "b" ], @result.columns
|
|
end
|
|
|
|
define_method( "test_close" ) do
|
|
stmt = @db.prepare( "select * from foo" )
|
|
result = stmt.execute
|
|
assert !result.closed?
|
|
result.close
|
|
assert result.closed?
|
|
assert stmt.closed?
|
|
assert_raise( SQLite3::Exception ) { result.reset }
|
|
assert_raise( SQLite3::Exception ) { result.next }
|
|
assert_raise( SQLite3::Exception ) { result.each }
|
|
assert_raise( SQLite3::Exception ) { result.close }
|
|
assert_raise( SQLite3::Exception ) { result.types }
|
|
assert_raise( SQLite3::Exception ) { result.columns }
|
|
end
|
|
end
|
|
const_set( "TC_ResultSet_#{driver}", test_case )
|
|
end
|
|
|
|
end
|