File Coverage

blib/lib/Test/Mock/Two.pm

Criterion	Covered	Total	%
statement	48	50	96.0
branch	18	22	81.8
condition	2	6	33.3
subroutine	10	10	100.0
pod	3	3	100.0
total	81	91	89.0

package Test::Mock::Two;
our $VERSION = '0.011';
use warnings;
use strict;

# ABSTRACT: Inspection module for Test::Mock::One

use Exporter 'import';
use Carp qw(croak carp);
use List::Util qw(any);

our @EXPORT_OK = qw(
    one_called
    one_called_ok
    one_called_times_ok
    one_called_params_ok
);

our %EXPORT_TAGS = (
    all => \@EXPORT_OK,
    test => [ grep { $_ =~ /_ok$/ } @EXPORT_OK ],
);

use Test::Builder;
my $tb = Test::Builder->new();

sub one_called {
    my ($one, $method, $func) = @_;

    if (!UNIVERSAL::isa($one, 'Test::Mock::One')) {
        croak "We only play well with Test::Mock::One";
    }

    if (!$one->{'X-Mock-Called'}) {
        croak "Test::Mock::One object did not specify call tracing", $/;
    }

    if (!defined $method) {
        croak "Failed to provide a method";
    }

    if (my $cs = $one->{'X-Mock-Called-By'}) {
        my $rv = $cs->{$method};

        return unless $rv;
        return $rv if !defined $func;

        return $rv->{$func} if exists $rv->{$func};
    }

    return if !defined $func;

    $func =~ s#::#/#g;
    $func .= '.pm';
    if (any { $func eq $_ } keys %INC) {
        carp("Using Pkg::Name instead of Pkg::Name::Function");
    }
    return;
}

sub one_called_ok {
    my ($one, $method, $func, $msg) = @_;

    $msg //= "${func} called Test::Mock::One->$method";
    my $rv = one_called($one, $method, $func);
    $tb->ok($rv, $msg);
    return $rv if defined $rv;
    return ;
}

sub one_called_times_ok {
    my ($one, $method, $func, $times, $msg) = @_;

    $msg //= "${func} called $times on Test::Mock::One->$method";
    if (my $rv = one_called($one, $method, $func)) {
        $tb->is_eq(scalar @$rv, $times, $msg);
        return $rv;
    }
    else {
        $tb->is_eq(0, $times, $msg);
        return;
    }
}

1;

__END__