context creation fixed in t/aggregate/unit_core_uri_for_action.t

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

diff --git a/t/lib/TestApp/Context.pm b/t/lib/TestApp/Context.pm
index 3b09d79..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';
@@ -21,5 +22,35 @@ sub count_leaks {
     return scalar @{ $ctx->leaks };
 }
 
+sub execute {
+    my $c      = shift;
+    my $class  = ref( $c->component( $_[0] ) ) || $_[0];
+    my $action = $_[1]->reverse;
+
+    my $method;
+
+    if ( $action =~ /->(\w+)$/ ) {
+        $method = $1;
+    }
+    elsif ( $action =~ /\/(\w+)$/ ) {
+        $method = $1;
+    }
+    elsif ( $action =~ /^(\w+)$/ ) {
+        $method = $action;
+    }
+
+    if ( $class && $method && $method !~ /^_/ ) {
+        my $executed = sprintf( "%s->%s", $class, $method );
+        my @executed = $c->response->headers->header('X-Catalyst-Executed');
+        push @executed, $executed;
+        $c->response->headers->header(
+            'X-Catalyst-Executed' => join ', ',
+            @executed
+        );
+    }
+    no warnings 'recursion';
+    return $c->SUPER::execute(@_);
+}
+
 1;