File Coverage

blib/lib/Test/Count/FileMutator.pm

Criterion	Covered	Total	%
statement	52	62	83.8
branch	9	12	75.0
condition	1	3	33.3
subroutine	10	14	71.4
pod	4	4	100.0
total	76	95	80.0

package Test::Count::FileMutator;
$Test::Count::FileMutator::VERSION = '0.1104';
use warnings;
use strict;

use parent 'Test::Count::Base';

use Test::Count      ();
use Test::Count::Lib ();


sub _counter
{
    my $self = shift;
    if (@_)
    {
        $self->{'_counter'} = shift;
    }
    return $self->{'_counter'};
}

sub _filename
{
    my $self = shift;
    if (@_)
    {
        $self->{'_filename'} = shift;
    }
    return $self->{'_filename'};
}

sub _plan_prefix_regex
{
    my $self = shift;
    if (@_)
    {
        $self->{'_plan_prefix_regex'} = shift;
    }
    return $self->{'_plan_prefix_regex'};
}

sub _assert_prefix_regex
{
    my $self = shift;
    if (@_)
    {
        $self->{'_assert_prefix_regex'} = shift;
    }
    return $self->{'_assert_prefix_regex'};
}


sub _init
{
    my $self = shift;
    my $args = shift;

    $args->{plan_prefix_regex} ||= Test::Count::Lib::perl_plan_prefix_regex();

    # Remmed out because Test::Count handles it by itself.
    # if (defined($args->{assert_prefix_regex}))
    # {
    #     $self->_assert_prefix_regex($args->{assert_prefix_regex});
    # }
    $self->_plan_prefix_regex( $args->{plan_prefix_regex} );
    $self->_filename( $args->{filename} );

    $self->_counter( Test::Count->new($args) );

    return 0;
}


sub modify
{
    my $self = shift;

    my $ret = $self->_counter()->process();

    my $count = $ret->{tests_count};

    my $plan_re = $self->_plan_prefix_regex();

    my @lines = @{ $ret->{lines} };

    open my $out_fh, ">:raw", $self->_filename()
        or die "Could not open file '"
        . $self->_filename()
        . "' for writing - $!.";
LINES_LOOP:
    while ( my $l = shift(@lines) )
    {
        if ( $l =~ s{^($plan_re)\d+}{$1$count} )
        {
            print {$out_fh} $l;
            last LINES_LOOP;
        }
        else
        {
            print {$out_fh} $l;
        }
    }
    print {$out_fh} @lines;

    close($out_fh);
    return 0;
}


sub update_assignments
{
    my ( $self, $args ) = @_;

    return $self->_parser()->assignments( $args->{text} );
}


sub update_count
{
    my ( $self, $args ) = @_;

    return $self->_parser()->update_count( $args->{text} );
}


sub get_count
{
    my $self = shift;

    return $self->_parser()->{count};
}


1;    # End of Test::Count::Parser

__END__