Module: Supergood::Vendor::NetHTTP

Defined in:
lib/supergood/vendors/net-http.rb

Class Method Summary collapse

Class Method Details

.existing_patch? ⇒ Boolean

Returns:

  • (Boolean) 


63
64
65
66
 
# File 'lib/supergood/vendors/net-http.rb', line 63

def self.existing_patch?
  (defined?(Net::HTTP) && Net::HTTP.method_defined?(:original_request_method)) ||
  (defined?(::Webmock) && WebMock::HttpLibAdapters::NetHttpAdapter.instance_variable_get("@webMockNetHTTP").method_defined?(:original_request_method))
end

.patch ⇒ Object 



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
 
# File 'lib/supergood/vendors/net-http.rb', line 8

def self.patch
  if !self.existing_patch?
    block = lambda do |x|
      alias original_request_method request
      def request(original_request_payload, body = nil, &block)
        http = self
        url = Supergood::Utils.request_url(http, original_request_payload)
        uri = URI.parse(url)
        request = {
          headers: Supergood::Utils.get_header(original_request_payload),
          method: original_request_payload.method,
          body: original_request_payload.body,
          url: url,
          path: original_request_payload.path,
          search: uri.query,
          domain: Supergood::Utils.get_host_without_www(uri.host)
        }
        Supergood.intercept(request) do
          original_response = original_request_method(original_request_payload, body, &block)
          {
            headers: Supergood::Utils.get_header(original_response),
            status: original_response.code,
            statusText: original_response.message,
            body: original_response.body,
            original_response: original_response
          }
        end
      end
    end

    if defined?(Net::HTTP)
      Net::HTTP.class_eval(&block)
    elsif defined?(::WebMock)
      WebMock::HttpLibAdapters::NetHttpAdapter.instance_variable_get("@webMockNetHTTP").class_eval(&block)
    end

  end
end

.unpatch ⇒ Object 



47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
 
# File 'lib/supergood/vendors/net-http.rb', line 47

def self.unpatch
  if self.existing_patch?
    block = lambda do |x|
      alias request original_request_method
    end

    if defined?(Net::HTTP)
      Net::HTTP.class_eval(&block)
      Net::HTTP.undef_method :original_request_method
    elsif defined?(::WebMock)
      WebMock::HttpLibAdapters::NetHttpAdapter.instance_variable_get("@webMockNetHTTP").class_eval(&block)
      WebMock::HttpLibAdapters::NetHttpAdapter.instance_variable_get("@webMockNetHTTP").undef_method :original_request_method
    end
  end
end