mongrel2 0.0.1

data/spec/mongrel2/httpresponse_spec.rb

@@ -0,0 +1,232 @@
+#!/usr/bin/env ruby
+
+BEGIN {
+	require 'pathname'
+	basedir = Pathname.new( __FILE__ ).dirname.parent.parent
+
+	libdir = basedir + "lib"
+
+	$LOAD_PATH.unshift( basedir ) unless $LOAD_PATH.include?( basedir )
+	$LOAD_PATH.unshift( libdir ) unless $LOAD_PATH.include?( libdir )
+}
+
+require 'rspec'
+
+require 'spec/lib/helpers'
+
+require 'mongrel2'
+require 'mongrel2/httprequest'
+require 'mongrel2/httpresponse'
+
+
+#####################################################################
+###	C O N T E X T S
+#####################################################################
+
+describe Mongrel2::HTTPResponse do
+
+	before( :all ) do
+		setup_logging( :fatal )
+	end
+
+	before( :each ) do
+		@response = Mongrel2::HTTPResponse.new( TEST_UUID, 299 )
+	end
+
+	after( :all ) do
+		reset_logging()
+	end
+
+
+	it "has a headers table" do
+		@response.headers.should be_a( Mongrel2::Table )
+	end
+
+	it "is a status 200 response if not set otherwise" do
+		@response.status_line.should == 'HTTP/1.1 200 OK'
+	end
+
+	it "sets Date and Content-length headers automatically if they haven't been set" do
+		@response.header_data.should =~ /Content-length: 0/i
+		@response.header_data.should =~ /Date: #{HTTP_DATE}/i
+	end
+
+	it "doesn't have a body" do
+		@response.body.should be_empty()
+	end
+
+	it "knows it hasn't been handled" do
+		@response.should_not be_handled()
+	end
+
+	it "stringifies to a valid RFC2616 response string" do
+		@response.to_s.should =~ HTTP_RESPONSE
+	end
+
+	it "has some default headers" do
+		@response.headers['Server'].should == Mongrel2.version_string( true )
+	end
+
+	it "can be reset to a pristine state" do
+		@response.body << "Some stuff we want to get rid of later"
+		@response.headers['x-lunch-packed-by'] = 'Your Mom'
+		@response.status = HTTP::OK
+
+		@response.reset
+
+		@response.should_not be_handled()
+		@response.body.should == ''
+		@response.headers.should have(1).keys
+	end
+
+
+	it "can find the length of its body if it's a String" do
+		test_body = 'A string full of stuff'
+		@response.body = test_body
+
+		@response.get_content_length.should == test_body.length
+	end
+
+	it "can find the length of its body if it's a seekable IO" do
+		test_body = File.open( __FILE__, 'r' )
+		test_body.seek( 0, IO::SEEK_END )
+		length = test_body.tell
+		test_body.seek( 0, IO::SEEK_SET )
+
+		@response.body = test_body
+
+		@response.get_content_length.should == length
+	end
+
+	it "can find the length of its body even if it's an IO that's been set to do a partial read" do
+		test_body = File.open( __FILE__, 'r' )
+		test_body.seek( 0, IO::SEEK_END )
+		length = test_body.tell
+		test_body.seek( 100, IO::SEEK_SET )
+
+		@response.body = test_body
+
+		@response.get_content_length.should == length - 100
+	end
+
+	it "knows that it has been handled even if the status is set to NOT_FOUND" do
+		@response.status = HTTP::NOT_FOUND
+		@response.should be_handled()
+	end
+
+	it "knows if it has not yet been handled" do
+		@response.should_not be_handled()
+		@response.status = HTTP::OK
+		@response.should be_handled()
+	end
+
+
+	it "knows what category of response it is" do
+		@response.status = HTTP::CREATED
+		@response.status_category.should == 2
+
+		@response.status = HTTP::NOT_ACCEPTABLE
+		@response.status_category.should == 4
+	end
+
+
+	it "knows if its status indicates it is an informational response" do
+		@response.status = HTTP::PROCESSING
+		@response.status_category.should == 1
+		@response.status_is_informational?.should == true
+	end
+
+
+	it "knows if its status indicates it is a successful response" do
+		@response.status = HTTP::ACCEPTED
+		@response.status_category.should == 2
+		@response.status_is_successful?.should == true
+	end
+
+
+	it "knows if its status indicates it is a redirected response" do
+		@response.status = HTTP::SEE_OTHER
+		@response.status_category.should == 3
+		@response.status_is_redirect?.should == true
+	end
+
+
+	it "knows if its status indicates there was a client error" do
+		@response.status = HTTP::GONE
+		@response.status_category.should == 4
+		@response.status_is_clienterror?.should == true
+	end
+
+
+	it "knows if its status indicates there was a server error" do
+		@response.status = HTTP::VERSION_NOT_SUPPORTED
+		@response.status_category.should == 5
+		@response.status_is_servererror?.should == true
+	end
+
+
+	it "knows what the response content type is" do
+		headers = mock( 'headers' )
+		@response.stub!( :headers ).and_return( headers )
+
+		headers.should_receive( :[] ).
+			with( :content_type ).
+			and_return( 'text/erotica' )
+
+		@response.content_type.should == 'text/erotica'
+	end
+
+
+	it "can modify the response content type" do
+		headers = mock( 'headers' )
+		@response.stub!( :headers ).and_return( headers )
+
+		headers.should_receive( :[]= ).
+			with( :content_type, 'image/nude' )
+
+		@response.content_type = 'image/nude'
+	end
+
+
+	it "can find the length of its body if it's an IO" do
+		test_body_content = 'A string with some stuff in it'
+		test_body = StringIO.new( test_body_content )
+		@response.body = test_body
+
+		@response.get_content_length.should == test_body_content.length
+	end
+
+
+	it "raises a descriptive error message if it can't get the body's length" do
+		@response.body = Object.new
+
+		lambda {
+			@response.get_content_length
+		}.should raise_error( Mongrel2::ResponseError, /content length/i )
+	end
+
+
+	it "can build a valid HTTP status line for its status" do
+		@response.status = HTTP::SEE_OTHER
+		@response.status_line.should == "HTTP/1.1 303 See Other"
+	end
+
+
+	it "has pipelining disabled by default" do
+		@response.should_not be_keepalive()
+	end
+
+
+	it "has pipelining disabled if it's explicitly disabled" do
+		@response.keepalive = false
+		@response.should_not be_keepalive()
+	end
+
+
+	it "can be set to allow pipelining" do
+		@response.keepalive = true
+		@response.should be_keepalive()
+	end
+
+end
+

