From: Sebastian Riedel <sri@labs.kraih.com>
Date: Sun, 20 Mar 2005 19:07:10 +0000 (+0000)
Subject: improved Catalyst::Test::request
X-Git-Tag: 5.7099_04~1753
X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?p=catagits%2FCatalyst-Runtime.git;a=commitdiff_plain;h=49faa307375117d01ccb6453642c0cf77244e66f

improved Catalyst::Test::request
---

diff --git a/Changes b/Changes
index 01bee69..958fb0b 100644
--- a/Changes
+++ b/Changes
@@ -2,6 +2,7 @@ This file documents the revision history for Perl extension Catalyst.
 
 4.30  XXX XXX XX XX:00:00 2005
         - added connection informations (Christian Hansen)
+        - HTTP::Request support in Catalyst::Test (Christian Hansen)
 
 4.28  Sat Mar 19 22:00:00 2005
         - fixed isa tree (Christian Hansen)
diff --git a/MANIFEST b/MANIFEST
index e26067b..c15290a 100644
--- a/MANIFEST
+++ b/MANIFEST
@@ -35,3 +35,4 @@ t/11stash.t
 t/12default.t
 t/13beginend.t
 t/14connection.t
+t/15post.t
diff --git a/lib/Catalyst/Engine/CGI.pm b/lib/Catalyst/Engine/CGI.pm
index e944526..1a9d04a 100644
--- a/lib/Catalyst/Engine/CGI.pm
+++ b/lib/Catalyst/Engine/CGI.pm
@@ -135,6 +135,8 @@ sub prepare_headers {
         ( my $field = $header ) =~ s/^HTTPS?_//;
         $c->req->headers->header( $field => $c->cgi->http($header) );
     }
+    $c->req->headers->header( 'Content-Type'   => $c->cgi->content_type );
+    $c->req->headers->header( 'Content-Length' => $c->cgi->content_length );
 }
 
 =item $c->prepare_parameters
diff --git a/lib/Catalyst/Test.pm b/lib/Catalyst/Test.pm
index fd53e18..1810aff 100644
--- a/lib/Catalyst/Test.pm
+++ b/lib/Catalyst/Test.pm
@@ -2,6 +2,8 @@ package Catalyst::Test;
 
 use strict;
 use UNIVERSAL::require;
+use IO::File;
+use HTTP::Request;
 use HTTP::Response;
 use Socket;
 use URI;
@@ -18,6 +20,11 @@ Catalyst::Test - Test Catalyst applications
 
 =head1 SYNOPSIS
 
+    # Helper
+    script/cgi-server.pl
+    script/server.pl
+    script/test.pl
+
     # Tests
     use Catalyst::Test 'TestApp';
     request('index.html');
@@ -74,17 +81,41 @@ sub import {
 }
 
 sub request {
-    my $uri = shift;
-    local *STDOUT;
-    my $output = '';
-    open STDOUT, '>', \$output;
-    $uri = URI->new($uri);
+    my $request = shift;
+    unless ( ref $request ) {
+        $request = URI->new( $request, 'http' );
+    }
+    unless ( ref $request eq 'HTTP::Request' ) {
+        $request = HTTP::Request->new( 'GET', $request );
+    }
+    local ( *STDIN, *STDOUT );
     my %clean = %ENV;
-    $ENV{REQUEST_METHOD} ||= 'GET';
-    $ENV{HTTP_HOST}      ||= $uri->authority || 'localhost';
-    $ENV{SCRIPT_NAME}    ||= $uri->path || '/';
-    $ENV{QUERY_STRING}   ||= $uri->query || '';
-    $ENV{CONTENT_TYPE}   ||= 'text/plain';
+    $ENV{CONTENT_TYPE}   ||= $request->header('Content-Type')   || '';
+    $ENV{CONTENT_LENGTH} ||= $request->header('Content-Length') || '';
+    $ENV{GATEWAY_INTERFACE} ||= 'CGI/1.1';
+    $ENV{HTTP_HOST}         ||= $request->uri->host || 'localhost';
+    $ENV{QUERY_STRING}      ||= $request->uri->query || '';
+    $ENV{REQUEST_METHOD}    ||= $request->method;
+    $ENV{SCRIPT_NAME}       ||= $request->uri->path || '/';
+    $ENV{SERVER_NAME}       ||= $request->uri->host || 'localhost';
+    $ENV{SERVER_PORT}       ||= $request->uri->port;
+    $ENV{SERVER_PROTOCOL}   ||= 'HTTP/1.1';
+
+    for my $field ( $request->header_field_names ) {
+        if ( $field =~ /^Content-(Length|Type)$/ ) {
+            next;
+        }
+        $field =~ s/-/_/g;
+        $ENV{ 'HTTP_' . uc($field) } = $request->header($field);
+    }
+    if ( $request->content_length ) {
+        my $body = IO::File->new_tmpfile;
+        $body->print( $request->content ) or die $!;
+        $body->seek( 0, SEEK_SET ) or die $!;
+        open( STDIN, "<&=", $body->fileno )
+          or die("Failed to dup \$body: $!");
+    }
+    open( STDOUT, '>', \$output );
     $class->handler;
     %ENV = %clean;
     return HTTP::Response->parse($output);
diff --git a/t/15post.t b/t/15post.t
new file mode 100644
index 0000000..459887c
--- /dev/null
+++ b/t/15post.t
@@ -0,0 +1,41 @@
+package TestApp;
+
+use Catalyst;
+
+__PACKAGE__->action(
+    echo => sub {
+        my ( $self, $c ) = @_;
+
+        for my $field ( $c->req->headers->header_field_names ) {
+            my $header = ( $field =~ /^X-/ ) ? $field : "X-$field";
+            $c->res->headers->header(
+                $header => $c->req->headers->header($field) );
+        }
+
+        $c->res->headers->content_type('text/plain');
+        $c->res->output('ok');
+    }
+);
+
+package main;
+
+use Test::More tests => 5;
+use Catalyst::Test 'TestApp';
+use HTTP::Request::Common;
+
+my $request = POST(
+    'http://localhost/echo',
+    'X-Whats-Cool' => 'Catalyst',
+    'Content-Type' => 'form-data',
+    'Content'      => [
+        catalyst => 'Rocks!',
+        file     => [$0],
+    ]
+);
+
+ok( my $response = request($request) );
+ok( $response->content_type eq 'text/plain' );
+ok( $response->headers->header('X-Content-Type') =~ /^multipart\/form-data/ );
+ok( $response->headers->header('X-Content-Length') ==
+      $request->content_length );
+ok( $response->headers->header('X-Whats-Cool') eq 'Catalyst' );