File Coverage

blib/lib/Test/Fixture/DBI/Util.pm

Criterion	Covered	Total	%
statement	21	41	51.2
branch	0	8	0.0
condition			n/a
subroutine	7	9	77.7
pod	2	2	100.0
total	30	60	50.0

package Test::Fixture::DBI::Util;

use strict;
use warnings;
use Exporter qw(import);

use Carp;
use DBI;
use YAML::Syck;
use UNIVERSAL::require;

our $VERSION = '0.01';
our @EXPORT  = qw(make_database_yaml make_fixture_yaml);

sub make_database_yaml {
    my ( $dbh, $filename ) = @_;
    my $driver       = $dbh->{Driver}{Name};
    my $driver_class = __PACKAGE__ . '::' . $driver;
    unless ( eval { $driver_class->require; 1 } ) {
        croak( sprintf( 'Driver %s is not supported yet', $driver ) );
    }
    my $data = $driver_class->make_database($dbh);
    if ($filename) {
        YAML::Syck::DumpFile( $filename, $data );
    }
    else {
        return $data;
    }
}

sub make_fixture_yaml {
    my ( $dbh, $schema, $name_column, $sql, $filename ) = @_;
    my $rows = $dbh->selectall_arrayref( $sql, +{ Slice => +{} } );

    my @data;
    for my $row (@$rows) {
        push(
            @data,
            +{
                name => ref $name_column
                ? join( '_', map { $row->{$_} } @$name_column )
                : $row->{$name_column},
                schema => $schema,
                data   => $row,
            }
        );
    }

    if ($filename) {
        YAML::Syck::DumpFile( $filename, \@data );
    }
    else {
        return \@data;
    }
}

1;

__END__