metasploit-framework/lib/rex/proto/http/client.rb.ut.rb

#!/usr/bin/env ruby

$:.unshift(File.join(File.dirname(__FILE__), '..', '..', '..'))

require 'test/unit'
require 'rex/proto/http'

class Rex::Proto::Http::Client::UnitTest < Test::Unit::TestCase

	Klass = Rex::Proto::Http::Client

	def test_parse
		c = Klass.new('www.google.com')

		# Set request factory parameters
		c.config(
			'vhost'      => 'www.google.com',
			'user-agent' => 'Metasploit Framework/3.0',
			'proto'      => '1.1',
			'cookie'     => 'NoCookie=NotACookie'
		)

		# Set client parameters
		c.config(
			'max-data' => 1024 * 1024
		)

		#
		# Request the main web pagfe
		#
		r = c.request(
			'method' => 'GET',
			'uri'    => '/'
		)

		resp = c.send_request(r)

		assert_equal(200, resp.code)
		assert_equal('OK', resp.message)
		assert_equal('1.1', resp.proto)

		#
		# Request a file that does not exist
		#
		r = c.request(
			'method' => 'GET',
			'uri'    => '/NoFileHere.404'
		)

		resp = c.send_request(r)
		
		assert_equal(404, resp.code)
		assert_equal('Not Found', resp.message)
		assert_equal('1.1', resp.proto)
		
		
		#
		# Send a POST request that results in a 302
		#
		c = Klass.new('beta.microsoft.com')
		c.request_option('vhost', 'beta.microsoft.com')

		r = c.request(
			'method' => 'POST',
			'uri'    => '/',
			'data'   => 'var=val',
			'content-type' => 'application/x-www-form-urlencoded'
		)

		resp = c.send_request(r)

		assert_equal(302, resp.code)
		assert_equal('Object moved', resp.message)
		assert_equal('1.1', resp.proto)
	end

end