X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?p=catagits%2FCatalyst-Runtime.git;a=blobdiff_plain;f=t%2Flib%2FTestApp%2FView%2FDump.pm;h=016df8116ae8d0b32da943a3c64e4654ef085837;hp=43cc976a1a10280dff16a104538f7d17e5dd3c82;hb=82010ea176741c7a4f2baf3f6f27377b1d9f6b15;hpb=0eb98ebd1624e8181a4bd88c26605f2a0f1c91d7

diff --git a/t/lib/TestApp/View/Dump.pm b/t/lib/TestApp/View/Dump.pm
index 43cc976..016df81 100644
--- a/t/lib/TestApp/View/Dump.pm
+++ b/t/lib/TestApp/View/Dump.pm
@@ -39,8 +39,10 @@ sub process {
     # Remove context from reference if needed
     my $context = delete $reference->{_context};
 
+    my $env = delete $reference->{env};
+
     if (my $log = $reference->{_log}) {
-        $log->clear_psgienv if $log->can('psgienv');
+        $log->clear_psgi if ($log->can('psgienv'));
     }
 
     if ( my $output =
@@ -62,6 +64,8 @@ sub process {
             $reference->{_body} = $body;
         }
 
+        if($env) { $reference->{env} = $env }
+
         return 1;
     }