X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=t%2F03_flash.t;h=f58222c3d3d8e9dd34780d5240f2d56ad18a96dd;hb=2842d93853af927a543214ccbe17d659f5ecd181;hp=fd1fadf53db713a76c4f5360f174f81760656161;hpb=d3e0ff80d7571b537b69edc656263335040f3ee4;p=catagits%2FCatalyst-Plugin-Session.git

diff --git a/t/03_flash.t b/t/03_flash.t
index fd1fadf..f58222c 100644
--- a/t/03_flash.t
+++ b/t/03_flash.t
@@ -3,9 +3,10 @@
 use strict;
 use warnings;
 
-use Test::More tests => 7;
+use Test::More tests => 12;
 use Test::MockObject::Extends;
 use Test::Exception;
+use Test::Deep;
 
 my $m;
 BEGIN { use_ok( $m = "Catalyst::Plugin::Session" ) }
@@ -19,38 +20,56 @@ $c->mock(
         return $key =~ /expire/ ? time() + 1000 : $flash;
     },
 );
-$c->set_true("store_session_data");
-$c->set_true("delete_session_data");
+$c->mock("debug" => sub { 0 });
+$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      => {} );
 
+is_deeply( $c->session, {}, "nothing in session" );
+
 is_deeply( $c->flash, {}, "nothing in flash" );
 
 $c->flash->{foo} = "moose";
 
-$c->finalize;
+$c->finalize_body;
 
 is_deeply( $c->flash, { foo => "moose" }, "one key in flash" );
 
-$c->flash->{bar} = "gorch";
+cmp_deeply( $c->session, { __updated => re('^\d+$'), __flash => $c->flash }, "session has __flash with flash data" );
+
+$c->flash(bar => "gorch");
 
 is_deeply( $c->flash, { foo => "moose", bar => "gorch" }, "two keys in flash" );
 
-$c->finalize;
+cmp_deeply( $c->session, { __updated => re('^\d+$'), __flash => $c->flash }, "session still has __flash with flash data" );
+
+$c->finalize_body;
 
 is_deeply( $c->flash, { bar => "gorch" }, "one key in flash" );
 
-$c->finalize;
+$c->finalize_body;
 
-is_deeply( $c->flash, {}, "nothing in flash" );
+$c->flash->{test} = 'clear_flash';
+
+$c->finalize_body;
+
+$c->clear_flash();
+
+is_deeply( $c->flash, {}, "nothing in flash after clear_flash" );
+
+$c->finalize_body;
+
+is_deeply( $c->flash, {}, "nothing in flash after finalize after clear_flash" );
+
+cmp_deeply( $c->session, { __updated => re('^\d+$'), }, "session has empty __flash after clear_flash + finalize" );
 
 $c->flash->{bar} = "gorch";
 
 $c->config->{session}{flash_to_stash} = 1;
 
-$c->finalize;
+$c->finalize_body;
 $c->prepare_action;
 
 is_deeply( $c->stash, { bar => "gorch" }, "flash copied to stash" );
-