dkubb-fakeweb 1.1.2.6

data/Rakefile

@@ -0,0 +1,57 @@
+require 'rubygems'
+require 'rake/testtask'
+require 'rake/rdoctask'
+require 'rcov/rcovtask'
+
+task :default => :test
+
+desc "Run All Tests"
+Rake::TestTask.new :test do |test|
+  test.test_files = ["test/**/*.rb"]
+  test.verbose = true
+end
+
+desc "Generate Documentation"
+Rake::RDocTask.new do |rdoc|
+  rdoc.main = "README.rdoc"
+  rdoc.rdoc_dir = "doc"
+  rdoc.rdoc_files.include("README.rdoc", "CHANGELOG", "LICENSE.txt", "lib/*.rb")
+  rdoc.title = "FakeWeb API Documentation"
+  rdoc.options << '--line-numbers' << '--inline-source'
+  rdoc.options << '--charset' << 'utf-8'
+end
+
+Rcov::RcovTask.new do |t|
+  t.test_files = FileList['test/**/test*.rb'] 
+  t.rcov_opts << "--sort coverage"
+  t.rcov_opts << "--exclude gems"
+  t.rcov_opts << "--no-validator-links"
+end
+
+desc %{Update ".manifest" with the latest list of project filenames. Respect\
+.gitignore by excluding everything that git ignores. Update `files` and\
+`test_files` arrays in "*.gemspec" file if it's present.}
+task :manifest do
+  list = Dir['**/*'].sort
+  spec_file = Dir['*.gemspec'].first
+  list -= [spec_file] if spec_file
+
+  File.read('.gitignore').each_line do |glob|
+    glob = glob.chomp.sub(/^\//, '')
+    list -= Dir[glob]
+    list -= Dir["#{glob}/**/*"] if File.directory?(glob) and !File.symlink?(glob)
+    puts "excluding #{glob}"
+  end
+
+  if spec_file
+    spec = File.read spec_file
+    spec.gsub! /^(\s* s.(test_)?files \s* = \s* )( \[ [^\]]* \] | %w\( [^)]* \) )/mx do
+      assignment = $1
+      bunch = $2 ? list.grep(/^test\//) : list
+      '%s%%w(%s)' % [assignment, bunch.join(' ')]
+    end
+
+    File.open(spec_file,   'w') {|f| f << spec }
+  end
+  File.open('.manifest', 'w') {|f| f << list.join("\n") }
+end

data/lib/fake_httpclient.rb

@@ -0,0 +1,67 @@
+# FakeWeb - Ruby Helper for Faking Web Requests
+# Copyright 2006 Blaine Cook <romeda@gmail.com>.
+# 
+# FakeWeb is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+# 
+# FakeWeb is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+# 
+# You should have received a copy of the GNU General Public License
+# along with FakeWeb; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+
+require 'rubygems'
+gem 'httpclient', '>=2.1.2'
+require 'httpclient'
+
+class HTTPClient
+
+  LIBRARY_IDENTIFIER = :httpclient
+  FakeWeb.register_client_library(LIBRARY_IDENTIFIER)
+
+  alias :original_httpclient_do_get_block :do_get_block
+
+  def self.socket_type
+    FakeWeb::SocketDelegator
+  end
+
+  def do_get_block(req, proxy, conn, &block)
+    method = req.header.request_method.downcase.to_sym
+    uri    = req.header.request_uri
+
+    if FakeWeb.registered_uri?(method, uri)
+      message = FakeWeb.response_for(method, LIBRARY_IDENTIFIER, uri, &block)
+      conn.push(message)
+    elsif FakeWeb.allow_net_connect?
+      original_httpclient_do_get_block(req, proxy, conn, &block)
+    else
+      raise FakeWeb::NetConnectNotAllowedError,
+            "Real HTTP connections are disabled. Unregistered request: #{req.header.request_method} #{uri}"
+    end
+  end
+
+=begin
+  # Do we want this? I'd imagine it'd make unregistered urls unhappy...
+
+  # Make sure there's no chance of a connection being made
+  class Session
+    def query(req)
+    end
+  end
+=end
+
+end
+
+# A uniform way to get fake content, for testing.
+module HTTP
+  class Message
+    def get_content
+      self.body.content
+    end
+  end
+end

data/lib/fake_net_http.rb

@@ -0,0 +1,67 @@
+require 'net/http'
+require 'net/https'
+require 'stringio'
+
+module Net  #:nodoc: all
+
+  class BufferedIO
+    def initialize(io, debug_output = nil)
+      @read_timeout = 60
+      @rbuf = ''
+      @debug_output = debug_output
+
+      @io = case io
+      when Socket, OpenSSL::SSL::SSLSocket, IO
+        io
+      when String
+        File.exists?(io) ? File.open(io, "r") : StringIO.new(io)
+      end
+      raise "Unable to create local socket" unless @io
+    end
+  end
+
+
+  class HTTP
+
+    LIBRARY_IDENTIFIER = :net_http
+    FakeWeb.register_client_library(LIBRARY_IDENTIFIER)
+
+    def self.socket_type
+      FakeWeb::SocketDelegator
+    end
+
+    alias :original_net_http_request :request
+    alias :original_net_http_connect :connect
+
+    def request(request, body = nil, &block)
+      protocol = use_ssl ? "https" : "http"
+
+      path = request.path
+      path = URI.parse(request.path).request_uri if request.path =~ /^http/
+
+      uri = "#{protocol}://#{self.address}:#{self.port}#{path}"
+      method = request.method.downcase.to_sym
+
+      if FakeWeb.registered_uri?(method, uri)
+        @socket = Net::HTTP.socket_type.new
+        FakeWeb.response_for(method, LIBRARY_IDENTIFIER, uri, &block)
+      elsif FakeWeb.allow_net_connect?
+        original_net_http_connect
+        original_net_http_request(request, body, &block)
+      else
+        raise FakeWeb::NetConnectNotAllowedError,
+              "Real HTTP connections are disabled. Unregistered request: #{request.method} #{uri}"
+      end
+    end
+
+    def connect
+    end
+  end
+
+  # A uniform way to get fake content, for testing.
+  class HTTPResponse
+    def get_content
+      self.body
+    end
+  end
+end

data/lib/fake_web.rb

@@ -0,0 +1,348 @@
+module FakeWeb
+  CLIENT_LIBRARIES = []
+  def self.register_client_library(library_identifier)
+    CLIENT_LIBRARIES << library_identifier
+  end
+end
+
+require File.expand_path(File.join(File.dirname(__FILE__), 'fake_net_http'))
+require File.expand_path(File.join(File.dirname(__FILE__), 'fake_httpclient'))
+require 'singleton'
+
+module OpenURI #:nodoc: all
+  class HTTPError < StandardError; end;
+end
+
+module FakeWeb
+
+
+  # Resets the FakeWeb Registry. This will force all subsequent web requests to
+  # behave as real requests.
+  def self.clean_registry
+    Registry.instance.clean_registry
+  end
+
+  # Enables or disables real HTTP connections for requests that don't match
+  # registered URIs.
+  #
+  # If you set <tt>FakeWeb.allow_net_connect = false</tt> and subsequently try
+  # to make a request to a URI you haven't registered with #register_uri, a
+  # NetConnectNotAllowedError will be raised. This is handy when you want to
+  # make sure your tests are self-contained, or want to catch the scenario
+  # when a URI is changed in implementation code without a corresponding test
+  # change.
+  #
+  # When <tt>FakeWeb.allow_net_connect = true</tt> (the default), requests to
+  # URIs not stubbed with FakeWeb are passed through to Net::HTTP.
+  def self.allow_net_connect=(allowed)
+    @allow_net_connect = allowed
+  end
+
+  # Enable pass-through to Net::HTTP by default.
+  self.allow_net_connect = true
+
+  # Returns +true+ if requests to URIs not registered with FakeWeb are passed
+  # through to Net::HTTP for normal processing (the default). Returns +false+
+  # if an exception is raised for these requests.
+  def self.allow_net_connect?
+    @allow_net_connect
+  end
+
+  # This exception is raised if you set <tt>FakeWeb.allow_net_connect =
+  # false</tt> and subsequently try to make a request to a URI you haven't
+  # stubbed.
+  class NetConnectNotAllowedError < StandardError; end;
+
+  # call-seq:
+  #   FakeWeb.register_uri(method, uri, options)
+  #   FakeWeb.register_uri(uri, options)
+  #
+  # Register requests using the HTTP method specified by the symbol +method+ for
+  # +uri+ to be handled according to +options+. If no +method+ is specified, or
+  # you explicitly specify <tt>:any</tt>, the response will be reigstered for
+  # any request for +uri+. +uri+ can be a +String+ or a +URI+ object. +options+
+  # must be either a +Hash+ or an +Array+ of +Hashes+ (see below) that must
+  # contain any one of the following keys:
+  #
+  # <tt>:string</tt>::
+  #   Takes a +String+ argument that is returned as the body of the response.
+  #     FakeWeb.register_uri(:get, 'http://example.com/', :string => "Hello World!")
+  # <tt>:file</tt>::
+  #   Takes a valid filesystem path to a file that is slurped and returned as
+  #   the body of the response.
+  #     FakeWeb.register_uri(:post, 'http://example.com/', :file => "/tmp/my_response_body.txt")
+  # <tt>:response</tt>:: 
+  #   Either an <tt>Net::HTTPResponse</tt>, an +IO+ or a +String+.
+  # 
+  #   The easier way by far is to pass the <tt>:response</tt> option to
+  #   +register_uri+ as a +String+ or an (open for reads) +IO+ object which
+  #   will be used as the complete HTTP response, including headers and body.
+  #   If the string points to a readable file, this file will be used as the
+  #   content for the request.
+  # 
+  #   To obtain a complete response document, you can use the +curl+ command,
+  #   like so:
+  #  
+  #     curl -i http://www.example.com/ > response_for_www.example.com
+  #
+  #   which can then be used in your test environment like so:
+  #
+  #     FakeWeb.register_uri(:get, 'http://www.example.com/', :response => 'response_for_www.example.com')
+  #
+  #   See the <tt>Net::HTTPResponse</tt>
+  #   documentation[http://ruby-doc.org/stdlib/libdoc/net/http/rdoc/classes/Net/HTTPResponse.html]
+  #   for more information on creating custom response objects.
+  # 
+  # +options+ may also be an +Array+ containing a list of the above-described +Hash+.
+  # In this case, FakeWeb will rotate through each provided response, you may optionally
+  # provide:
+  #
+  # <tt>:times</tt>::
+  #   The number of times this response will be used. Decremented by one each time it's called.
+  #   FakeWeb will use the final provided request indefinitely, regardless of its :times parameter.
+  # 
+  # Two optional arguments are also accepted:
+  #
+  # <tt>:status</tt>::
+  #   Passing <tt>:status</tt> as a two-value array will set the response code
+  #   and message. The defaults are <tt>200</tt> and <tt>OK</tt>, respectively.
+  #   Example:
+  #     FakeWeb.register_uri('http://www.example.com/', :response => "Go away!", :status => [ 404, "Not Found" ])
+  # <tt>:exception</tt>::
+  #   The argument passed via <tt>:exception</tt> will be raised when the
+  #   specified URL is requested. Any +Exception+ class is valid. Example:
+  #     FakeWeb.register_uri('http://www.example.com/', :exception => Net::HTTPError)
+  #
+  def self.register_uri(*args)
+    method = :any
+    case args.length
+    when 3 then method, uri, options = *args
+    when 2 then         uri, options = *args
+    else   raise ArgumentError.new("wrong number of arguments (#{args.length} for method = :any, uri, options)")
+    end
+    Registry.instance.register_uri(method, uri, options)
+  end
+
+  # call-seq:
+  #   FakeWeb.response_for(method, uri, client_library)
+  #   FakeWeb.response_for(uri, client_library)
+  #
+  # When using Net::HTTP
+  #   Returns the faked Net::HTTPResponse object associated with +uri+.
+  # When using HTTPClient
+  #   Returns the faked HTTP::Message object associated with +uri+.
+  def self.response_for(*args, &block) #:nodoc: :yields: response
+    method = :any
+    case args.length
+    when 3 then method, client_library, uri = *args
+    when 2 then         client_library, uri = *args
+    else   raise ArgumentError.new("wrong number of arguments (#{args.length} for method = :any, uri)")
+    end
+    Registry.instance.response_for(method, uri, client_library, &block)
+  end
+
+  # call-seq:
+  #   FakeWeb.registered_uri?(method, uri)
+  #   FakeWeb.registered_uri?(uri)
+  #
+  # Returns true if +uri+ is registered with FakeWeb. You can optionally
+  # specify +method+ to limit the search to a certain HTTP method (or use
+  # <tt>:any</tt> to explicitly check against any method).
+  def self.registered_uri?(*args)
+    method = :any
+    case args.length
+    when 2 then method, uri = *args
+    when 1 then         uri = *args
+    else   raise ArgumentError.new("wrong number of arguments (#{args.length} for method = :any, uri)")
+    end
+    Registry.instance.registered_uri?(method, uri)
+  end
+
+  class Registry #:nodoc:
+    include Singleton
+
+    attr_accessor :uri_map
+
+    def initialize
+      clean_registry
+    end
+
+    def clean_registry
+      self.uri_map = Hash.new { |hash, key| hash[key] = Hash.new(&hash.default_proc) }
+    end
+
+    def register_uri(method, uri, options)
+      uri_map[normalize_uri(uri)][method] = [*[options]].flatten.collect { |option|
+        FakeWeb::Responder.new(method, uri, option, option[:times])
+      }
+    end
+
+    def registered_uri?(method, uri)
+      normalized_uri = normalize_uri(uri)
+      uri_map[normalized_uri].has_key?(method) || uri_map[normalized_uri].has_key?(:any)
+    end
+
+    def registered_uri(method, uri)
+      uri = normalize_uri(uri)
+      registered = registered_uri?(method, uri)
+      if registered && uri_map[uri].has_key?(method)
+        uri_map[uri][method]
+      elsif registered
+        uri_map[uri][:any]
+      else
+        nil
+      end
+    end
+
+    def response_for(method, uri, client_library, &block)
+      responses = registered_uri(method, uri)
+      return nil if responses.nil?
+
+      next_response = responses.last
+      responses.each { |response|
+        if response.times and response.times > 0
+          response.times -= 1
+          next_response = response
+          break
+        end
+      }
+
+      return next_response.response(client_library, &block)
+    end
+
+    private
+
+    def normalize_uri(uri)
+      case uri
+      when URI then uri
+      else
+        uri = 'http://' + uri unless uri.match('^https?://')
+        parsed_uri = URI.parse(uri)
+        parsed_uri.query = sort_query_params(parsed_uri.query)
+        parsed_uri
+      end
+    end
+
+    def sort_query_params(query)
+      return nil if query.nil? or query.empty?
+      query.split('&').sort.join('&')
+    end
+  end
+
+  module Response #:nodoc:
+    def read_body(*args, &block)
+      yield @body if block_given?
+      @body
+    end
+  end
+
+  class Responder #:nodoc:
+
+    attr_accessor :method, :uri, :options, :times
+
+    def initialize(method, uri, options, times)
+      self.method = method
+      self.uri = uri
+      self.options = options
+      self.times = times ? times : 1
+    end
+
+    def response(client_library, &block)
+      if has_baked_response?
+        response = baked_response
+      else
+        case client_library
+          when :net_http
+            code, msg = meta_information
+            response = Net::HTTPResponse.send(:response_class, code.to_s).new(uri, code.to_s, msg)
+            response.instance_variable_set(:@body, content)
+          when :httpclient
+            response = HTTP::Message.new_response(content)
+        end
+      end
+      case client_library
+        when :net_http
+          response.instance_variable_set(:@read, true)
+          response.extend FakeWeb::Response
+      end
+
+      optionally_raise(response)
+
+      yield response if block_given?
+
+      response
+    end
+
+    private
+
+    def content
+      [ :file, :string ].each do |map_option|
+        next unless options.has_key?(map_option)
+        return self.send("#{map_option}_response", options[map_option])
+      end
+
+      return ''
+    end
+
+    def file_response(path)
+      IO.readlines(path).join("\n")
+    end
+
+    def string_response(string)
+      string
+    end
+
+    def baked_response
+      resp = case options[:response]
+      when Net::HTTPResponse then options[:response]
+      when String
+        socket = Net::BufferedIO.new(options[:response])
+        r = Net::HTTPResponse.read_new(socket)
+        r.instance_eval { @header['transfer-encoding'] = nil }
+        r.reading_body(socket, true) {}
+        r
+      else raise StandardError, "Handler unimplemented for response #{options[:response]}"
+      end
+    end
+
+    def has_baked_response?
+      options.has_key?(:response)
+    end
+
+    def optionally_raise(response)
+      return unless options.has_key?(:exception)
+      ex_alloc = options[:exception].allocate
+      ex_instance = case ex_alloc
+      when Net::HTTPError, OpenURI::HTTPError
+        options[:exception].new('Exception from FakeWeb', response)
+      else options[:exception].new
+      end
+      raise ex_instance
+    end
+
+    def meta_information
+      if options.has_key?(:status); options[:status]
+      else; [ 200, 'OK' ]
+      end
+    end
+  end
+
+  class SocketDelegator #:nodoc:
+
+    def initialize(delegate=nil)
+      @delegate = nil
+    end
+
+    def method_missing(method, *args, &block)
+      return @delegate.send(method, *args, &block) if @delegate
+      return self.send("my_#{method}", *args, &block)
+    end
+
+    def my_closed?
+      @closed ||= true
+    end
+
+    def my_readuntil(*args)
+    end
+  end
+end