data/spec/mongrel2/logging_spec.rb

@@ -0,0 +1,76 @@
+#!/usr/bin/env ruby
+
+BEGIN {
+	require 'pathname'
+	basedir = Pathname.new( __FILE__ ).dirname.parent.parent
+
+	libdir = basedir + "lib"
+
+	$LOAD_PATH.unshift( basedir ) unless $LOAD_PATH.include?( basedir )
+	$LOAD_PATH.unshift( libdir ) unless $LOAD_PATH.include?( libdir )
+}
+
+require 'rspec'
+require 'stringio'
+
+require 'spec/lib/helpers'
+
+require 'mongrel2'
+require 'mongrel2/logging'
+
+
+#####################################################################
+###	C O N T E X T S
+#####################################################################
+
+describe Mongrel2::Logging, "-extended module" do
+
+	before( :each ) do
+		@extended_module = Module.new do
+			extend Mongrel2::Logging
+		end
+	end
+
+	it "should have a default Logger" do
+		@extended_module.logger.should be_a( Logger )
+		@extended_module.default_logger.should equal( @extended_module.logger )
+	end
+
+	it "should know if its default logger is replaced" do
+		@extended_module.should be_using_default_logger
+		@extended_module.logger = Logger.new( $stderr )
+		@extended_module.should_not be_using_default_logger
+	end
+
+	it "has the default logger instance after being reset" do
+		@extended_module.reset_logger
+		@extended_module.logger.should equal( @extended_module.default_logger )
+	end
+
+	it "has the default log formatter instance after being reset" do
+		@extended_module.reset_logger
+		@extended_module.logger.formatter.should equal( @extended_module.default_log_formatter )
+	end
+
+
+	context "with new defaults" do
+
+		before( :each ) do
+			@sink = StringIO.new
+			@logger = Logger.new( @sink )
+			@formatter = Mongrel2::Logging::ColorFormatter.new( @logger )
+
+			@extended_module.default_logger = @logger
+			@extended_module.default_log_formatter = @formatter
+		end
+
+		it "uses the new defaults when the logging subsystem is reset" do
+			@extended_module.reset_logger
+			@extended_module.logger.should equal( @logger )
+		end
+
+	end
+
+
+end
+

