Class: Excon::Middleware::Mock

Inherits:
Base
  • Object
show all
Defined in:
lib/excon/middlewares/mock.rb

Instance Method Summary collapse

Methods inherited from Base

#initialize, #response_call

Constructor Details

This class inherits a constructor from Excon::Middleware::Base

Instance Method Details

#request_call(datum) ⇒ Object 



4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
 
# File 'lib/excon/middlewares/mock.rb', line 4

def request_call(datum)
  if datum[:mock]
    # convert File/Tempfile body to string before matching:
    unless datum[:body].nil? || datum[:body].is_a?(String)
     if datum[:body].respond_to?(:binmode)
       datum[:body].binmode
     end
     if datum[:body].respond_to?(:rewind)
       datum[:body].rewind
     end
     datum[:body] = datum[:body].read
    end

    datum[:captures] = {:headers => {}} # setup data to hold captures
    Excon.stubs.each do |stub, response|
      headers_match = !stub.has_key?(:headers) || stub[:headers].keys.all? do |key|
        case value = stub[:headers][key]
        when Regexp
          if match = value.match(datum[:headers][key])
            datum[:captures][:headers][key] = match.captures
          end
          match
        else
          value == datum[:headers][key]
        end
      end
      non_headers_match = (stub.keys - [:headers]).all? do |key|
        case value = stub[key]
        when Regexp
          if match = value.match(datum[key])
            datum[:captures][key] = match.captures
          end
          match
        else
          value == datum[key]
        end
      end
      if headers_match && non_headers_match
        datum[:response] = {
          :body       => '',
          :headers    => {},
          :status     => 200,
          :remote_ip  => '127.0.0.1'
        }

        stub_datum = case response
        when Proc
          response.call(datum)
        else
          response
        end

        datum[:response].merge!(stub_datum.reject {|key,value| key == :headers})
        if stub_datum.has_key?(:headers)
          datum[:response][:headers].merge!(stub_datum[:headers])
        end

        if datum[:expects] && ![*datum[:expects]].include?(datum[:response][:status])
          # don't pass stuff into a block if there was an error
        elsif datum.has_key?(:response_block) && datum[:response].has_key?(:body)
          body = datum[:response].delete(:body)
          content_length = remaining = body.bytesize
          i = 0
          while i < body.length
            datum[:response_block].call(body[i, datum[:chunk_size]], [remaining - datum[:chunk_size], 0].max, content_length)
            remaining -= datum[:chunk_size]
            i += datum[:chunk_size]
          end
        end
        return @stack.request_call(datum)
      end
    end
    # if we reach here no stubs matched
    raise(Excon::Errors::StubNotFound.new('no stubs matched ' << datum.inspect))
  else
    @stack.request_call(datum)
  end
end