move C::E::LWP to C::E::Test and make C::E::H::D a subclass of C::E::T
[catagits/Catalyst-Runtime.git] / lib / Catalyst / Engine / Test.pm
index 45da156..59a6246 100644 (file)
@@ -1,20 +1,44 @@
 package Catalyst::Engine::Test;
 
 use strict;
-use base 'Catalyst::Engine::CGI::NPH';
+use base 'Catalyst::Engine';
 
+use CGI::Simple::Cookie;
+use Class::Struct ();
+use HTTP::Headers::Util 'split_header_words';
 use HTTP::Request;
 use HTTP::Response;
 use IO::File;
 use URI;
 
+__PACKAGE__->mk_accessors(qw/lwp/);
+
+Class::Struct::struct 'Catalyst::Engine::Test::LWP' => {
+    request  => 'HTTP::Request',
+    response => 'HTTP::Response',
+    hostname => '$',
+    address  => '$'
+};
+
 =head1 NAME
 
 Catalyst::Engine::Test - Catalyst Test Engine
 
 =head1 SYNOPSIS
 
-See L<Catalyst>.
+A script using the Catalyst::Engine::Test module might look like:
+
+    #!/usr/bin/perl -w
+
+    BEGIN { 
+       $ENV{CATALYST_ENGINE} = 'Test';
+    }
+
+    use strict;
+    use lib '/path/to/MyApp/lib';
+    use MyApp;
+
+    MyApp->run('/a/path');
 
 =head1 DESCRIPTION
 
@@ -22,10 +46,184 @@ This is the Catalyst engine specialized for testing.
 
 =head1 OVERLOADED METHODS
 
-This class overloads some methods from C<Catalyst::Engine::CGI::NPH>.
+This class overloads some methods from C<Catalyst::Engine>.
 
 =over 4
 
