foo

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

diff --git a/t/03_flash.t b/t/03_flash.t
index c37624e..0ed2242 100644 (file)
--- a/t/03_flash.t
+++ b/t/03_flash.t
@@ -19,8 +19,8 @@ $c->mock(
         return $key =~ /expire/ ? time() + 1000 : $flash;
     },
 );
-$c->set_true("store_session_data");
-$c->set_true("delete_session_data");
+$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      => {} );
@@ -33,7 +33,7 @@ $c->finalize;
 
 is_deeply( $c->flash, { foo => "moose" }, "one key in flash" );
 
-$c->flash->{bar} = "gorch";
+$c->flash(bar => "gorch");
 
 is_deeply( $c->flash, { foo => "moose", bar => "gorch" }, "two keys in flash" );