use Test::More;

use strict;

use warnings;

use DBI;

BEGIN {

    eval { require DBD::SQLite; 1 }

        or plan skip_all => 'DBD::SQLite required';

    eval { DBD::SQLite->VERSION >= 1 }

        or plan skip_all => 'DBD::SQLite >= 1.00 required';

    plan 'no_plan';

    use_ok('DBIx::Custom::Result');

}

my $test;

sub test {

    $test = shift;

}

sub query {

    my ($dbh, $sql) = @_;

    my $sth = $dbh->prepare($sql);

    $sth->execute;

    return DBIx::Custom::Result->new(sth => $sth);

}

my $dbh;

my $sql;

my $sth;

my @row;

my $row;

my @rows;

my $rows;

my $result;

my $filter;

my @error;

my $error;

$dbh = DBI->connect('dbi:SQLite:dbname=:memory:', undef, undef, {PrintError => 0, RaiseError => 1});

$dbh->do("create table table1 (key1 char(255), key2 char(255));");

$dbh->do("insert into table1 (key1, key2) values ('1', '2');");

$dbh->do("insert into table1 (key1, key2) values ('3', '4');");

$sql = "select key1, key2 from table1";

test 'fetch scalar context';

$result = query($dbh, $sql);

@rows = ();

while (my $row = $result->fetch) {

    push @rows, [@$row];

}

is_deeply(\@rows, [[1, 2], [3, 4]], $test);

test 'fetch list context';

$result = query($dbh, $sql);

@rows = ();

while (my @row = $result->fetch) {

    push @rows, [@row];

}

is_deeply(\@rows, [[1, 2], [3, 4]], $test);

test 'fetch_hash scalar context';

$result = query($dbh, $sql);

@rows = ();

while (my $row = $result->fetch_hash) {

    push @rows, {%$row};

}

is_deeply(\@rows, [{key1 => 1, key2 => 2}, {key1 => 3, key2 => 4}], $test);

test 'fetch hash list context';

$result = query($dbh, $sql);

@rows = ();

while (my %row = $result->fetch_hash) {

    push @rows, {%row};

}

is_deeply(\@rows, [{key1 => 1, key2 => 2}, {key1 => 3, key2 => 4}], $test);

test 'fetch_single';

$result = query($dbh, $sql);

$row = $result->fetch_single;

is_deeply($row, [1, 2], "$test : row");

$row = $result->fetch;

ok(!$row, "$test : finished");

test 'fetch_single list context';

$result = query($dbh, $sql);

@row = $result->fetch_single;

is_deeply([@row], [1, 2], "$test : row");

@row = $result->fetch;

ok(!@row, "$test : finished");

test 'fetch_hash_single';

$result = query($dbh, $sql);

$row = $result->fetch_hash_single;

is_deeply($row, {key1 => 1, key2 => 2}, "$test : row");

$row = $result->fetch_hash;

ok(!$row, "$test : finished");

test 'fetch_hash_single list context';

$result = query($dbh, $sql);

@row = $result->fetch_hash_single;

is_deeply({@row}, {key1 => 1, key2 => 2}, "$test : row");

@row = $result->fetch_hash;

ok(!@row, "$test : finished");

test 'fetch_multi';

$dbh->do("insert into table1 (key1, key2) values ('5', '6');");

$dbh->do("insert into table1 (key1, key2) values ('7', '8');");

$dbh->do("insert into table1 (key1, key2) values ('9', '10');");

$result = query($dbh, $sql);

$rows = $result->fetch_multi(2);

is_deeply($rows, [[1, 2],

                  [3, 4]], "$test : fetch_multi first");

$rows = $result->fetch_multi(2);

is_deeply($rows, [[5, 6],

                  [7, 8]], "$test : fetch_multi secound");

$rows = $result->fetch_multi(2);

is_deeply($rows, [[9, 10]], "$test : fetch_multi third");

$rows = $result->fetch_multi(2);

ok(!$rows);

test 'fetch_multi list context';

$result = query($dbh, $sql);

@rows = $result->fetch_multi(2);

is_deeply([@rows], [[1, 2],

                  [3, 4]], "$test : fetch_multi first");

@rows = $result->fetch_multi(2);

is_deeply([@rows], [[5, 6],

                  [7, 8]], "$test : fetch_multi secound");

@rows = $result->fetch_multi(2);

is_deeply([@rows], [[9, 10]], "$test : fetch_multi third");

@rows = $result->fetch_multi(2);

ok(!@rows);

test 'fetch_multi error';

$result = query($dbh, $sql);

eval {$result->fetch_multi};

like($@, qr/Row count must be specified/, "$test : Not specified row count");

test 'fetch_hash_multi';

$result = query($dbh, $sql);

$rows = $result->fetch_hash_multi(2);

is_deeply($rows, [{key1 => 1, key2 => 2},

                  {key1 => 3, key2 => 4}], "$test : fetch_multi first");

$rows = $result->fetch_hash_multi(2);

is_deeply($rows, [{key1 => 5, key2 => 6},

                  {key1 => 7, key2 => 8}], "$test : fetch_multi secound");

$rows = $result->fetch_hash_multi(2);

is_deeply($rows, [{key1 => 9, key2 => 10}], "$test : fetch_multi third");

$rows = $result->fetch_hash_multi(2);

ok(!$rows);

test 'fetch_multi list context';

$result = query($dbh, $sql);

@rows = $result->fetch_hash_multi(2);

is_deeply([@rows], [{key1 => 1, key2 => 2},

                    {key1 => 3, key2 => 4}], "$test : fetch_multi first");

@rows = $result->fetch_hash_multi(2);

is_deeply([@rows], [{key1 => 5, key2 => 6},

                    {key1 => 7, key2 => 8}], "$test : fetch_multi secound");

@rows = $result->fetch_hash_multi(2);

is_deeply([@rows], [{key1 => 9, key2 => 10}], "$test : fetch_multi third");

@rows = $result->fetch_hash_multi(2);

ok(!@rows);

$dbh->do("delete from table1 where key1 = 5 or key1 = 7 or key1 = 9");

test 'fetch_multi error';

$result = query($dbh, $sql);

eval {$result->fetch_hash_multi};

like($@, qr/Row count must be specified/, "$test : Not specified row count");

test 'fetch_all';

$result = query($dbh, $sql);

$rows = $result->fetch_all;

is_deeply($rows, [[1, 2], [3, 4]], $test);

test 'fetch_all list context';

$result = query($dbh, $sql);

@rows = $result->fetch_all;

is_deeply(\@rows, [[1, 2], [3, 4]], $test);

test 'fetch_hash_all';

$result = query($dbh, $sql);

@rows = $result->fetch_hash_all;

is_deeply($rows, [[1, 2], [3, 4]], $test);

test 'fetch_hash_all list context';

$result = query($dbh, $sql);

@rows = $result->fetch_all;

is_deeply(\@rows, [[1, 2], [3, 4]], $test);

test 'fetch filter';

$result = query($dbh, $sql);

$result->filters({three_times => sub { $_[0] * 3}});

$result->filter({key1 => 'three_times'});

$rows = $result->fetch_all;

is_deeply($rows, [[3, 2], [9, 4]], "$test array");

$result = query($dbh, $sql);

$result->filters({three_times => sub { $_[0] * 3}});

$result->filter({key1 => 'three_times'});

$rows = $result->fetch_hash_all;

is_deeply($rows, [{key1 => 3, key2 => 2}, {key1 => 9, key2 => 4}], "$test hash");