File Coverage

blib/lib/Test/EasyMock/MockControl/Class.pm

Criterion	Covered	Total	%
statement	37	37	100.0
branch	1	2	50.0
condition			n/a
subroutine	10	10	100.0
pod	3	3	100.0
total	51	52	98.0

package Test::EasyMock::MockControl::Class;
use strict;
use warnings;

=head1 NAME

Test::EasyMock::MockControl::Class - Control behavior of the class method mocking.

=cut
use parent qw(Test::EasyMock::MockControl);
use Carp qw(confess);
use Scalar::Util qw(weaken);
use Test::MockModule;

=head1 CONSTRUCTORS

=head2 new($module)

Create a control instance for class method mocking.

=cut
sub new {
    my $class = shift;
    my $self = $class->SUPER::new(@_);

    confess('Requires module or object at an argument.(e.g. create_class_mock')
        unless $self->{_module};

    return $self;
}

=head1 INSTANCE METHODS

=head2 replay

Change to I mode.
Override for creating C and mocking class method.

=cut
sub replay {
    my $self = shift;
    $self->SUPER::replay(@_);

    # prevent circular reference
    my ($mock) = @_;
    weaken($mock);

    my $mock_module = Test::MockModule->new($self->{_module});
    foreach my $expectation (@{$self->{_expectations}}) {
        my $method = $expectation->method;
        $mock_module->mock($method => sub {
                               my $class = shift;
                               return $mock->$method(@_);
                           });
    }

    $self->{_mock_module} = $mock_module;
}

=head2 reset

Clear expectations and change to I mode.
Override for release C instance.

=cut
sub reset {
    my $self = shift;
    $self->SUPER::reset(@_);
    $self->{_mock_module} = undef;
}

1;