sub.t

use v6;

use Test;

=begin comment

This tests the &?ROUTINE magical value

=end comment

plan 10;

# L<S06/The C<&?ROUTINE> object>
# L<S02/Names/Which routine am I in>
sub factorial { @_[0] < 2 ?? 1 !! @_[0] * &?ROUTINE(@_[0] - 1) }

my $result1 = factorial(3);
is($result1, 6, 'the &?ROUTINE magical works correctly');

my $factorial = sub { @_[0] < 2 ?? 1 !! @_[0] * &?ROUTINE(@_[0] - 1) };
my $result2 = $factorial(3);
is($result2, 6, 'the &?ROUTINE magical works correctly in anon-subs');

sub postfix:<!!!> (Int $n) { $n < 2 ?? 1 !! $n * &?ROUTINE($n - 1) }
my $result3 = 3!!!;
is($result3, 6, 'the &?ROUTINE magical works correctly in overloaded operators' );

{
    my $variable;
    my regex foo { a { $variable = &?ROUTINE; } }
    my token bar { b { $variable = &?ROUTINE; } }
    my rule baz  { c { $variable = &?ROUTINE; } }
    "a" ~~ &foo;
    is $variable, &foo, '&?ROUTINE is correct inside a regex';
    "b" ~~ &bar;
    is $variable, &bar, '&?ROUTINE is correct inside a token';
    "c" ~~ &baz;
    is $variable, &baz, '&?ROUTINE is correct inside a rule';
}

#?rakudo.jvm skip 'java.lang.NullPointerException (openjdk)'
{
    my @collected;
    sub foo($a) {
        sub bar($n) {
            @collected.push($a);
            $n == 1 ?? 1 !! $n * &?ROUTINE($n - 1)
        }
    }
    my $r1 = foo('a');
    my $r2 = foo('b');
    is $r1(4), 24, 'Correct result from function generator returning function using &?ROUTINE';
    is @collected.join(''), 'aaaa', 'Correct closure semantics with &?ROUTINE';
}

{
    my @collected;
    sub foo($a) {
        sub bar($n) {
            my $what-to-do = {
                @collected.push($a);
                $n == 1 ?? 1 !! $n * &?ROUTINE($n - 1)
            }
            $what-to-do();
        }
    }
    my $r1 = foo('a');
    my $r2 = foo('b');
    is $r1(4), 24, 'Correct result using &?ROUTINE in nested closure';
    is @collected.join(''), 'aaaa', 'Correct closure semantics with &?ROUTINE in nested closure';
}

# vim: ft=perl6