#!perl
use strict;
use warnings;
use lib 't/lib';
use Stomp_LogCalls;
use Test::More;
use Test::Deep;
use Test::Fatal;
use Data::Printer;
use Net::Stomp::Producer::Transactional;

my $p=Net::Stomp::Producer::Transactional->new({
    connection_builder => sub { return Stomp_LogCalls->new(@_) },
    servers => [ {
        hostname => 'test-host', port => 9999,
    } ],
});

my $first_call=1;
sub assert_message_sent {
    my $diag = pop;
    my @bodies = @_;

    cmp_deeply(\@Stomp_LogCalls::calls,
               bag(
                   $first_call ? (
                       [ 'new', ignore(), ignore() ],
                       [ 'connect', ignore(), ignore() ],
                   ) : (),
                   map { [
                       'send',
                       ignore(),
                       {
                           body  => $_,
                           destination => '/queue/foo',
                       },
                   ] } @bodies
               ),
               $diag)
        or note p @Stomp_LogCalls::calls;

    $first_call=0;
    @Stomp_LogCalls::calls=();
}

sub assert_nothing_sent {
    my ($diag) = @_;

    cmp_deeply(\@Stomp_LogCalls::calls,
               [],
               $diag)
        or note p @Stomp_LogCalls::calls;
}

sub send_message {
    my ($body) = @_;

    $p->send('/queue/foo',{},$body);
}

subtest 'direct send' => sub {
    send_message('11');
    assert_message_sent('11','direct send');
};

subtest 'transactional send' => sub {
    $p->txn_begin();
    send_message('21');
    assert_nothing_sent('nothing sent when in a transaction');

    $p->txn_commit();
    assert_message_sent('21','sent on commit');

    my $e = exception { $p->txn_commit() };
    isa_ok($e,'Net::Stomp::Producer::Exceptions::Transactional',
           q{can't commit without a transaction}); #'});

    $p->txn_begin();
    send_message('21');
    $p->txn_rollback();
    assert_nothing_sent('nothing sent on rollback');

    send_message('23');
    assert_message_sent('23','no transaction still means send directly');
};

subtest 'transactions are re-entrant' => sub {
    $p->txn_begin;
    send_message('41');
    assert_nothing_sent('nothing sent when in a transaction');

    $p->txn_begin;
    send_message('42');
    assert_nothing_sent('nothing sent when in 2 transactions');

    $p->txn_commit;
    assert_nothing_sent('begin twice, commit once: nothing sent');

    $p->txn_commit;
    assert_message_sent('41','42','all messages sent when commited as many times as begun');

    my $e = exception { $p->txn_commit() };
    isa_ok($e,'Net::Stomp::Producer::Exceptions::Transactional',
           q{can't commit without a transaction}); #'});
};

subtest 'txn_do' => sub {

    subtest 'simple' => sub {
        $p->txn_do(sub {
                            send_message('51');
                            assert_nothing_sent('nothing sent inside txn_do');
                        });
        assert_message_sent('51','message sent at the end of txn_do');
    };

    subtest 'exception handling' => sub {
        my $e = exception {
            $p->txn_do(sub {
                                send_message('52');
                                die "boom\n";
                            });
        };
        is($e,"boom\n",'exception is propagated');
        assert_nothing_sent('nothing sent when exception thrown');
    };

    subtest 'nested exception handling' => sub {
        my $e = exception {
            $p->txn_do(sub {
                                send_message('53');
                                eval {
                                    $p->txn_do(sub {
                                                        send_message('54');
                                                        die "boom\n";
                                                    });
                                };
                                send_message('55');
                            });
        };
        is($e,undef,'exception caught inside our code');
        assert_message_sent('53','55','inner block rolled back');
    };

};

done_testing();