use Test::More;

use strict;

use warnings;

# user password database

our ($USER, $PASSWORD, $DATABASE) = connect_info();

plan skip_all => 'private MySQL test' unless $USER;

plan 'no_plan';

require DBIx::Connector;

# Function for test name

sub test { print "# $_[0]\n" }

# Functions for tests

sub connect_info {

    my $file = 'password.tmp';

    open my $fh, '<', $file

      or return;

    my ($user, $password, $database) = split(/\s/, (<$fh>)[0]);

    close $fh;

    return ($user, $password, $database);

}

# Varialbes for tests

my $dbi;

my $dbname;

my $rows;

my $result;

{

    package DBIx::Custom::MySQL;

    use strict;

    use warnings;

    use base 'DBIx::Custom';

    __PACKAGE__->attr([qw/database host port/]);

    sub connect {

        my $proto = shift;

        # Create a new object

        my $self = ref $proto ? $proto : $proto->new(@_);

        # Data source

        if (!$self->data_source) {

            my $database = $self->database;

            my $host     = $self->host;

            my $port     = $self->port;

            my $data_source = "dbi:mysql:";

            $data_source .= "database=$database;" if $database;

            $data_source .= "host=$host;"         if $host;

            $data_source .= "port=$port;"         if $port;

            $self->data_source($data_source);

        }

        return $self->SUPER::connect;

    }

    1;

}

test 'connect';

$dbi = DBIx::Custom::MySQL->new(user => $USER, password => $PASSWORD,

                    database => $DATABASE, host => 'localhost', port => '10000');

$dbi->connect;

like($dbi->data_source, qr/dbi:mysql:database=.*;host=localhost;port=10000;/, "created data source");

is(ref $dbi->dbh, 'DBI::db');

test 'attributes';

$dbi = DBIx::Custom::MySQL->new;

$dbi->host('a');

is($dbi->host, 'a', "host");

$dbi->port('b');

is($dbi->port, 'b', "port");

test 'limit';

$dbi = DBIx::Custom->connect(

    data_source => "dbi:mysql:database=$DATABASE",

    user => $USER,

    password => $PASSWORD

);

$dbi->delete_all(table => 'table1');

$dbi->insert(table => 'table1', param => {key1 => 1, key2 => 2});

$dbi->insert(table => 'table1', param => {key1 => 1, key2 => 4});

$dbi->insert(table => 'table1', param => {key1 => 1, key2 => 6});

$dbi->query_builder->register_tag_processor(

    limit => sub {

        my ($count, $offset) = @_;

        my $s = '';

        $offset = 0 unless defined $offset;

        $s .= "limit $offset";

        $s .= ", $count";

        return [$s, []];

    }

);

$rows = $dbi->select(

  table => 'table1',

  where => {key1 => 1},

  append => "order by key2 {limit 1 0}"

)->fetch_hash_all;

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

$rows = $dbi->select(

  table => 'table1',

  where => {key1 => 1},

  append => "order by key2 {limit 2 1}"

)->fetch_hash_all;

is_deeply($rows, [{key1 => 1, key2 => 4},{key1 => 1, key2 => 6}]);

$rows = $dbi->select(

  table => 'table1',

  where => {key1 => 1},

  append => "order by key2 {limit 1}"

)->fetch_hash_all;

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

$dbi->delete_all(table => 'table1');

test 'dbh';

{

    my $connector = DBIx::Connector->new(

        "dbi:mysql:database=$DATABASE",

        $USER,

        $PASSWORD,

        DBIx::Custom->new->default_dbi_option

    );

    my $dbi = DBIx::Custom->connect(connector => $connector);

    $dbi->delete_all(table => 'table1');

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

    is($dbi->select(table => 'table1')->fetch_hash_first->{key1}, 1);

    $dbi = DBIx::Custom->new;

    $dbi->dbh('a');

    is($dbi->{dbh}, 'a');

}

test 'transaction';

test 'dbh';

{

    my $connector = DBIx::Connector->new(

        "dbi:mysql:database=$DATABASE",

        $USER,

        $PASSWORD,

        DBIx::Custom->new->default_dbi_option

    );

    my $dbi = DBIx::Custom->connect(connector => $connector);

    $dbi->delete_all(table => 'table1');

    $dbi->connector->txn(sub {

        $dbi->insert(table => 'table1', param => {key1 => 1, key2 => 2});

        $dbi->insert(table => 'table1', param => {key1 => 3, key2 => 4});

    });

    is_deeply($dbi->select(table => 'table1')->fetch_hash_all,

              [{key1 => 1, key2 => 2}, {key1 => 3, key2 => 4}]);

    $dbi->delete_all(table => 'table1');

    eval {

        $dbi->connector->txn(sub {

            $dbi->insert(table => 'table1', param => {key1 => 1, key2 => 2});

            die "Error";

            $dbi->insert(table => 'table1', param => {key1 => 3, key2 => 4});

        });

    };

    is_deeply($dbi->select(table => 'table1')->fetch_hash_all,

              []);

}

test 'fork';

{

    my $connector = DBIx::Connector->new(

        "dbi:mysql:database=$DATABASE",

        $USER,

        $PASSWORD,

        DBIx::Custom->new->default_dbi_option

    );

    my $dbi = DBIx::Custom->new(connector => $connector);

    $dbi->delete_all(table => 'table1');

    $dbi->insert(table => 'table1', param => {key1 => 1, key2 => 2});

    die "Can't fork" unless defined (my $pid = fork);

    if ($pid) {

        # Parent

        my $result = $dbi->select(table => 'table1');

        is_deeply($result->fetch_hash_first, {key1 => 1, key2 => 2});

    }

    else {

        # Child

        my $result = $dbi->select(table => 'table1');

        die "Not OK" unless $result->fetch_hash_first->{key1} == 1;

    }

}