X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=lib%2FCatalyst%2FTest.pm;h=191c6374a0d798c655fa9548c9803b842cf671b2;hb=6db8aa1d6fcd8b008971db965bf8e3190216b77b;hp=cf05f5ff56e5d0bd1f0f66a51fab2158ccc5ca83;hpb=a1791811465a12062122c0bf4cb73c56a5d5fad2;p=catagits%2FCatalyst-Runtime.git diff --git a/lib/Catalyst/Test.pm b/lib/Catalyst/Test.pm index cf05f5f..191c637 100644 --- a/lib/Catalyst/Test.pm +++ b/lib/Catalyst/Test.pm @@ -225,10 +225,13 @@ sub local_request { my $app = shift; my $request = Catalyst::Utils::request(shift); - _customize_request($request, @_); + my %extra_env; + _customize_request($request, \%extra_env, @_); my $ret; - test_psgi app => $app, client => sub { $ret = shift->($request) }; + test_psgi + app => sub { $app->({ %{ $_[0] }, %extra_env }) }, + client => sub { $ret = shift->($request) }; return $ret; } @@ -303,11 +306,16 @@ sub remote_request { sub _customize_request { my $request = shift; + my $extra_env = shift; my $opts = pop(@_) || {}; $opts = {} unless ref($opts) eq 'HASH'; if ( my $host = exists $opts->{host} ? $opts->{host} : $default_host ) { $request->header( 'Host' => $host ); } + + if (my $extra = $opts->{extra_env}) { + @{ $extra_env }{keys %{ $extra }} = values %{ $extra }; + } } =head2 action_ok