+=item $c->finalize_headers
+
+=cut
+
+sub finalize_headers {
+    my $c = shift;
+
+    my $status   = $c->response->status || 200;
+    my $headers  = $c->response->headers;
+    my $response = HTTP::Response->new( $status, undef, $headers );
+
+    while ( my ( $name, $cookie ) = each %{ $c->response->cookies } ) {
+        my $cookie = CGI::Simple::Cookie->new(
+            -name    => $name,
+            -value   => $cookie->{value},
+            -expires => $cookie->{expires},
+            -domain  => $cookie->{domain},
+            -path    => $cookie->{path},
+            -secure  => $cookie->{secure} || 0
+        );
+
+        $response->header( 'Set-Cookie' => $cookie->as_string );
+    }
+
+    $c->lwp->response($response);
+}
+
+=item $c->finalize_output
+
+=cut
+
+sub finalize_output {
+    my $c = shift;
+    $c->lwp->response->content_ref( \$c->response->{output} );
+}
+
+=item $c->prepare_connection
+
+=cut
+
+sub prepare_connection {
+    my $c = shift;
+    $c->req->hostname( $c->lwp->hostname );
+    $c->req->address( $c->lwp->address );
+}
+
+=item $c->prepare_cookies
+
+=cut
+
+sub prepare_cookies {
+    my $c = shift;
+
+    if ( my $header = $c->request->headers->header('Cookie') ) {
+        $c->req->cookies( { CGI::Simple::Cookie->parse($header) } );
+    }
+}
+
+=item $c->prepare_headers
+
+=cut
+
+sub prepare_headers {
+    my $c = shift;
+    $c->req->method( $c->lwp->request->method );
+    $c->req->headers( $c->lwp->request->headers );
+}
+
+=item $c->prepare_parameters
+
+=cut
+
+sub prepare_parameters {
+    my $c = shift;
+
+    my @params  = ();
+    my $request = $c->lwp->request;
+
+    push( @params, $request->uri->query_form );
+
+    if ( $request->content_type eq 'application/x-www-form-urlencoded' ) {
+        my $uri = URI->new('http:');
+        $uri->query( $request->content );
+        push( @params, $uri->query_form );
+    }
+
+    if ( $request->content_type eq 'multipart/form-data' ) {
+
+        for my $part ( $request->parts ) {
+
+            my $disposition = $part->header('Content-Disposition');
+            my %parameters  = @{ ( split_header_words($disposition) )[0] };
+
+            if ( $parameters{filename} ) {
+
+                my $fh = IO::File->new_tmpfile;
+                $fh->write( $part->content ) or die $!;
+                $fh->seek( SEEK_SET, 0 ) or die $!;
+
+                $c->req->uploads->{ $parameters{filename} } = {
+                    fh   => $fh,
+                    size => ( stat $fh )[7],
+                    type => $part->content_type
+                };
+
+                push( @params, $parameters{filename}, $fh );
+            }
+            else {
+                push( @params, $parameters{name}, $part->content );
+            }
+        }
+    }
+
+    my $parameters = $c->req->parameters;
+
+    while ( my ( $name, $value ) = splice( @params, 0, 2 ) ) {
+
+        if ( exists $parameters->{$name} ) {
+            for ( $parameters->{$name} ) {
+                $_ = [$_] unless ref($_) eq "ARRAY";
+                push( @$_, $value );
+            }
+        }
+        else {
+            $parameters->{$name} = $value;
+        }
+    }
+}
+
+=item $c->prepare_path
+
+=cut
+
+sub prepare_path {
+    my $c = shift;
+
+    my $base;
+    {
+        my $scheme = $c->lwp->request->uri->scheme;
+        my $host   = $c->lwp->request->uri->host;
+        my $port   = $c->lwp->request->uri->port;
+
+        $base = URI->new;
+        $base->scheme($scheme);
+        $base->host($host);
+        $base->port($port);
+
+        $base = $base->canonical->as_string;
+    }
+
+    my $path = $c->lwp->request->uri->path || '/';
+    $path =~ s/^\///;
+
+    $c->req->base($base);
+    $c->req->path($path);
+}
+
+=item $c->prepare_request($r)
+
+=cut
+
+sub prepare_request {
+    my ( $c, $lwp ) = @_;
+    $c->lwp($lwp);
+}
+
+=item $c->prepare_uploads
+
+=cut
+
+sub prepare_uploads {
+    my $c = shift;
+}
+
 =item $c->run
 
 =cut
@@ -35,49 +233,27 @@ sub run {
     my $request = shift || '/';
 
     unless ( ref $request ) {
-        $request = URI->new( $request, 'http' );
+
+        my $uri = ( $request =~ m/http/i )
+          ? URI->new($request)
+          : URI->new( 'http://localhost' . $request );
+
+        $request = $uri->canonical;
     }
+
     unless ( ref $request eq 'HTTP::Request' ) {
         $request = HTTP::Request->new( 'GET', $request );
     }
 
-    local ( *STDIN, *STDOUT );
-
-    my %clean  = %ENV;
-    my $output = '';
-    $ENV{CONTENT_TYPE}   ||= $request->header('Content-Type')   || '';
-    $ENV{CONTENT_LENGTH} ||= $request->header('Content-Length') || '';
-    $ENV{GATEWAY_INTERFACE} ||= 'CGI/1.1';
-    $ENV{HTTP_USER_AGENT}   ||= 'Catalyst';
-    $ENV{HTTP_HOST}         ||= $request->uri->host || 'localhost';
-    $ENV{QUERY_STRING}      ||= $request->uri->query || '';
-    $ENV{REQUEST_METHOD}    ||= $request->method;
-    $ENV{PATH_INFO}         ||= $request->uri->path || '/';
-    $ENV{SCRIPT_NAME}       ||= '/';
-    $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);
-    }
+    my $lwp = Catalyst::Engine::Test::LWP->new(
+        request  => $request,
+        address  => '127.0.0.1',
+        hostname => 'localhost'
+    );
 
-    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: $!");
-    }
+    $class->handler($lwp);
 
-    open( STDOUT, '>', \$output );
-    $class->handler;
-    %ENV = %clean;
-    return HTTP::Response->parse($output);
+    return $lwp->response;
 }
 
 =back