1#!./perl -w
2
3# Tests for the coderef-in-@INC feature
4
5BEGIN {
6    chdir 't' if -d 't';
7    @INC = qw(. ../lib);
8}
9
10use File::Spec;
11
12require "test.pl";
13plan(tests => 45);
14
15my @tempfiles = ();
16
17sub get_temp_fh {
18    my $f = "DummyModule0000";
19    1 while -e ++$f;
20    push @tempfiles, $f;
21    open my $fh, ">$f" or die "Can't create $f: $!";
22    print $fh "package ".substr($_[0],0,-3).";\n1;\n";
23    print $fh $_[1] if @_ > 1;
24    close $fh or die "Couldn't close: $!";
25    open $fh, $f or die "Can't open $f: $!";
26    return $fh;
27}
28
29END { 1 while unlink @tempfiles }
30
31sub fooinc {
32    my ($self, $filename) = @_;
33    if (substr($filename,0,3) eq 'Foo') {
34	return get_temp_fh($filename);
35    }
36    else {
37        return undef;
38    }
39}
40
41push @INC, \&fooinc;
42
43my $evalret = eval { require Bar; 1 };
44ok( !$evalret,      'Trying non-magic package' );
45
46$evalret = eval { require Foo; 1 };
47die $@ if $@;
48ok( $evalret,                      'require Foo; magic via code ref'  );
49ok( exists $INC{'Foo.pm'},         '  %INC sees Foo.pm' );
50is( ref $INC{'Foo.pm'}, 'CODE',    '  val Foo.pm is a coderef in %INC' );
51is( $INC{'Foo.pm'}, \&fooinc,	   '  val Foo.pm is correct in %INC' );
52
53$evalret = eval "use Foo1; 1;";
54die $@ if $@;
55ok( $evalret,                      'use Foo1' );
56ok( exists $INC{'Foo1.pm'},        '  %INC sees Foo1.pm' );
57is( ref $INC{'Foo1.pm'}, 'CODE',   '  val Foo1.pm is a coderef in %INC' );
58is( $INC{'Foo1.pm'}, \&fooinc,     '  val Foo1.pm is correct in %INC' );
59
60$evalret = eval { do 'Foo2.pl'; 1 };
61die $@ if $@;
62ok( $evalret,                      'do "Foo2.pl"' );
63ok( exists $INC{'Foo2.pl'},        '  %INC sees Foo2.pl' );
64is( ref $INC{'Foo2.pl'}, 'CODE',   '  val Foo2.pl is a coderef in %INC' );
65is( $INC{'Foo2.pl'}, \&fooinc,     '  val Foo2.pl is correct in %INC' );
66
67pop @INC;
68
69
70sub fooinc2 {
71    my ($self, $filename) = @_;
72    if (substr($filename, 0, length($self->[1])) eq $self->[1]) {
73	return get_temp_fh($filename);
74    }
75    else {
76        return undef;
77    }
78}
79
80my $arrayref = [ \&fooinc2, 'Bar' ];
81push @INC, $arrayref;
82
83$evalret = eval { require Foo; 1; };
84die $@ if $@;
85ok( $evalret,                     'Originally loaded packages preserved' );
86$evalret = eval { require Foo3; 1; };
87ok( !$evalret,                    'Original magic INC purged' );
88
89$evalret = eval { require Bar; 1 };
90die $@ if $@;
91ok( $evalret,                     'require Bar; magic via array ref' );
92ok( exists $INC{'Bar.pm'},        '  %INC sees Bar.pm' );
93is( ref $INC{'Bar.pm'}, 'ARRAY',  '  val Bar.pm is an arrayref in %INC' );
94is( $INC{'Bar.pm'}, $arrayref,    '  val Bar.pm is correct in %INC' );
95
96ok( eval "use Bar1; 1;",          'use Bar1' );
97ok( exists $INC{'Bar1.pm'},       '  %INC sees Bar1.pm' );
98is( ref $INC{'Bar1.pm'}, 'ARRAY', '  val Bar1.pm is an arrayref in %INC' );
99is( $INC{'Bar1.pm'}, $arrayref,   '  val Bar1.pm is correct in %INC' );
100
101ok( eval { do 'Bar2.pl'; 1 },     'do "Bar2.pl"' );
102ok( exists $INC{'Bar2.pl'},       '  %INC sees Bar2.pl' );
103is( ref $INC{'Bar2.pl'}, 'ARRAY', '  val Bar2.pl is an arrayref in %INC' );
104is( $INC{'Bar2.pl'}, $arrayref,   '  val Bar2.pl is correct in %INC' );
105
106pop @INC;
107
108sub FooLoader::INC {
109    my ($self, $filename) = @_;
110    if (substr($filename,0,4) eq 'Quux') {
111	return get_temp_fh($filename);
112    }
113    else {
114        return undef;
115    }
116}
117
118my $href = bless( {}, 'FooLoader' );
119push @INC, $href;
120
121$evalret = eval { require Quux; 1 };
122die $@ if $@;
123ok( $evalret,                      'require Quux; magic via hash object' );
124ok( exists $INC{'Quux.pm'},        '  %INC sees Quux.pm' );
125is( ref $INC{'Quux.pm'}, 'FooLoader',
126				   '  val Quux.pm is an object in %INC' );
127is( $INC{'Quux.pm'}, $href,        '  val Quux.pm is correct in %INC' );
128
129pop @INC;
130
131my $aref = bless( [], 'FooLoader' );
132push @INC, $aref;
133
134$evalret = eval { require Quux1; 1 };
135die $@ if $@;
136ok( $evalret,                      'require Quux1; magic via array object' );
137ok( exists $INC{'Quux1.pm'},       '  %INC sees Quux1.pm' );
138is( ref $INC{'Quux1.pm'}, 'FooLoader',
139				   '  val Quux1.pm is an object in %INC' );
140is( $INC{'Quux1.pm'}, $aref,       '  val Quux1.pm  is correct in %INC' );
141
142pop @INC;
143
144my $sref = bless( \(my $x = 1), 'FooLoader' );
145push @INC, $sref;
146
147$evalret = eval { require Quux2; 1 };
148die $@ if $@;
149ok( $evalret,                      'require Quux2; magic via scalar object' );
150ok( exists $INC{'Quux2.pm'},       '  %INC sees Quux2.pm' );
151is( ref $INC{'Quux2.pm'}, 'FooLoader',
152				   '  val Quux2.pm is an object in %INC' );
153is( $INC{'Quux2.pm'}, $sref,       '  val Quux2.pm is correct in %INC' );
154
155pop @INC;
156
157push @INC, sub {
158    my ($self, $filename) = @_;
159    if (substr($filename,0,4) eq 'Toto') {
160	$INC{$filename} = 'xyz';
161	return get_temp_fh($filename);
162    }
163    else {
164        return undef;
165    }
166};
167
168$evalret = eval { require Toto; 1 };
169die $@ if $@;
170ok( $evalret,                      'require Toto; magic via anonymous code ref'  );
171ok( exists $INC{'Toto.pm'},        '  %INC sees Toto.pm' );
172ok( ! ref $INC{'Toto.pm'},         q/  val Toto.pm isn't a ref in %INC/ );
173is( $INC{'Toto.pm'}, 'xyz',	   '  val Toto.pm is correct in %INC' );
174
175pop @INC;
176
177push @INC, sub {
178    my ($self, $filename) = @_;
179    if ($filename eq 'abc.pl') {
180	return get_temp_fh($filename, qq(return "abc";\n));
181    }
182    else {
183	return undef;
184    }
185};
186
187$ret = "";
188$ret ||= do 'abc.pl';
189is( $ret, 'abc', 'do "abc.pl" sees return value' );
190
191pop @INC;
192
193my $filename = $^O eq 'MacOS' ? ':Foo:Foo.pm' : './Foo.pm';
194{
195    local @INC;
196    @INC = sub { $filename = 'seen'; return undef; };
197    eval { require $filename; };
198    is( $filename, 'seen', 'the coderef sees fully-qualified pathnames' );
199}
200