shard-crystal-sqlite3/spec/db_spec.cr

132 lines
4.1 KiB
Crystal
Raw Normal View History

require "./spec_helper"
require "db/spec"
private class NotSupportedType
end
private def cast_if_blob(expr, sql_type)
case sql_type
when "blob"
"cast(#{expr} as blob)"
else
expr
end
end
DB::DriverSpecs(DB::Any).run do
support_unprepared false
before do
File.delete(DB_FILENAME) if File.exists?(DB_FILENAME)
end
after do
File.delete(DB_FILENAME) if File.exists?(DB_FILENAME)
end
connection_string "sqlite3:#{DB_FILENAME}"
# ? can use many ... (:memory:)
sample_value true, "int", "1", type_safe_value: false
sample_value false, "int", "0", type_safe_value: false
sample_value 2, "int", "2", type_safe_value: false
sample_value 1_i64, "int", "1"
sample_value "hello", "text", "'hello'"
sample_value 1.5_f32, "float", "1.5", type_safe_value: false
sample_value 1.5, "float", "1.5"
sample_value Time.utc(2016, 2, 15), "text", "'2016-02-15 00:00:00.000'", type_safe_value: false
sample_value Time.utc(2016, 2, 15, 10, 15, 30), "text", "'2016-02-15 10:15:30.000'", type_safe_value: false
sample_value Time.local(2016, 2, 15, 7, 15, 30, location: Time::Location.fixed("fixed", -3*3600)), "text", "'2016-02-15 10:15:30.000'", type_safe_value: false
ary = UInt8[0x53, 0x51, 0x4C, 0x69, 0x74, 0x65]
sample_value Bytes.new(ary.to_unsafe, ary.size), "blob", "X'53514C697465'" # , type_safe_value: false
binding_syntax do |index|
"?"
end
create_table_1column_syntax do |table_name, col1|
"create table #{table_name} (#{col1.name} #{col1.sql_type} #{col1.null ? "NULL" : "NOT NULL"})"
end
create_table_2columns_syntax do |table_name, col1, col2|
"create table #{table_name} (#{col1.name} #{col1.sql_type} #{col1.null ? "NULL" : "NOT NULL"}, #{col2.name} #{col2.sql_type} #{col2.null ? "NULL" : "NOT NULL"})"
end
select_1column_syntax do |table_name, col1|
"select #{cast_if_blob(col1.name, col1.sql_type)} from #{table_name}"
end
select_2columns_syntax do |table_name, col1, col2|
"select #{cast_if_blob(col1.name, col1.sql_type)}, #{cast_if_blob(col2.name, col2.sql_type)} from #{table_name}"
end
select_count_syntax do |table_name|
"select count(*) from #{table_name}"
end
select_scalar_syntax do |expression, sql_type|
"select #{cast_if_blob(expression, sql_type)}"
end
insert_1column_syntax do |table_name, col, expression|
"insert into #{table_name} (#{col.name}) values (#{expression})"
end
insert_2columns_syntax do |table_name, col1, expr1, col2, expr2|
"insert into #{table_name} (#{col1.name}, #{col2.name}) values (#{expr1}, #{expr2})"
end
drop_table_if_exists_syntax do |table_name|
"drop table if exists #{table_name}"
end
it "gets last insert row id", prepared: :both do |db|
db.exec "create table person (name text, age integer)"
db.exec %(insert into person values ("foo", 10))
res = db.exec %(insert into person values ("foo", 10))
res.last_insert_id.should eq(2)
res.rows_affected.should eq(1)
end
# TODO timestamp support
it "raises on unsupported param types" do |db|
expect_raises Exception, "SQLite3::Statement does not support NotSupportedType params" do
db.query "select ?", NotSupportedType.new
end
# TODO raising exception does not close the connection and pool is exhausted
end
it "ensures statements are closed" do |db|
db.exec %(create table if not exists a (i int not null, str text not null);)
db.exec %(insert into a (i, str) values (23, "bai bai");)
2.times do |i|
DB.open db.uri do |db|
begin
db.query("SELECT i, str FROM a WHERE i = ?", 23) do |rs|
rs.move_next
break
end
rescue e : SQLite3::Exception
fail("Expected no exception, but got \"#{e.message}\"")
end
begin
db.exec("UPDATE a SET i = ? WHERE i = ?", 23, 23)
rescue e : SQLite3::Exception
fail("Expected no exception, but got \"#{e.message}\"")
end
end
end
end
2017-09-14 18:51:25 +00:00
it "handles single-step pragma statements" do |db|
db.exec %(PRAGMA synchronous = OFF)
end
it "handles multi-step pragma statements" do |db|
db.exec %(PRAGMA journal_mode = memory)
end
end