X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=t%2Flib%2FTestApp%2FContext.pm;h=22b43b0902b6930ec01ee19a57f6a3e36082a2f6;hb=f7367fe2884ee17d55064722b9136f244e74f86d;hp=e3e8f757fc0a40a9c5069c55683727b56f3da332;hpb=8458a66a25f3bd41bc9b6f13a1cd59000a2da411;p=catagits%2FCatalyst-Runtime.git

diff --git a/t/lib/TestApp/Context.pm b/t/lib/TestApp/Context.pm
index e3e8f75..22b43b0 100644
--- a/t/lib/TestApp/Context.pm
+++ b/t/lib/TestApp/Context.pm
@@ -1,6 +1,7 @@
 package TestApp::Context;
 use Moose;
 extends 'Catalyst::Context'; 
+with 'Catalyst::TraitFor::Context::TestHeaders';
 
 if (eval { Class::MOP::load_class('CatalystX::LeakChecker'); 1 }) {
     with 'CatalystX::LeakChecker';
@@ -51,10 +52,5 @@ sub execute {
     return $c->SUPER::execute(@_);
 }
 
-after prepare_action => sub{
-    my $c = shift;
-    $c->res->header( 'X-Catalyst-Action' => $c->req->action );
-};
-
 1;