File Coverage

blib/lib/Test/Mock/Simple.pm

Criterion	Covered	Total	%
statement	45	46	97.8
branch	16	18	88.8
condition	4	6	66.6
subroutine	7	7	100.0
pod	2	2	100.0
total	74	79	93.6

package Test::Mock::Simple;

use 5.008008;
use strict;
use warnings;

our $VERSION = '0.10';

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");
  }

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

  if ($self->{no_load}) {
    $INC{$module_location} = 1;
  }
  else {
    require $module_location;
  }

  $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__