#!/usr/bin/perl

# Load testing for prefork.pm

use strict;

BEGIN {

	$|  = 1;

	$^W = 1;

}

use Test::More tests => 19;

use prefork ();

ok( ! $prefork::FORKING, '$FORKING is false' );

# Prepare the things we need

my $notify1 = 0;

my $notify2 = 0;

sub _notify1 { $notify1++ }

sub _notify2 { $notify2++ }

# Start with some bad notify calls

eval { prefork::notify(undef) };

ok( $@, 'prefork::notify(undef) dies' );

ok( $@ =~ /prefork::notify was not passed a CODE reference/, 'die message matches expected' );

eval { prefork::notify('foo') };

ok( $@, 'prefork::notify("foo") dies' );

ok( $@ =~ /prefork::notify was not passed a CODE reference/, 'die message matches expected' );

eval { prefork::notify(\"foo") };

ok( $@, 'prefork::notify(\"foo") dies' );

ok( $@ =~ /prefork::notify was not passed a CODE reference/, 'die message matches expected' );

# Give notify the valid function ref

ok( prefork::notify( \&_notify1 ), 'prefork::notify(\&func) returns true' );

is( $notify1, 0, 'prefork::notify in non-FORKING context does not call callback' );

eval { prefork::notify(\&_notify1) };

ok( $@, 'Duplicate prefork::notify(\&func) call dies' );

ok( $@ =~ /Callback function already registered/, 'die message matches expected' );

# Call ->enable and see if the callback is called

ok( prefork::enable(), 'prefork::enable returns true' );

is( $notify1, 1, 'Callback appears to be executed correctly' );

ok( prefork::enable(), 'prefork::enable returns true' );

is( $notify1, 1, 'Callback is not called again' );

# Pass the second callback, which should be called immediately

ok( prefork::notify( \&_notify2 ), 'prefork::notify(\&func2) returns true' );

is( $notify2, 1, 'prefork::notify in FORKING context does call callback' );

ok( prefork::notify( \&_notify2 ), 'prefork::notify(\&func2) returns true' );

is( $notify2, 2, 'prefork::notify in FORKING context calls callback again' );