#!/usr/bin/env ruby<\/span>\n#<\/span>\n# unknown source :(<\/span>\n\nrequire<\/span> 'rubygems'<\/span>\nrequire<\/span> 'rack'<\/span>\nrequire<\/span> 'thin'<\/span>\n\n# Mock server for testing bad or heavyweight server responses; runs on<\/span>\n# localhost:4000 by default<\/span>\n#<\/span>\n# Example of use<\/span>\n#<\/span>\n# In setup method:<\/span>\n#<\/span>\n# @mock_server = MockServer.new # creating a new instance spawns a<\/span>\n# thread running the TCP server<\/span>\n# @mock_server.register( { 'REQUEST_METHOD' => 'GET' }, <\/span>\n# [ 200, { 'Content-Type' => 'text/plain', 'Content-Length' => '11' }, <\/span>\n# [ 'Hello World' ]])<\/span>\n# @mock_server.register( { `REQUEST_METHOD' => `GET' }, [<\/span>\n#  200, { 'Content-Type' => 'text/plain', 'Content-Length' => '11' }, <\/span>\n#  [ 'Hello Again' ]])<\/span>\n#<\/span>\n# After each test, to remove all expectations:<\/span>\n#<\/span>\n# @mock_server.clear<\/span>\n#<\/span>\n# In teardown method:<\/span>\n#<\/span>\n# @mock_server.stop<\/span>\n\nclass<\/span> MockServer<\/span>\n\n  def<\/span> initialize<\/span>(<\/span>options<\/span>=<\/span>{})<\/span>\n    host<\/span> =<\/span> options<\/span>[<\/span>:host<\/span>]<\/span> ||<\/span> '127.0.0.1'<\/span>\n    port<\/span> =<\/span> options<\/span>[<\/span>:port<\/span>]<\/span> ||<\/span> 4000<\/span>\n    @expectations<\/span> =<\/span> []<\/span>\n    @server<\/span> =<\/span> Thin<\/span>::<\/span>Server<\/span>.<\/span>new<\/span>(<\/span>host<\/span>,<\/span> port<\/span>,<\/span> self<\/span>)<\/span>\n    @thread<\/span> =<\/span> Thread<\/span>.<\/span>new<\/span> {<\/span> @server<\/span>.<\/span>start<\/span> }<\/span>\n  end<\/span>\n\n  def<\/span> stop<\/span>\n    @server<\/span>.<\/span>stop!<\/span>\n    Thread<\/span>.<\/span>kill<\/span>(<\/span>@thread<\/span>)<\/span>\n  end<\/span>\n\n# env should be a hash mapping elements of a Rack env to expected values <\/span>\n# (see examples above); note that an expected value can be a Proc which <\/span>\n# will be passed the value from the request<\/span>\n# and executed - if the Proc returns true on execution, the expectation is met<\/span>\n#<\/span>\n# For example, to check that the querystring contains the value `1234', env could be:<\/span>\n#<\/span>\n# { `QUERY_STRING' => lambda { |qs| !((qs =~ /1234/).nil?) } }<\/span>\n#<\/span>\n# response should be a Rack-formatted response; i.e. [response_code,<\/span>\n# {'header' => 'value', ...}, response_body]<\/span>\n#<\/span>\n# options:<\/span>\n# :transient => false to prevent a response being removed after it has <\/span>\n# been served (default is true)<\/span>\n  def<\/span> register<\/span>(<\/span>env<\/span>,<\/span> response<\/span>,<\/span> options<\/span>=<\/span>{})<\/span>\n    transient<\/span> =<\/span> options<\/span>[<\/span>:transient<\/span>]<\/span>\n    transient<\/span> =<\/span> true<\/span> if<\/span> transient<\/span>.<\/span>nil?<\/span>\n\n    @expectations<\/span> =<\/span> []<\/span>\n\n    @expectations<\/span>.<\/span>each_with_index<\/span> do<\/span> |<\/span>expectation<\/span>,<\/span> index<\/span>|<\/span>\n      expectation_env<\/span>,<\/span> matched_response<\/span>,<\/span> transient<\/span> =<\/span> expectation<\/span>\n      matched<\/span> =<\/span> false<\/span>\n\n      expectation_env<\/span>.<\/span>each<\/span> do<\/span> |<\/span>env_key<\/span>,<\/span> value<\/span>|<\/span>\n        puts<\/span> "Trying to match <\/span>#{<\/span>env_key<\/span>}<\/span> => <\/span>#{<\/span>value<\/span>}<\/span> to request"<\/span>\n        matched<\/span> =<\/span> true<\/span>\n\n        req_value<\/span> =<\/span> env<\/span>[<\/span>env_key<\/span>]<\/span>\n\n        if<\/span> value<\/span>.<\/span>is_a?<\/span> Proc<\/span>\n          req_element_matches<\/span> =<\/span> value<\/span>.<\/span>call<\/span>(<\/span>req_value<\/span>)<\/span>\n        else<\/span>\n          req_element_matches<\/span> =<\/span> (<\/span>value<\/span> ==<\/span> req_value<\/span>)<\/span>\n        end<\/span>\n\n        unless<\/span> req_element_matches<\/span>\n          puts<\/span> " Value NOT matched: request value was <\/span>#{<\/span>env<\/span>[<\/span>env_key<\/span>]<\/span>}<\/span> (needed <\/span>#{<\/span>value<\/span>}<\/span> to match)"<\/span>\n          matched<\/span> =<\/span> false<\/span>\n          break<\/span>\n        end<\/span>\n      end<\/span>\n\n      if<\/span> matched<\/span>\n        if<\/span> transient<\/span>\n          @expectations<\/span>.<\/span>delete_at<\/span>(<\/span>index<\/span>)<\/span>\n        end<\/span>\n        response<\/span> =<\/span> matched_response<\/span>\n        break<\/span>\n      end<\/span>\n    end<\/span>\n    response<\/span>\n  end<\/span>\n\nend<\/span>\n<\/pre><\/div>\n