#!/usr/local/bin/perl -w
my @custom_inc;
BEGIN {
    if ($ENV{PERL_CORE}) {
        chdir 't' if -d 't';
        @custom_inc = @INC = '../lib';
    } elsif (!grep /blib/, @INC) {
        chdir 't' if -d 't';
        unshift @INC, (@custom_inc = ('../blib/lib', '../blib/arch'));
    }
}

BEGIN {delete $ENV{THREADS_DEBUG}} # no debugging during testing!

# "Unpatch" Test::More, who internally tries to disable threads
BEGIN {
    no warnings 'redefine';
    if ($] < 5.008001) {
        require forks::shared::global_filter;
        import forks::shared::global_filter 'Test::Builder';
        require Test::Builder;
        *Test::Builder::share = \&threads::shared::share;
        *Test::Builder::lock = \&threads::shared::lock;
        Test::Builder->new->reset;
    }
}

# Patch Test::Builder to add fork-thread awareness
{
    no warnings 'redefine';
    my $_sanity_check_old = \&Test::Builder::_sanity_check;
    *Test::Builder::_sanity_check = sub {
        my $self = $_[0];
        # Don't bother with an ending if this is a forked copy.  Only the parent
        # should do the ending.
        if( $self->{Original_Pid} != $$ ) {
            return;
        }
        $_sanity_check_old->(@_);
    };
}

use Test::More tests => 4;
use Config;
use strict;
use warnings;

diag( <<EOD );

These tests validate main thread exit values.

EOD

my $libs;
if (@custom_inc) {
    $libs = '"-Mlib='.join(',', @custom_inc).'"';
} else {
    $libs = '"-Mlib='.join(',', ('blib/lib', 'blib/arch')).'"';
}
my $desired_exit_val = 42;

my $secure_perl_path = $Config{perlpath};
if ($^O ne 'VMS') {
    $secure_perl_path .= $Config{_exe}
        unless $secure_perl_path =~ m/$Config{_exe}$/i;
}

my $cmd = qq{$secure_perl_path $libs -e '}
    .q|BEGIN {delete $ENV{THREADS_DEBUG}; delete $ENV{THREADS_DAEMON_MODEL};}|
    .qq{ use forks; exit($desired_exit_val);'};
my $cmd2 = qq{$secure_perl_path $libs -e '}
    .q|BEGIN {delete $ENV{THREADS_DEBUG}; $ENV{THREADS_DAEMON_MODEL} = 1;}|
    .qq{ use forks; exit($desired_exit_val);'};
my $cmd3 = qq{$secure_perl_path $libs -e '}
    .q|BEGIN {delete $ENV{THREADS_DEBUG}; delete $ENV{THREADS_DAEMON_MODEL};}|
    .qq{ use forks; threads->new(sub { exit($desired_exit_val);} )->join(); sleep 10; sleep 10;'};
my $cmd4 = qq{$secure_perl_path $libs -e '}
    .q|BEGIN {delete $ENV{THREADS_DEBUG}; $ENV{THREADS_DAEMON_MODEL} = 1;}|
    .qq{ use forks; threads->new(sub { exit($desired_exit_val);} )->join(); sleep 10; sleep 10;'};

my $exit_val = system($cmd) >> 8;
cmp_ok($exit_val, '==', $desired_exit_val, 'Check that perl exit value is correct with forks');
$exit_val = system($cmd2) >> 8;
cmp_ok($exit_val, '==', $desired_exit_val, 'Check that perl exit value is correct with forks');
SKIP: { #TODO perl 5.6 compatibility, unclear why exit() is not handled (possibly a signal issue)
    skip 'Case not  supported in perl 5.6 (yet)', 1;
    $exit_val = system($cmd3) >> 8;
    cmp_ok($exit_val, '==', $desired_exit_val, 'Check that perl exit value is correct with forks');
}
$exit_val = system($cmd4) >> 8;
cmp_ok($exit_val, '==', $desired_exit_val, 'Check that perl exit value is correct with forks');

1;