data/spec/mongrel2/mixins_spec.rb

@@ -0,0 +1,62 @@
+#!/usr/bin/env ruby
+
+BEGIN {
+	require 'pathname'
+	basedir = Pathname.new( __FILE__ ).dirname.parent.parent
+
+	libdir = basedir + "lib"
+
+	$LOAD_PATH.unshift( basedir ) unless $LOAD_PATH.include?( basedir )
+	$LOAD_PATH.unshift( libdir ) unless $LOAD_PATH.include?( libdir )
+}
+
+require 'rspec'
+
+require 'spec/lib/helpers'
+
+require 'mongrel2'
+require 'mongrel2/mixins'
+
+
+#####################################################################
+###	C O N T E X T S
+#####################################################################
+
+describe Mongrel2, "mixins" do
+
+	describe Mongrel2::Loggable do
+
+		before(:each) do
+			@logfile = StringIO.new('')
+			Mongrel2.logger = Logger.new( @logfile )
+
+			@test_class = Class.new do
+				include Mongrel2::Loggable
+
+				def log_test_message( level, msg )
+					self.log.send( level, msg )
+				end
+
+				def logdebug_test_message( msg )
+					self.log_debug.debug( msg )
+				end
+			end
+			@obj = @test_class.new
+		end
+
+
+		it "is able to output to the log via its #log method" do
+			@obj.log_test_message( :debug, "debugging message" )
+			@logfile.rewind
+			@logfile.read.should =~ /debugging message/
+		end
+
+		it "is able to output to the log via its #log_debug method" do
+			@obj.logdebug_test_message( "sexydrownwatch" )
+			@logfile.rewind
+			@logfile.read.should =~ /sexydrownwatch/
+		end
+	end
+
+end
+

data/spec/mongrel2/request_spec.rb

