use warnings;
use strict;

use lib '.';
#use Count;

print One::foo();
{
    #my $count = Count->new;
    my $bar = Count->mock;
    print One::foo();
}
print One::foo();

BEGIN{
package One;

sub foo {
    return "foo\n";
}
1;

package Count;

sub new {
    return bless {}, shift;
}
sub unmock {
    my $self = shift;
    *One::foo = \&{ $self->{sub} };
}
sub mock {
    my $thing = shift;
    my $self;

    if (ref($thing) eq __PACKAGE__){
        $self = $thing;
    }
    else {
        $self = bless {}, __PACKAGE__;
    }
    use Scalar::Util 'weaken';
    weaken $self;

    $self->{sub} = \&One::foo;
    {
        no warnings 'redefine';
        *One::foo = sub { $self->{x} = 'x'; return "baz\n"; };
    }
    return $self;
}
sub DESTROY {
    my $self = shift;
    print "destroying...\n";
    $self->unmock;
}
1;

}
[download]

outputs

foo
Use of uninitialized value in subroutine dereference at script.pl line
+ 30.
destroying...
baz
baz
    (in cleanup) Unable to create sub named "" at script.pl line 30.
[download]

If you weaken $self then you immediately lower the reference count to 0. If you copy $self to $closure_self then you raise the count to 2, then lower it to 1. $self does not go out of scope until after the return, at which point you've passed an external reference to your main block and thus still have a ref count of 1.

I ran your posted code under ActiveState 5.20, and got the output

foo
destroying...
baz
baz
[download]

and if I swap the chunk in question to

    my $closure_self = $self;
    weaken $closure_self;
    $self->{sub} = \&One::foo;
    *One::foo = sub { $closure_self->{x} = 'x'; return "baz\n"; };
    return $self;
[download]

I get

foo
baz
destroying...
foo
[download]

Your code cannot restore the old function, because your object was out of scope before your assignment. This could be demonstrated by using Data::Dump in DESTROY:

sub DESTROY {
    use Data::Dump;
    my $self = shift;
    print "destroying...\n";
    print dd($self), "\n";
    $self->unmock;
}
[download]

which, for your code, outputs

foo
destroying...
bless({}, "Count")
1
baz
baz
[download]

and for mine

foo
baz
destroying...
bless({ sub => sub { ... }, x => "x" }, "Count")
1
foo
[download]

You will also note you'll start getting the errors and warnings I posted if you put strict and warnings in your modules instead of just in your script.

This all raises the point that unmock is actually written terribly, because it creates a whole additional closure about $self. It should really be

sub unmock {
    my $self = shift;
    *One::foo = $self->{sub};
}
[download]

so that you restore the original sub reference.

---

---

Posts are HTML formatted. Put <p> </p> tags around your paragraphs. Put <code> </code> tags around your code and data!

Titles consisting of a single word are discouraged, and in most cases are disallowed outright.

Read Where should I post X? if you're not absolutely sure you're posting in the right place.

Please read these before you post! —

• How do I compose an effective node title?
• How do I post a question effectively?
• Markup in the Monastery

Posts may use any of the Perl Monks Approved HTML tags:

a, abbr, b, big, blockquote, br, caption, center, col, colgroup, dd, del, details, div, dl, dt, em, font, h1, h2, h3, h4, h5, h6, hr, i, ins, li, ol, p, pre, readmore, small, span, spoiler, strike, strong, sub, summary, sup, table, tbody, td, tfoot, th, thead, tr, tt, u, ul, wbr

You may need to use entities for some characters, as follows. (Exception: Within code tags, you can put the characters literally.)

	For:		Use:
	&		&
	<		&lt;
	>		&gt;
	[		&#91;
	]		&#93;

Link using PerlMonks shortcuts! What shortcuts can I use for linking?

See Writeup Formatting Tips and other pages linked from there for more info.