use Test::More; use strict; use warnings; use utf8; use Encode qw/encode_utf8 decode_utf8/; use FindBin; use lib "$FindBin::Bin/common"; BEGIN { eval { require DBD::SQLite; 1 } or plan skip_all => 'DBD::SQLite required'; eval { DBD::SQLite->VERSION >= 1.25 } or plan skip_all => 'DBD::SQLite >= 1.25 required'; plan 'no_plan'; use_ok('DBIx::Custom'); } $SIG{__WARN__} = sub { warn $_[0] unless $_[0] =~ /DEPRECATED/}; sub test { print "# $_[0]\n" } use DBIx::Custom; { package DBIx::Custom; has dsn => sub { 'dbi:SQLite:dbname=:memory:' } } use MyDBI1; { package MyDBI4; use strict; use warnings; use base 'DBIx::Custom'; sub connect { my $self = shift->SUPER::connect(@_); $self->include_model( MyModel2 => [ 'book', {class => 'Company', name => 'company'} ] ); } package MyModel2::Base1; use strict; use warnings; use base 'DBIx::Custom::Model'; package MyModel2::book; use strict; use warnings; use base 'MyModel2::Base1'; sub insert { my ($self, $param) = @_; return $self->SUPER::insert(param => $param); } sub list { shift->select; } package MyModel2::Company; use strict; use warnings; use base 'MyModel2::Base1'; sub insert { my ($self, $param) = @_; return $self->SUPER::insert(param => $param); } sub list { shift->select; } } { package MyDBI5; use strict; use warnings; use base 'DBIx::Custom'; sub connect { my $self = shift->SUPER::connect(@_); $self->include_model('MyModel4'); } } { package MyDBI6; use base 'DBIx::Custom'; sub connect { my $self = shift->SUPER::connect(@_); $self->include_model('MyModel5'); return $self; } } { package MyDBI7; use base 'DBIx::Custom'; sub connect { my $self = shift->SUPER::connect(@_); $self->include_model('MyModel6'); return $self; } } { package MyDBI8; use base 'DBIx::Custom'; sub connect { my $self = shift->SUPER::connect(@_); $self->include_model('MyModel7'); return $self; } } { package MyDBI9; use base 'DBIx::Custom'; sub connect { my $self = shift->SUPER::connect(@_); $self->include_model('MyModel8')->setup_model; return $self; } } # Constant my $create_table1 = 'create table table1 (key1 varchar, key2 varchar);'; my $create_table1_2 = 'create table table1 (key1 varchar, key2 varchar, key3 varchar, key4 varchar, key5 varchar);'; my $create_table2 = 'create table table2 (key1 varchar, key3 varchar);'; my $create_table2_2 = "create table table2 (key1 varchar, key2 varchar, key3 varchar)"; my $create_table3 = "create table table3 (key1 varchar, key2 varchar, key3 varchar)"; my $create_table_reserved = 'create table "table" ("select" varchar, "update" varchar)'; my $q = '"'; my $p = '"'; # Variables my $dbi; my $param; my $params; my $sql; my $result; my $row; my @rows; my $rows; my $model; my $model2; my $where; my $binary; # Prepare table $dbi = DBIx::Custom->connect; ### a little complex test test 'table_alias'; $dbi = DBIx::Custom->connect; eval { $dbi->execute('drop table table1') }; $dbi->execute("create table table1 (key1 Date, key2 datetime)"); $dbi->type_rule( into1 => { date => sub { uc $_[0] } } ); $dbi->execute("insert into table1 (key1) values (:table2.key1)", {'table2.key1' => 'a'}, table_alias => {table2 => 'table1'}); $result = $dbi->select(table => 'table1'); is($result->one->{key1}, 'A'); test 'type_rule into'; $dbi = DBIx::Custom->connect; $dbi->execute("create table table1 (key1 Date, key2 datetime)"); $dbi->type_rule( into1 => { date => sub { uc $_[0] } } ); $dbi->insert({key1 => 'a'}, table => 'table1'); $result = $dbi->select(table => 'table1'); is($result->one->{key1}, 'A'); $dbi = DBIx::Custom->connect; $dbi->execute("create table table1 (key1 date, key2 datetime)"); $dbi->type_rule( into1 => [ [qw/date datetime/] => sub { uc $_[0] } ] ); $dbi->insert({key1 => 'a', key2 => 'b'}, table => 'table1'); $result = $dbi->select(table => 'table1'); $row = $result->one; is($row->{key1}, 'A'); is($row->{key2}, 'B'); $dbi = DBIx::Custom->connect; $dbi->execute("create table table1 (key1 Date, key2 datetime)"); $dbi->insert({key1 => 'a', key2 => 'B'}, table => 'table1'); $dbi->type_rule( into1 => [ [qw/date datetime/] => sub { uc $_[0] } ] ); $result = $dbi->execute( "select * from table1 where key1 = :key1 and key2 = :table1.key2;", param => {key1 => 'a', 'table1.key2' => 'b'} ); $row = $result->one; is($row->{key1}, 'a'); is($row->{key2}, 'B'); $dbi = DBIx::Custom->connect; $dbi->execute("create table table1 (key1 Date, key2 datetime)"); $dbi->insert({key1 => 'A', key2 => 'B'}, table => 'table1'); $dbi->type_rule( into1 => [ [qw/date datetime/] => sub { uc $_[0] } ] ); $result = $dbi->execute( "select * from table1 where key1 = :key1 and key2 = :table1.key2;", param => {key1 => 'a', 'table1.key2' => 'b'}, table => 'table1' ); $row = $result->one; is($row->{key1}, 'A'); is($row->{key2}, 'B'); $dbi = DBIx::Custom->connect; $dbi->execute("create table table1 (key1 date, key2 datetime)"); $dbi->register_filter(twice => sub { $_[0] * 2 }); $dbi->type_rule( from1 => { date => 'twice', }, into1 => { date => 'twice', } ); $dbi->insert({key1 => 2}, table => 'table1'); $result = $dbi->select(table => 'table1'); is($result->fetch->[0], 8); test 'type_rule and filter order'; $dbi = DBIx::Custom->connect; $dbi->execute("create table table1 (key1 Date, key2 datetime)"); $dbi->type_rule( into1 => { date => sub { $_[0] . 'b' } }, into2 => { date => sub { $_[0] . 'c' } }, from1 => { date => sub { $_[0] . 'd' } }, from2 => { date => sub { $_[0] . 'e' } } ); $dbi->insert({key1 => '1'}, table => 'table1', filter => {key1 => sub { $_[0] . 'a' }}); $result = $dbi->select(table => 'table1'); $result->filter(key1 => sub { $_[0] . 'f' }); is($result->fetch_first->[0], '1abcdef'); $dbi = DBIx::Custom->connect; $dbi->execute("create table table1 (key1 Date, key2 datetime)"); $dbi->type_rule( from1 => { date => sub { $_[0] . 'p' } }, from2 => { date => sub { $_[0] . 'q' } }, ); $dbi->insert({key1 => '1'}, table => 'table1'); $result = $dbi->select(table => 'table1'); $result->type_rule( from1 => { date => sub { $_[0] . 'd' } }, from2 => { date => sub { $_[0] . 'e' } } ); $result->filter(key1 => sub { $_[0] . 'f' }); is($result->fetch_first->[0], '1def'); test 'type_rule_off'; $dbi = DBIx::Custom->connect; $dbi->execute("create table table1 (key1 Date, key2 datetime)"); $dbi->type_rule( from1 => { date => sub { $_[0] * 2 }, }, into1 => { date => sub { $_[0] * 2 }, } ); $dbi->insert({key1 => 2}, table => 'table1', type_rule_off => 1); $result = $dbi->select(table => 'table1', type_rule_off => 1); is($result->type_rule_off->fetch->[0], 2); $dbi = DBIx::Custom->connect; $dbi->execute("create table table1 (key1 Date, key2 datetime)"); $dbi->type_rule( from1 => { date => sub { $_[0] * 2 }, }, into1 => { date => sub { $_[0] * 3 }, } ); $dbi->insert({key1 => 2}, table => 'table1', type_rule_off => 1); $result = $dbi->select(table => 'table1', type_rule_off => 1); is($result->one->{key1}, 4); $dbi = DBIx::Custom->connect; $dbi->execute("create table table1 (key1 Date, key2 datetime)"); $dbi->type_rule( from1 => { date => sub { $_[0] * 2 }, }, into1 => { date => sub { $_[0] * 3 }, } ); $dbi->insert({key1 => 2}, table => 'table1'); $result = $dbi->select(table => 'table1'); is($result->one->{key1}, 12); $dbi = DBIx::Custom->connect; $dbi->execute("create table table1 (key1 Date, key2 datetime)"); $dbi->type_rule( from1 => { date => sub { $_[0] * 2 }, }, into1 => { date => sub { $_[0] * 3 }, } ); $dbi->insert({key1 => 2}, table => 'table1'); $result = $dbi->select(table => 'table1'); is($result->fetch->[0], 12); $dbi = DBIx::Custom->connect; $dbi->execute("create table table1 (key1 Date, key2 datetime)"); $dbi->register_filter(ppp => sub { uc $_[0] }); $dbi->type_rule( into1 => { date => 'ppp' } ); $dbi->insert({key1 => 'a'}, table => 'table1'); $result = $dbi->select(table => 'table1'); is($result->one->{key1}, 'A'); eval{$dbi->type_rule( into1 => { date => 'pp' } )}; like($@, qr/not registered/); $dbi = DBIx::Custom->connect; $dbi->execute("create table table1 (key1 Date, key2 datetime)"); eval { $dbi->type_rule( from1 => { Date => sub { $_[0] * 2 }, } ); }; like($@, qr/lower/); eval { $dbi->type_rule( into1 => { Date => sub { $_[0] * 2 }, } ); }; like($@, qr/lower/); $dbi = DBIx::Custom->connect; $dbi->execute("create table table1 (key1 Date, key2 datetime)"); $dbi->type_rule( from1 => { date => sub { $_[0] * 2 }, }, into1 => { date => sub { $_[0] * 3 }, } ); $dbi->insert({key1 => 2}, table => 'table1'); $result = $dbi->select(table => 'table1'); $result->type_rule_off; is($result->one->{key1}, 6); $dbi = DBIx::Custom->connect; $dbi->execute("create table table1 (key1 Date, key2 datetime)"); $dbi->type_rule( from1 => { date => sub { $_[0] * 2 }, datetime => sub { $_[0] * 4 }, }, ); $dbi->insert({key1 => 2, key2 => 2}, table => 'table1'); $result = $dbi->select(table => 'table1'); $result->type_rule( from1 => { date => sub { $_[0] * 3 } } ); $row = $result->one; is($row->{key1}, 6); is($row->{key2}, 2); $result = $dbi->select(table => 'table1'); $result->type_rule( from1 => { date => sub { $_[0] * 3 } } ); $row = $result->one; is($row->{key1}, 6); is($row->{key2}, 2); $result = $dbi->select(table => 'table1'); $result->type_rule( from1 => { date => sub { $_[0] * 3 } } ); $row = $result->one; is($row->{key1}, 6); is($row->{key2}, 2); $result = $dbi->select(table => 'table1'); $result->type_rule( from1 => [date => sub { $_[0] * 3 }] ); $row = $result->one; is($row->{key1}, 6); is($row->{key2}, 2); $dbi->register_filter(fivetimes => sub { $_[0] * 5}); $result = $dbi->select(table => 'table1'); $result->type_rule( from1 => [date => 'fivetimes'] ); $row = $result->one; is($row->{key1}, 10); is($row->{key2}, 2); $result = $dbi->select(table => 'table1'); $result->type_rule( from1 => [date => undef] ); $row = $result->one; is($row->{key1}, 2); is($row->{key2}, 2); $dbi = DBIx::Custom->connect; $dbi->execute("create table table1 (key1 Date, key2 datetime)"); $dbi->type_rule( from1 => { date => sub { $_[0] * 2 }, }, ); $dbi->insert({key1 => 2}, table => 'table1'); $result = $dbi->select(table => 'table1'); $result->filter(key1 => sub { $_[0] * 3 }); is($result->one->{key1}, 12); $dbi = DBIx::Custom->connect; $dbi->execute("create table table1 (key1 Date, key2 datetime)"); $dbi->type_rule( from1 => { date => sub { $_[0] * 2 }, }, ); $dbi->insert({key1 => 2}, table => 'table1'); $result = $dbi->select(table => 'table1'); $result->filter(key1 => sub { $_[0] * 3 }); is($result->fetch->[0], 12); $dbi = DBIx::Custom->connect; $dbi->execute("create table table1 (key1 Date, key2 datetime)"); $dbi->type_rule( into1 => { date => sub { $_[0] . 'b' } }, into2 => { date => sub { $_[0] . 'c' } }, from1 => { date => sub { $_[0] . 'd' } }, from2 => { date => sub { $_[0] . 'e' } } ); $dbi->insert({key1 => '1'}, table => 'table1', type_rule_off => 1); $result = $dbi->select(table => 'table1'); is($result->type_rule_off->fetch_first->[0], '1'); $result = $dbi->select(table => 'table1'); is($result->type_rule_on->fetch_first->[0], '1de'); $dbi = DBIx::Custom->connect; $dbi->execute("create table table1 (key1 Date, key2 datetime)"); $dbi->type_rule( into1 => { date => sub { $_[0] . 'b' } }, into2 => { date => sub { $_[0] . 'c' } }, from1 => { date => sub { $_[0] . 'd' } }, from2 => { date => sub { $_[0] . 'e' } } ); $dbi->insert({key1 => '1'}, table => 'table1', type_rule1_off => 1); $result = $dbi->select(table => 'table1'); is($result->type_rule1_off->fetch_first->[0], '1ce'); $result = $dbi->select(table => 'table1'); is($result->type_rule1_on->fetch_first->[0], '1cde'); $dbi = DBIx::Custom->connect; $dbi->execute("create table table1 (key1 Date, key2 datetime)"); $dbi->type_rule( into1 => { date => sub { $_[0] . 'b' } }, into2 => { date => sub { $_[0] . 'c' } }, from1 => { date => sub { $_[0] . 'd' } }, from2 => { date => sub { $_[0] . 'e' } } ); $dbi->insert({key1 => '1'}, table => 'table1', type_rule2_off => 1); $result = $dbi->select(table => 'table1'); is($result->type_rule2_off->fetch_first->[0], '1bd'); $result = $dbi->select(table => 'table1'); is($result->type_rule2_on->fetch_first->[0], '1bde'); test 'Model class'; use MyDBI1; $dbi = MyDBI1->connect; eval { $dbi->execute('drop table book') }; $dbi->execute("create table book (title, author)"); $model = $dbi->model('book'); $model->insert({title => 'a', author => 'b'}); is_deeply($model->list->all, [{title => 'a', author => 'b'}], 'basic'); $dbi->execute("create table company (name)"); $model = $dbi->model('company'); $model->insert({name => 'a'}); is_deeply($model->list->all, [{name => 'a'}], 'basic'); is($dbi->models->{'book'}, $dbi->model('book')); is($dbi->models->{'company'}, $dbi->model('company')); $dbi = MyDBI4->connect; eval { $dbi->execute('drop table book') }; $dbi->execute("create table book (title, author)"); $model = $dbi->model('book'); $model->insert({title => 'a', author => 'b'}); is_deeply($model->list->all, [{title => 'a', author => 'b'}], 'basic'); $dbi->execute("create table company (name)"); $model = $dbi->model('company'); $model->insert({name => 'a'}); is_deeply($model->list->all, [{name => 'a'}], 'basic'); $dbi = MyDBI5->connect; eval { $dbi->execute('drop table company') }; eval { $dbi->execute('drop table table1') }; $dbi->execute("create table company (name)"); $dbi->execute("create table table1 (key1)"); $model = $dbi->model('company'); $model->insert({name => 'a'}); is_deeply($model->list->all, [{name => 'a'}], 'include all model'); $dbi->insert(table => 'table1', param => {key1 => 1}); $model = $dbi->model('book'); is_deeply($model->list->all, [{key1 => 1}], 'include all model'); test 'primary_key'; use MyDBI1; $dbi = MyDBI1->connect; $model = $dbi->model('book'); $model->primary_key(['id', 'number']); is_deeply($model->primary_key, ['id', 'number']); test 'columns'; use MyDBI1; $dbi = MyDBI1->connect; $model = $dbi->model('book'); $model->columns(['id', 'number']); is_deeply($model->columns, ['id', 'number']); test 'setup_model'; use MyDBI1; $dbi = MyDBI1->connect; eval { $dbi->execute('drop table book') }; eval { $dbi->execute('drop table company') }; eval { $dbi->execute('drop table test') }; $dbi->execute('create table book (id)'); $dbi->execute('create table company (id, name);'); $dbi->execute('create table test (id, name);'); $dbi->setup_model; is_deeply($dbi->model('book')->columns, ['id']); is_deeply($dbi->model('company')->columns, ['id', 'name']); ### SQLite only test test 'prefix'; $dbi = DBIx::Custom->connect; eval { $dbi->execute('drop table table1') }; $dbi->execute('create table table1 (key1 varchar, key2 varchar, primary key(key1));'); $dbi->insert(table => 'table1', param => {key1 => 1, key2 => 2}); $dbi->insert(table => 'table1', param => {key1 => 1, key2 => 4}, prefix => 'or replace'); $result = $dbi->execute('select * from table1;'); $rows = $result->all; is_deeply($rows, [{key1 => 1, key2 => 4}], "basic"); $dbi = DBIx::Custom->connect; eval { $dbi->execute('drop table table1') }; $dbi->execute('create table table1 (key1 varchar, key2 varchar, primary key(key1));'); $dbi->insert(table => 'table1', param => {key1 => 1, key2 => 2}); $dbi->update(table => 'table1', param => {key2 => 4}, where => {key1 => 1}, prefix => 'or replace'); $result = $dbi->execute('select * from table1;'); $rows = $result->all; is_deeply($rows, [{key1 => 1, key2 => 4}], "basic"); test 'quote'; $dbi = DBIx::Custom->connect; $dbi->quote('"'); eval { $dbi->execute("drop table ${q}table$p") }; $dbi->execute($create_table_reserved); $dbi->apply_filter('table', select => {out => sub { $_[0] * 2}}); $dbi->insert(table => 'table', param => {select => 1}); $dbi->delete(table => 'table', where => {select => 1}); $result = $dbi->execute("select * from ${q}table$p"); $rows = $result->all; is_deeply($rows, [], "reserved word"); test 'finish statement handle'; $dbi = DBIx::Custom->connect; $dbi->execute($create_table1); $dbi->insert({key1 => 1, key2 => 2}, table => 'table1'); $dbi->insert({key1 => 3, key2 => 4}, table => 'table1'); $result = $dbi->select(table => 'table1'); $row = $result->fetch_first; is_deeply($row, [1, 2], "row"); $row = $result->fetch; ok(!$row, "finished"); $result = $dbi->select(table => 'table1'); $row = $result->fetch_hash_first; is_deeply($row, {key1 => 1, key2 => 2}, "row"); $row = $result->fetch_hash; ok(!$row, "finished"); $dbi->execute('create table table2 (key1, key2);'); $result = $dbi->select(table => 'table2'); $row = $result->fetch_hash_first; ok(!$row, "no row fetch"); $dbi = DBIx::Custom->connect; eval { $dbi->execute('drop table table1') }; $dbi->execute($create_table1); $dbi->insert({key1 => 1, key2 => 2}, table => 'table1'); $dbi->insert({key1 => 3, key2 => 4}, table => 'table1'); $dbi->insert({key1 => 5, key2 => 6}, table => 'table1'); $dbi->insert({key1 => 7, key2 => 8}, table => 'table1'); $dbi->insert({key1 => 9, key2 => 10}, table => 'table1'); $result = $dbi->select(table => 'table1'); $rows = $result->fetch_multi(2); is_deeply($rows, [[1, 2], [3, 4]], "fetch_multi first"); $rows = $result->fetch_multi(2); is_deeply($rows, [[5, 6], [7, 8]], "fetch_multi secound"); $rows = $result->fetch_multi(2); is_deeply($rows, [[9, 10]], "fetch_multi third"); $rows = $result->fetch_multi(2); ok(!$rows); $result = $dbi->select(table => 'table1'); eval {$result->fetch_multi}; like($@, qr/Row count must be specified/, "Not specified row count"); $result = $dbi->select(table => 'table1'); $rows = $result->fetch_hash_multi(2); is_deeply($rows, [{key1 => 1, key2 => 2}, {key1 => 3, key2 => 4}], "fetch_multi first"); $rows = $result->fetch_hash_multi(2); is_deeply($rows, [{key1 => 5, key2 => 6}, {key1 => 7, key2 => 8}], "fetch_multi secound"); $rows = $result->fetch_hash_multi(2); is_deeply($rows, [{key1 => 9, key2 => 10}], "fetch_multi third"); $rows = $result->fetch_hash_multi(2); ok(!$rows); $result = $dbi->select(table => 'table1'); eval {$result->fetch_hash_multi}; like($@, qr/Row count must be specified/, "Not specified row count"); test 'type option'; # DEPRECATED! $dbi = DBIx::Custom->connect( data_source => 'dbi:SQLite:dbname=:memory:', dbi_option => { $DBD::SQLite::VERSION > 1.26 ? (sqlite_unicode => 1) : (unicode => 1) } ); $binary = pack("I3", 1, 2, 3); eval { $dbi->execute('drop table table1') }; $dbi->execute('create table table1(key1, key2)'); $dbi->insert(table => 'table1', param => {key1 => $binary, key2 => 'あ'}, type => [key1 => DBI::SQL_BLOB]); $result = $dbi->select(table => 'table1'); $row = $result->one; is_deeply($row, {key1 => $binary, key2 => 'あ'}, "basic"); $result = $dbi->execute('select length(key1) as key1_length from table1'); $row = $result->one; is($row->{key1_length}, length $binary); test 'type_rule from'; $dbi = DBIx::Custom->connect; $dbi->type_rule( from1 => { date => sub { uc $_[0] } } ); $dbi->execute("create table table1 (key1 Date, key2 datetime)"); $dbi->insert({key1 => 'a'}, table => 'table1'); $result = $dbi->select(table => 'table1'); is($result->fetch_first->[0], 'A'); $result = $dbi->select(table => 'table1'); is($result->one->{key1}, 'A'); # DEPRECATED! test test 'filter __ expression'; $dbi = DBIx::Custom->connect; eval { $dbi->execute('drop table company') }; eval { $dbi->execute('drop table location') }; $dbi->execute('create table company (id, name, location_id)'); $dbi->execute('create table location (id, name)'); $dbi->apply_filter('location', name => {in => sub { uc $_[0] } } ); $dbi->insert(table => 'company', param => {id => 1, name => 'a', location_id => 2}); $dbi->insert(table => 'location', param => {id => 2, name => 'b'}); $result = $dbi->select( table => ['company', 'location'], relation => {'company.location_id' => 'location.id'}, column => ['location.name as location__name'] ); is($result->fetch_first->[0], 'B'); $result = $dbi->select( table => 'company', relation => {'company.location_id' => 'location.id'}, column => ['location.name as location__name'] ); is($result->fetch_first->[0], 'B'); $result = $dbi->select( table => 'company', relation => {'company.location_id' => 'location.id'}, column => ['location.name as "location.name"'] ); is($result->fetch_first->[0], 'B'); test 'reserved_word_quote'; $dbi = DBIx::Custom->connect; eval { $dbi->execute("drop table ${q}table$p") }; $dbi->reserved_word_quote('"'); $dbi->execute($create_table_reserved); $dbi->apply_filter('table', select => {out => sub { $_[0] * 2}}); $dbi->apply_filter('table', update => {out => sub { $_[0] * 3}}); $dbi->insert(table => 'table', param => {select => 1}); $dbi->update(table => 'table', where => {'table.select' => 1}, param => {update => 2}); $result = $dbi->execute("select * from ${q}table$p"); $rows = $result->all; is_deeply($rows, [{select => 2, update => 6}], "reserved word");