package Test::Exit;
use strict;
use warnings;
use Test::Builder;
use base qw( Exporter );
our @EXPORT = qw( exit_ok exit_nok exit_with );

my $test = Test::Builder->new;

my $exit_callback = sub {};

BEGIN {
    *CORE::GLOBAL::exit = sub { $exit_callback->(@_) };
}


sub exit_with(&$$) {
    my ($coderef, $expected_exit_code, $message) = @_;
    my $exit_called = 0;
    my $exit_code   = 0;

    $exit_callback = sub {
        $exit_called = 1;
        $exit_code = shift
    };

    $coderef->();
    if ($exit_called) {
        if (! $test->ok( $expected_exit_code == $exit_code, $message)) {
            $test->diag("expected $expected_exit_code but received $exit_code");
        }
    }
    else {
        fail($message . " - exit not called");
    }
}
sub exit_ok(&$) {
    my ($coderef, $message) = @_;
    my $exit_called = 0;

    $exit_callback = sub { $exit_called = 1};
    $coderef->();
    $test->ok($exit_called == 1, $message);
}

sub exit_nok(&$) {
    my ($coderef, $message) = @_;
    my $exit_called = 0;

    $exit_callback = sub { $exit_called = 1};
    $coderef->();
    $test->ok($exit_called == 0, $message);
}



1;
=head1 NAME

 Test::Exit - Tests whether exit was called

=head2 exit_ok

 exit_ok {
     exit;
 }, 'assert that exit was called';


=head2 exit_nok

 exit_nok {
 }, 'assert that exit was not called';


=head2 exit_with

 exit_with {
     exit 2
 }, 2, 'expect exit(2)';

=cut