fix tests for clear_flash (mocking was bad)

[catagits/Catalyst-Plugin-Session.git] / t / 03_flash.t

diff --git a/t/03_flash.t b/t/03_flash.t
index edbf607..ca9b886 100644 (file)
--- a/t/03_flash.t
+++ b/t/03_flash.t
@@ -3,40 +3,63 @@
 use strict;
 use warnings;
 
-use Test::More tests => 6;
+use Test::More tests => 8;
 use Test::MockObject::Extends;
 use Test::Exception;
 
 my $m;
 BEGIN { use_ok( $m = "Catalyst::Plugin::Session" ) }
 
-my $c = Test::MockObject::Extends->new( $m );
+my $c = Test::MockObject::Extends->new($m);
 
-$c->set_always( get_session_data => { } );
-$c->set_true( "store_session_data" );
-$c->set_always( _sessionid => "deadbeef");
+my $flash = {};
+$c->mock(
+    get_session_data => sub {
+        my ( $c, $key ) = @_;
+        return $key =~ /expire/ ? time() + 1000 : $flash;
+    },
+);
+$c->mock("store_session_data" => sub { $flash = $_[2] });
+$c->mock("delete_session_data" => sub { $flash = {} });
+$c->set_always( _sessionid => "deadbeef" );
+$c->set_always( config     => { session => { expires => 1000 } } );
+$c->set_always( stash      => {} );
 
-$c->_load_flash;
-
-is_deeply( $c->flash, {}, "nothing in flash");
+is_deeply( $c->flash, {}, "nothing in flash" );
 
 $c->flash->{foo} = "moose";
 
 $c->finalize;
-$c->_load_flash;
 
 is_deeply( $c->flash, { foo => "moose" }, "one key in flash" );
 
 $c->flash->{bar} = "gorch";
 
-is_deeply( $c->flash, { foo => "moose", bar => "gorch" }, "two keys in flash");
+is_deeply( $c->flash, { foo => "moose", bar => "gorch" }, "two keys in flash" );
 
 $c->finalize;
-$c->_load_flash;
 
 is_deeply( $c->flash, { bar => "gorch" }, "one key in flash" );
 
 $c->finalize;
-$c->_load_flash;
 
-is_deeply( $c->flash, {}, "nothing in flash");
+$c->flash->{test} = 'clear_flash';
+
+$c->finalize;
+
+$c->clear_flash();
+
+is_deeply( $c->flash, {}, "nothing in flash after clear_flash" );
+
+$c->finalize;
+
+is_deeply( $c->flash, {}, "nothing in flash after finalize after clear_flash" );
+
+$c->flash->{bar} = "gorch";
+
+$c->config->{session}{flash_to_stash} = 1;
+
+$c->finalize;
+$c->prepare_action;
+
+is_deeply( $c->stash, { bar => "gorch" }, "flash copied to stash" );