@@ -0,0 +1,157 @@
+#!/usr/bin/env ruby
+
+BEGIN {
+	require 'pathname'
+	basedir = Pathname.new( __FILE__ ).dirname.parent.parent
+
+	libdir = basedir + "lib"
+
+	$LOAD_PATH.unshift( basedir ) unless $LOAD_PATH.include?( basedir )
+	$LOAD_PATH.unshift( libdir ) unless $LOAD_PATH.include?( libdir )
+}
+
+require 'rspec'
+require 'tnetstring'
+
+require 'spec/lib/helpers'
+
+require 'mongrel2'
+require 'mongrel2/request'
+
+
+#####################################################################
+###	C O N T E X T S
+#####################################################################
+
+describe Mongrel2::Request do
+
+	before( :all ) do
+		setup_logging( :fatal )
+	end
+
+	after( :all ) do
+		reset_logging()
+	end
+
+
+	it "can parse a request message" do
+
+		message = make_request()
+		req = Mongrel2::Request.parse( message )
+
+		req.should be_a( Mongrel2::Request )
+		req.sender_id.should == TEST_UUID
+		req.conn_id.should == TEST_ID
+
+		req.headers.should be_a( Mongrel2::Table )
+		req.headers['Host'].should == TEST_HEADERS['host']
+	end
+
+	it "can parse a request message with TNetstring headers" do
+
+		message = make_tn_request()
+		req = Mongrel2::Request.parse( message )
+
+		req.should be_a( Mongrel2::Request )
+		req.sender_id.should == TEST_UUID
+		req.conn_id.should == TEST_ID
+
+		req.headers.should be_a( Mongrel2::Table )
+		req.headers.host.should == TEST_HEADERS['host']
+	end
+
+	it "can parse a request message with a JSON body" do
+
+		message = make_json_request()
+		req = Mongrel2::Request.parse( message )
+
+		req.should be_a( Mongrel2::JSONRequest )
+		req.sender_id.should == TEST_UUID
+		req.conn_id.should == TEST_ID
+
+		req.headers.should be_a( Mongrel2::Table )
+		req.headers.path.should == TEST_JSON_PATH
+
+		req.data.should == TEST_JSON_BODY
+	end
+
+	it "raises an UnhandledMethodError with the name of the method for METHOD verbs that " +
+	   "don't look like HTTP ones" do
+
+		message = make_request( :headers => {'METHOD' => '!DIVULGE'} )
+		expect {
+			Mongrel2::Request.parse( message )
+		}.to raise_error( Mongrel2::UnhandledMethodError, /!DIVULGE/ )
+	end
+
+
+	describe "instances" do
+
+		before( :each ) do
+			message = make_json_request() # HTTPRequest overrides the #response method
+			@req = Mongrel2::Request.parse( message )
+		end
+
+		it "can return an appropriate response instance for themselves" do
+			result = @req.response
+			result.should be_a( Mongrel2::Response )
+			result.sender_id.should == @req.sender_id
+			result.conn_id.should == @req.conn_id
+		end
+
+	end
+
+
+	describe "framework support" do
+
+		before( :all ) do
+			@oldtypes = Mongrel2::Request.request_types
+			@original_default_proc = Mongrel2::Request.request_types.default_proc
+		end
+
+		before( :each ) do
+			Mongrel2::Request.request_types.default_proc = @original_default_proc
+			Mongrel2::Request.request_types.clear
+		end
+
+		after( :all ) do
+			Mongrel2::Request.request_types.default_proc = @original_default_proc
+			Mongrel2::Request.request_types.replace( @oldtypes )
+		end
+
+
+		it "includes a mechanism for overriding the default Request subclass" do
+			subclass = Class.new( Mongrel2::Request ) do
+				register_request_type self, :__default
+			end
+
+			Mongrel2::Request.subclass_for_method( 'GET' ).should == subclass
+			Mongrel2::Request.subclass_for_method( 'POST' ).should == subclass
+			Mongrel2::Request.subclass_for_method( 'JSON' ).should == subclass
+		end
+
+		it "includes a mechanism for overriding the Request subclass for a particular request " +
+		   "method" do
+			subclass = Class.new( Mongrel2::Request ) do
+				register_request_type self, :GET
+			end
+
+			Mongrel2::Request.subclass_for_method( 'GET' ).should == subclass
+			Mongrel2::Request.subclass_for_method( 'POST' ).should_not == subclass
+			Mongrel2::Request.subclass_for_method( 'JSON' ).should_not == subclass
+		end
+
+		it "clears any cached method -> subclass lookups when the default subclass changes" do
+			Mongrel2::Request.subclass_for_method( 'OPTIONS' ) # cache OPTIONS -> Mongrel2::Request
+
+			subclass = Class.new( Mongrel2::Request ) do
+				register_request_type self, :__default
+			end
+
+			Mongrel2::Request.subclass_for_method( 'OPTIONS' ).should == subclass
+		end
+
+	end
+
+end
+