File Coverage

lib/DBIx/MoCo/Fixture.pm

Criterion	Covered	Total	%
statement	59	60	98.3
branch	5	10	50.0
condition	3	8	37.5
subroutine	14	14	100.0
pod	1	5	20.0
total	82	97	84.5

package DBIx::MoCo::Fixture;
use strict;
use warnings;
use Exporter::Lite;

use Carp qw/croak/;
use FindBin;
use Kwalify ();
use Path::Class;
use UNIVERSAL::require;
use YAML::Syck;

our $VERSION = 0.01;

our @EXPORT_OK = qw/fixtures/;
our @EXPORT = @EXPORT_OK;

sub fixtures (@) {
    my $option = pop if ref $_[-1] and ref $_[-1] eq 'HASH';
    $option ||= {};
    croak 'usage: fixtures(@models, { yaml_dir => '...' })' unless @_;

    my $dir = $option->{yaml_dir} || dir($FindBin::Bin)->parent->subdir('fixtures');
    my $res;
    for (@_) {
        my $config = validate_fixture(
            load_fixture( file($dir, sprintf("%s.yml", $_)) )
        );
        $config->{model}->require or die $@;

        delete_all( $config->{model} );
        $res->{$_} = insert( $config->{model}, $config->{records} );
    }
    $res;
}

sub load_fixture {
    my $file = shift;
    my $yaml = $file->slurp or die $!;

    YAML::Syck::Load($yaml);
}

sub validate_fixture {
    my $stuff = shift;

    Kwalify::validate({
        type    => 'map',
        mapping => {
            model   => { type => 'str', required => 1 },
            records => { type => 'any', required => 1 },
        }
    }, $stuff);

    $stuff;
}

sub delete_all {
    my $model = shift;
    if ($model->db->vendor eq 'SQLite') {
        $_->delete for $model->search;
    } else {
        $model->delete_all(where => { 1 => '1' });
    }
}

sub insert {
    my ($model, $records) = @_;

    my $res = {};
    for my $name (keys %$records) {
        my $obj = $model->create( %{ $records->{$name} } );
        $obj->save;

        my %query;
        for (@{ $obj->primary_keys }) {
            $query{$_} = $obj->$_;
        }

        $res->{$name} = $model->retrieve(%query);
    }
    $res;
}

1;

__END__