Class: Net::HTTP

Inherits:

Object

• Object
• Net::HTTP

Defined in:

lib/mechanize/test_case.rb

Overview

:nodoc:

Constant Summary

PAGE_CACHE =

{}

Instance Method Summary

• #do_start ⇒ Object
• #old_do_start ⇒ Object
• #old_request ⇒ Object
• #request(req, *data) {|response| ... } ⇒ Object

Instance Method Details

#do_start ⇒ Object

# File 'lib/mechanize/test_case.rb', line 200

def do_start
  @started = true
end

#old_do_start ⇒ Object

# File 'lib/mechanize/test_case.rb', line 198

alias :old_do_start :do_start

#old_request ⇒ Object

# File 'lib/mechanize/test_case.rb', line 206

alias :old_request :request

#request(req, *data) {|response| ... } ⇒ Object

Yields:

• (response)

# File 'lib/mechanize/test_case.rb', line 208

def request(req, *data, &block)
  url = URI.parse(req.path)
  path = WEBrick::HTTPUtils.unescape(url.path)

  path = '/index.html' if path == '/'

  res = ::Response.new
  res.query_params = url.query

  req.query = if 'POST' != req.method && url.query then
                WEBrick::HTTPUtils.parse_query url.query
              elsif req['content-type'] =~ /www-form-urlencoded/ then
                WEBrick::HTTPUtils.parse_query req.body
              elsif req['content-type'] =~ /boundary=(.+)/ then
                boundary = WEBrick::HTTPUtils.dequote $1
                WEBrick::HTTPUtils.parse_form_data req.body, boundary
              else
                {}
              end

  req.cookies = WEBrick::Cookie.parse(req['Cookie'])

  Mechanize::TestCase::REQUESTS << req

  if servlet_klass = MECHANIZE_TEST_CASE_SERVLETS[path]
    servlet = servlet_klass.new({})
    servlet.send "do_#{req.method}", req, res
  else
    filename = "htdocs#{path.gsub(/[^\/\\.\w\s]/, '_')}"
    unless PAGE_CACHE[filename]
      ::File.open("#{Mechanize::TestCase::TEST_DIR}/#{filename}", 'rb') do |io|
        PAGE_CACHE[filename] = io.read
      end
    end

    res.body = PAGE_CACHE[filename]
    case filename
    when /\.txt$/
      res['Content-Type'] = 'text/plain'
    when /\.jpg$/
      res['Content-Type'] = 'image/jpeg'
    end
  end

  res['Content-Type'] ||= 'text/html'
  res.code ||= "200"

  response_klass = Net::HTTPResponse::CODE_TO_OBJ[res.code.to_s]
  response = response_klass.new res.http_version, res.code, res.message

  res.header.each do |k,v|
    v = v.first if v.length == 1
    response[k] = v
  end

  res.cookies.each do |cookie|
    response.add_field 'Set-Cookie', cookie.to_s
  end

  response['Content-Type'] ||= 'text/html'
  response['Content-Length'] = res['Content-Length'] || res.body.length.to_s

  io = StringIO.new(res.body)
  response.instance_variable_set :@socket, io
  def io.read clen, dest = nil, _ = nil
    if dest then
      dest << super(clen)
    else
      super clen
    end
  end

  body_exist = req.response_body_permitted? &&
    response_klass.body_permitted?

  response.instance_variable_set :@body_exist, body_exist

  yield response if block_given?

  response
end