File Coverage

blib/lib/Test/MockSleep.pm

Criterion	Covered	Total	%
statement	38	39	97.4
branch	7	10	70.0
condition	3	8	37.5
subroutine	9	9	100.0
pod	0	3	0.0
total	57	69	82.6

package Test::MockSleep;
use strict;
use warnings;

our $VERSION = 0.02;

no warnings 'redefine';
my $time_hires_sleep;
my $have_mock_time;

our $use_mock_time = 0;

our $Slept;

sub mock_sleep (;@) {
    my $seconds = shift;
    $seconds ||= 0;
    $Slept += $seconds;
    if($have_mock_time && $use_mock_time) {
        Test::MockTime::set_relative_time($seconds);
    }
}

sub slept {
    my $ret = $Slept;
    $Slept = 0;
    return $ret;
}

eval {
    require Time::HiRes;
    $time_hires_sleep = \&Time::HiRes::sleep;
};

eval {
    require Test::MockTime;
    $have_mock_time = 1;
};

sub import {
    my $cls = shift;
    my @args = @_;

    if(grep $_ eq ':with_mocktime', @args) {
        $use_mock_time = 1;
    } else {
        $use_mock_time = 0;
    }

    my $cpkg = caller();
    no strict 'refs';
    *{$cpkg.'::slept'} = \&slept;
    *CORE::GLOBAL::sleep = \&mock_sleep;
    if($time_hires_sleep) {
        *Time::HiRes::sleep = \&mock_sleep;
    }
    1;
}

sub restore {
    my ($cpkg,@args);

    if($have_mock_time && $use_mock_time) {
        Test::MockTime::restore();
    }

    *CORE::GLOBAL::sleep = sub (;@) { CORE::sleep(shift) };
    if($time_hires_sleep) {
        *Time::HiRes::sleep = $time_hires_sleep;
    }
}
1;

__END__