File Coverage

blib/lib/Test/Mock/Simple.pm

Criterion	Covered	Total	%
statement	43	43	100.0
branch	14	16	87.5
condition	4	6	66.6
subroutine	7	7	100.0
pod	2	2	100.0
total	70	74	94.5

package Test::Mock::Simple;

use 5.008008;
use strict;
use warnings;

our $VERSION = '0.09';

my $allow_new_methods = 0;

sub new {
  my $package = shift;
  my $class = ref($package) || $package;

  my $self = {@_};
  bless($self, $class);

  if (!$self->{module}) {
    require Carp;
    Carp::croak("No module name provided to mock");
  }

  if (!$self->{no_load}) {
    my $module = $self->{module} . '.pm';
    $module =~ s/::/\//g;
    require $module;
  }

  $allow_new_methods = 1 if $self->{allow_new_methods} || $self->{no_load};

  return $self;
}

sub add {
  my $self = shift;
  my $name = shift;
  my $sub = shift;

  return if $ENV{TEST_MOCK_SIMPLE_DISABLE};

  if (!$name) {
    require Carp;
    Carp::croak("No method name provided to mock");
  }
  if (!$sub) {
    require Carp;
    Carp::croak("No sub ref provided to mock");
  }

  if (!$allow_new_methods) {
      die("Module (" . $self->{module} . ") does not have a method named '$name'\n")
        unless $self->{module}->can($name);
  }

  {
    no strict;
    no warnings;

    *{$self->{module} . '::' . $name} = $sub;
  }
}

1;
__END__