t/op/inccode.t

   1 #!./perl -w
   2
   3 # Tests for the coderef-in-@INC feature
   4
   5 BEGIN {
   6     chdir 't' if -d 't';
   7     @INC = qw(. ../lib);
   8     require './test.pl';
   9 }
  10
  11 use Config;
  12
  13 my $can_fork   = 0;
  14 my $has_perlio = $Config{useperlio};
  15
  16 unless (is_miniperl()) {
  17     if ($Config{d_fork} && eval 'require POSIX; 1') {
  18         $can_fork = 1;
  19     }
  20 }
  21
  22 use strict;
  23
  24 plan(tests => 49 + !is_miniperl() * (3 + 14 * $can_fork));
  25
  26 sub get_temp_fh {
  27     my $f = tempfile();
  28     open my $fh, ">$f" or die "Can't create $f: $!";
  29     print $fh "package ".substr($_[0],0,-3).";\n1;\n";
  30     print $fh $_[1] if @_ > 1;
  31     close $fh or die "Couldn't close: $!";
  32     open $fh, $f or die "Can't open $f: $!";
  33     return $fh;
  34 }
  35
  36 sub fooinc {
  37     my ($self, $filename) = @_;
  38     if (substr($filename,0,3) eq 'Foo') {
  39         return get_temp_fh($filename);
  40     }
  41     else {
  42         return undef;
  43     }
  44 }
  45
  46 push @INC, \&fooinc;
  47
  48 my $evalret = eval { require Bar; 1 };
  49 ok( !$evalret,      'Trying non-magic package' );
  50
  51 $evalret = eval { require Foo; 1 };
  52 die $@ if $@;
  53 ok( $evalret,                      'require Foo; magic via code ref'  );
  54 ok( exists $INC{'Foo.pm'},         '  %INC sees Foo.pm' );
  55 is( ref $INC{'Foo.pm'}, 'CODE',    '  val Foo.pm is a coderef in %INC' );
  56 is( $INC{'Foo.pm'}, \&fooinc,      '  val Foo.pm is correct in %INC' );
  57
  58 $evalret = eval "use Foo1; 1;";
  59 die $@ if $@;
  60 ok( $evalret,                      'use Foo1' );
  61 ok( exists $INC{'Foo1.pm'},        '  %INC sees Foo1.pm' );
  62 is( ref $INC{'Foo1.pm'}, 'CODE',   '  val Foo1.pm is a coderef in %INC' );
  63 is( $INC{'Foo1.pm'}, \&fooinc,     '  val Foo1.pm is correct in %INC' );
  64
  65 $evalret = eval { do 'Foo2.pl'; 1 };
  66 die $@ if $@;
  67 ok( $evalret,                      'do "Foo2.pl"' );
  68 ok( exists $INC{'Foo2.pl'},        '  %INC sees Foo2.pl' );
  69 is( ref $INC{'Foo2.pl'}, 'CODE',   '  val Foo2.pl is a coderef in %INC' );
  70 is( $INC{'Foo2.pl'}, \&fooinc,     '  val Foo2.pl is correct in %INC' );
  71
  72 pop @INC;
  73
  74
  75 sub fooinc2 {
  76     my ($self, $filename) = @_;
  77     if (substr($filename, 0, length($self->[1])) eq $self->[1]) {
  78         return get_temp_fh($filename);
  79     }
  80     else {
  81         return undef;
  82     }
  83 }
  84
  85 my $arrayref = [ \&fooinc2, 'Bar' ];
  86 push @INC, $arrayref;
  87
  88 $evalret = eval { require Foo; 1; };
  89 die $@ if $@;
  90 ok( $evalret,                     'Originally loaded packages preserved' );
  91 $evalret = eval { require Foo3; 1; };
  92 ok( !$evalret,                    'Original magic INC purged' );
  93
  94 $evalret = eval { require Bar; 1 };
  95 die $@ if $@;
  96 ok( $evalret,                     'require Bar; magic via array ref' );
  97 ok( exists $INC{'Bar.pm'},        '  %INC sees Bar.pm' );
  98 is( ref $INC{'Bar.pm'}, 'ARRAY',  '  val Bar.pm is an arrayref in %INC' );
  99 is( $INC{'Bar.pm'}, $arrayref,    '  val Bar.pm is correct in %INC' );
 100
 101 ok( eval "use Bar1; 1;",          'use Bar1' );
 102 ok( exists $INC{'Bar1.pm'},       '  %INC sees Bar1.pm' );
 103 is( ref $INC{'Bar1.pm'}, 'ARRAY', '  val Bar1.pm is an arrayref in %INC' );
 104 is( $INC{'Bar1.pm'}, $arrayref,   '  val Bar1.pm is correct in %INC' );
 105
 106 ok( eval { do 'Bar2.pl'; 1 },     'do "Bar2.pl"' );
 107 ok( exists $INC{'Bar2.pl'},       '  %INC sees Bar2.pl' );
 108 is( ref $INC{'Bar2.pl'}, 'ARRAY', '  val Bar2.pl is an arrayref in %INC' );
 109 is( $INC{'Bar2.pl'}, $arrayref,   '  val Bar2.pl is correct in %INC' );
 110
 111 pop @INC;
 112
 113 sub FooLoader::INC {
 114     my ($self, $filename) = @_;
 115     if (substr($filename,0,4) eq 'Quux') {
 116         return get_temp_fh($filename);
 117     }
 118     else {
 119         return undef;
 120     }
 121 }
 122
 123 my $href = bless( {}, 'FooLoader' );
 124 push @INC, $href;
 125
 126 $evalret = eval { require Quux; 1 };
 127 die $@ if $@;
 128 ok( $evalret,                      'require Quux; magic via hash object' );
 129 ok( exists $INC{'Quux.pm'},        '  %INC sees Quux.pm' );
 130 is( ref $INC{'Quux.pm'}, 'FooLoader',
 131                                    '  val Quux.pm is an object in %INC' );
 132 is( $INC{'Quux.pm'}, $href,        '  val Quux.pm is correct in %INC' );
 133
 134 pop @INC;
 135
 136 my $aref = bless( [], 'FooLoader' );
 137 push @INC, $aref;
 138
 139 $evalret = eval { require Quux1; 1 };
 140 die $@ if $@;
 141 ok( $evalret,                      'require Quux1; magic via array object' );
 142 ok( exists $INC{'Quux1.pm'},       '  %INC sees Quux1.pm' );
 143 is( ref $INC{'Quux1.pm'}, 'FooLoader',
 144                                    '  val Quux1.pm is an object in %INC' );
 145 is( $INC{'Quux1.pm'}, $aref,       '  val Quux1.pm  is correct in %INC' );
 146
 147 pop @INC;
 148
 149 my $sref = bless( \(my $x = 1), 'FooLoader' );
 150 push @INC, $sref;
 151
 152 $evalret = eval { require Quux2; 1 };
 153 die $@ if $@;
 154 ok( $evalret,                      'require Quux2; magic via scalar object' );
 155 ok( exists $INC{'Quux2.pm'},       '  %INC sees Quux2.pm' );
 156 is( ref $INC{'Quux2.pm'}, 'FooLoader',
 157                                    '  val Quux2.pm is an object in %INC' );
 158 is( $INC{'Quux2.pm'}, $sref,       '  val Quux2.pm is correct in %INC' );
 159
 160 pop @INC;
 161
 162 push @INC, sub {
 163     my ($self, $filename) = @_;
 164     if (substr($filename,0,4) eq 'Toto') {
 165         $INC{$filename} = 'xyz';
 166         return get_temp_fh($filename);
 167     }
 168     else {
 169         return undef;
 170     }
 171 };
 172
 173 $evalret = eval { require Toto; 1 };
 174 die $@ if $@;
 175 ok( $evalret,                      'require Toto; magic via anonymous code ref'  );
 176 ok( exists $INC{'Toto.pm'},        '  %INC sees Toto.pm' );
 177 ok( ! ref $INC{'Toto.pm'},         q/  val Toto.pm isn't a ref in %INC/ );
 178 is( $INC{'Toto.pm'}, 'xyz',        '  val Toto.pm is correct in %INC' );
 179
 180 pop @INC;
 181
 182 push @INC, sub {
 183     my ($self, $filename) = @_;
 184     if ($filename eq 'abc.pl') {
 185         return get_temp_fh($filename, qq(return "abc";\n));
 186     }
 187     else {
 188         return undef;
 189     }
 190 };
 191
 192 my $ret = "";
 193 $ret ||= do 'abc.pl';
 194 is( $ret, 'abc', 'do "abc.pl" sees return value' );
 195
 196 {
 197     my $filename = './Foo.pm';
 198     #local @INC; # local fails on tied @INC
 199     my @old_INC = @INC; # because local doesn't work on tied arrays
 200     @INC = sub { $filename = 'seen'; return undef; };
 201     eval { require $filename; };
 202     is( $filename, 'seen', 'the coderef sees fully-qualified pathnames' );
 203     @INC = @old_INC;
 204 }
 205
 206 # this will segfault if it fails
 207
 208 sub PVBM () { 'foo' }
 209 { my $dummy = index 'foo', PVBM }
 210
 211 # I don't know whether these requires should succeed or fail. 5.8 failed
 212 # all of them; 5.10 with an ordinary constant in place of PVBM lets the
 213 # latter two succeed. For now I don't care, as long as they don't
 214 # segfault :).
 215
 216 unshift @INC, sub { PVBM };
 217 eval 'require foo';
 218 ok( 1, 'returning PVBM doesn\'t segfault require' );
 219 eval 'use foo';
 220 ok( 1, 'returning PVBM doesn\'t segfault use' );
 221 shift @INC;
 222 unshift @INC, sub { \PVBM };
 223 eval 'require foo';
 224 ok( 1, 'returning PVBM ref doesn\'t segfault require' );
 225 eval 'use foo';
 226 ok( 1, 'returning PVBM ref doesn\'t segfault use' );
 227 shift @INC;
 228
 229 exit if is_miniperl();
 230
 231 SKIP: {
 232     skip( "No PerlIO available", 3 ) unless $has_perlio;
 233     pop @INC;
 234
 235     push @INC, sub {
 236         my ($cr, $filename) = @_;
 237         my $module = $filename; $module =~ s,/,::,g; $module =~ s/\.pm$//;
 238         open my $fh, '<',
 239              \"package $module; sub complain { warn q() }; \$::file = __FILE__;"
 240             or die $!;
 241         $INC{$filename} = "/custom/path/to/$filename";
 242         return $fh;
 243     };
 244
 245     require Publius::Vergilius::Maro;
 246     is( $INC{'Publius/Vergilius/Maro.pm'},
 247         '/custom/path/to/Publius/Vergilius/Maro.pm', '%INC set correctly');
 248     is( our $file, '/custom/path/to/Publius/Vergilius/Maro.pm',
 249         '__FILE__ set correctly' );
 250     {
 251         my $warning;
 252         local $SIG{__WARN__} = sub { $warning = shift };
 253         Publius::Vergilius::Maro::complain();
 254         like( $warning, qr{something's wrong at /custom/path/to/Publius/Vergilius/Maro.pm}, 'warn() reports correct file source' );
 255     }
 256 }
 257 pop @INC;
 258
 259 if ($can_fork) {
 260     require PerlIO::scalar;
 261     # This little bundle of joy generates n more recursive use statements,
 262     # with each module chaining the next one down to 0. If it works, then we
 263     # can safely nest subprocesses
 264     my $use_filter_too;
 265     push @INC, sub {
 266         return unless $_[1] =~ /^BBBLPLAST(\d+)\.pm/;
 267         my $pid = open my $fh, "-|";
 268         if ($pid) {
 269             # Parent
 270             return $fh unless $use_filter_too;
 271             # Try filters and state in addition.
 272             return ($fh, sub {s/$_[1]/pass/; return}, "die")
 273         }
 274         die "Can't fork self: $!" unless defined $pid;
 275
 276         # Child
 277         my $count = $1;
 278         # Lets force some fun with odd sized reads.
 279         $| = 1;
 280         print 'push @main::bbblplast, ';
 281         print "$count;\n";
 282         if ($count--) {
 283             print "use BBBLPLAST$count;\n";
 284         }
 285         if ($use_filter_too) {
 286             print "die('In $_[1]');";
 287         } else {
 288             print "pass('In $_[1]');";
 289         }
 290         print '"Truth"';
 291         POSIX::_exit(0);
 292         die "Can't get here: $!";
 293     };
 294
 295     @::bbblplast = ();
 296     require BBBLPLAST5;
 297     is ("@::bbblplast", "0 1 2 3 4 5", "All ran");
 298
 299     foreach (keys %INC) {
 300         delete $INC{$_} if /^BBBLPLAST/;
 301     }
 302
 303     @::bbblplast = ();
 304     $use_filter_too = 1;
 305
 306     require BBBLPLAST5;
 307
 308     is ("@::bbblplast", "0 1 2 3 4 5", "All ran with a filter");
 309 }