TraitFor headers

[catagits/Catalyst-Runtime.git] / t / lib / TestApp / Context.pm

diff --git a/t/lib/TestApp/Context.pm b/t/lib/TestApp/Context.pm
index e3e8f75..22b43b0 100644 (file)
--- 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;