DBIx-Custom / t / 02-sqlite.t /
Newer Older
109 lines | 2.443kb
add test
yuki-kimoto authored on 2009-10-18
1
use Test::More;
2
use strict;
3
use warnings;
add test module
yuki-kimoto authored on 2009-10-19
4
use DBI;
add test
yuki-kimoto authored on 2009-10-18
5

            
6
BEGIN {
7
    eval { require DBD::SQLite; 1 }
8
        or plan skip_all => 'DBD::SQLite required';
9
    eval { DBD::SQLite->VERSION >= 1 }
10
        or plan skip_all => 'DBD::SQLite >= 1.00 required';
11

            
12
    plan 'no_plan';
13
    use_ok('DBI::Custom');
14
}
15

            
add test module
yuki-kimoto authored on 2009-10-19
16
package Test::DBI::Custom;
17
use Object::Simple;
add test
yuki-kimoto authored on 2009-10-18
18

            
add test module
yuki-kimoto authored on 2009-10-19
19
sub dbi : Attr {}
add test
yuki-kimoto authored on 2009-10-18
20

            
add test module
yuki-kimoto authored on 2009-10-19
21
sub new {
22
    my $self = shift->SUPER::new;
23
    my $dbi = DBI::Custom->new->connect_info(data_source => 'dbi:SQLite:dbname=:memory:');
24
    
25
    $dbi->connect;
26
    $self->dbi($dbi);
27
    return $self;
28
}
29

            
30
sub create_table {
31
    my ($self, $create_table) = @_;
32
    $self->dbi->query_raw_sql($create_table);
33
    return $self;
34
}
35

            
36
sub create_table1 {
37
    my $self = shift;
38
    $self->create_table("create table t1 (k1 char(255), k2 char(255), k3 char(255), k4 char(255), k5 char(255));");
39
    return $self;
40
}
41

            
42
sub insert {
43
    my ($self, @values_list) = @_;
44
    my $table = ref $values_list[0] ? '' : shift;
45
    $table ||= 't1';
46
    
47
    foreach my $values (@values_list) {
48
        my $sql = $self->dbi->query(
49
            "insert into $table {insert_values}", {insert_values => $values}
50
        );
51
    }
52
    return $self;
53
}
54

            
55
sub test {
56
    my ($self, $code) = @_;
57
    $code->($self->dbi);
58
}
59

            
60
Object::Simple->build_class;
61

            
62
package main;
63
my $t = Test::DBI::Custom->new;
64

            
65
$t->new->create_table1->insert({k1 => 1, k2 => 2}, {k1 => 3, k2 => 4})->test(sub {
66
    my $dbi = shift;
add test
yuki-kimoto authored on 2009-10-18
67
    
cleanup
yuki-kimoto authored on 2009-10-19
68
    my $r;     # resultset
69
    my @rows;
70
    
71
    # Simple query array
72
    $r = $dbi->query("select k1, k2 from t1");
73
    
74
    @rows = ();
75
    while (my $row = $r->fetch) {
76
        push @rows, [@$row];
77
    }
78
    is_deeply(\@rows, [[1, 2], [3, 4]], 'Simple query array');
79
    
80
    # Simple query hash
81
    $r = $dbi->query("select k1, k2 from t1;");
82
    
83
    @rows = ();
84
    while (my $row = $r->fetch_hash) {
85
        push @rows, {%$row};
86
    }
87
    is_deeply(\@rows, [{k1 => 1, k2 => 2}, {k1 => 3, k2 => 4}], 'Simple query array');
88
    
89
    
90
    
91
    
92
    
add test
yuki-kimoto authored on 2009-10-18
93
    $dbi->fetch_filter(sub {
94
        my ($key, $value) = @_;
95
        if ($key eq 'k1' && $value == 1 ) {
96
            return $value * 3;
97
        }
98
        return $value;
99
    });
100
    
cleanup
yuki-kimoto authored on 2009-10-19
101
    $r = $dbi->query("select k1, k2 from t1");
add test
yuki-kimoto authored on 2009-10-18
102
    
cleanup
yuki-kimoto authored on 2009-10-19
103
    my $row = $r->fetch;
add test
yuki-kimoto authored on 2009-10-18
104
    my @values = @$row;
cleanup
yuki-kimoto authored on 2009-10-19
105
    $r->finish;
add test
yuki-kimoto authored on 2009-10-18
106
    
107
    is_deeply(\@values, [3, 2]);
add test module
yuki-kimoto authored on 2009-10-19
108
});
109