96 lines
3.8 KiB
Ruby
96 lines
3.8 KiB
Ruby
require_relative '../test_helper'
|
|
require_relative '../support/storage_contract'
|
|
|
|
class SqliteStorageTest < PipelineTest
|
|
include StorageContract
|
|
|
|
def contract_storage dir, purpose
|
|
db = "#{dir}/test.sqlite3"
|
|
unless File.exist?(db)
|
|
DataPipeline::SqliteStorage.new(database: db).tap do |s|
|
|
s.query(query: 'create table rows (id text, code text, name text, note text)')
|
|
s.close
|
|
end
|
|
end
|
|
purpose == :write ? DataPipeline::SqliteStorage.new(database: db, table: 'rows') : DataPipeline::SqliteStorage.new(database: db, query: 'select * from rows order by id')
|
|
end
|
|
|
|
def memory_db
|
|
DataPipeline::SqliteStorage.new(database: ':memory:').tap do |s|
|
|
s.query(query: 'create table t (id integer, name text, betrag real, datum text)')
|
|
end
|
|
end
|
|
|
|
def test_types_null_and_quotes
|
|
db = memory_db
|
|
db.write('t', [{ 'id' => 1, 'name' => "O'Brien", 'betrag' => 1.5, 'datum' => Date.new(2025, 11, 3) }, { id: 2, name: nil }])
|
|
assert_equal [{ 'id' => 1, 'name' => "O'Brien", 'betrag' => 1.5, 'datum' => '2025-11-03' },
|
|
{ 'id' => 2, 'name' => nil, 'betrag' => nil, 'datum' => nil }],
|
|
db.query(query: 'select * from t order by id')
|
|
db.close
|
|
end
|
|
|
|
def test_positional_and_named_params
|
|
db = memory_db
|
|
db.write('t', [{ 'id' => 1, 'name' => 'a' }, { 'id' => 2, 'name' => 'b' }])
|
|
assert_equal [{ 'name' => 'b' }], db.query(query: 'select name from t where id = ?', params: [2])
|
|
assert_equal [{ 'name' => 'a' }], db.query(query: 'select name from t where id = :id', params: { id: 1 })
|
|
db.close
|
|
end
|
|
|
|
def test_write_is_one_transaction
|
|
db = memory_db
|
|
db.query(query: 'create table u (id integer not null)')
|
|
assert_raises(SQLite3::Exception) { db.write('u', [{ 'id' => 1 }, { 'id' => nil }]) }
|
|
assert_equal 0, db.query(query: 'select count(*) as n from u').first['n']
|
|
db.close
|
|
end
|
|
|
|
def test_sql_injection_in_values_and_identifiers
|
|
db = memory_db
|
|
db.query(query: 'create table "Mixed ""Name""" ("Col A" text)')
|
|
db.write('Mixed "Name"', [{ 'Col A' => "x'); drop table t; --" }])
|
|
assert_equal [{ 'Col A' => "x'); drop table t; --" }], db.query(query: 'select * from "Mixed ""Name"""')
|
|
assert_equal 0, db.query(query: 'select count(*) as n from t').first['n']
|
|
db.close
|
|
end
|
|
|
|
def test_create_truncate_and_lazy_open
|
|
with_tmpdir do |dir|
|
|
path = "#{dir}/neu.sqlite3"
|
|
storage = DataPipeline::SqliteStorage.new(database: path, table: 'k', create: true, truncate: true)
|
|
refute File.exist?(path), 'opened only on first use'
|
|
storage.write([{ 'id' => 1, 'betrag' => 1.5, 'name' => 'a' }, { 'id' => 2, 'betrag' => 2.0, 'name' => 'b' }])
|
|
storage.close
|
|
|
|
again = DataPipeline::SqliteStorage.new(database: path, table: 'k', create: true, truncate: true)
|
|
again.write_row({ 'id' => 3, 'betrag' => 3.0, 'name' => 'c' })
|
|
assert_equal [{ 'id' => 3, 'betrag' => 3.0, 'name' => 'c' }], again.query(query: 'select * from k')
|
|
assert_equal %w[integer real text], again.query(query: 'pragma table_info(k)').map { |c| c['type'].downcase }
|
|
again.close
|
|
end
|
|
end
|
|
|
|
def test_missing_options
|
|
assert_raises(ArgumentError) { DataPipeline::SqliteStorage.new }
|
|
db = memory_db
|
|
assert_raises(RuntimeError) { db.write([{ 'id' => 1 }]) }
|
|
assert_raises(RuntimeError) { db.read }
|
|
db.close
|
|
end
|
|
|
|
def test_short_name_and_lookup
|
|
with_tmpdir do |dir|
|
|
path = "#{dir}/l.sqlite3"
|
|
names = nil
|
|
DataPipeline.new do
|
|
db = container :sqlite3, database: path
|
|
db.query(query: 'create table land (code text, name text)')
|
|
db.write('land', [{ 'code' => 'DE', 'name' => 'Deutschland' }])
|
|
names = lookup db, query: 'select * from land where code like ?', params: ['%'], key: 'code', unique: true
|
|
end
|
|
assert_equal 'Deutschland', names['DE']['name']
|
|
end
|
|
end
|
|
end
|