Class: Apipie::Extractor::Recorder

Inherits:
Object
  • Object
show all
Defined in:
lib/apipie/extractor/recorder.rb

Defined Under Namespace

Modules: FunctionalTestRecording Classes: Middleware

Constant Summary collapse

MULTIPART_BOUNDARY = 
'APIPIE_RECORDER_EXAMPLE_BOUNDARY'

Instance Method Summary collapse

Constructor Details

#initialize ⇒ Recorder

Returns a new instance of Recorder.



6
7
8
 
# File 'lib/apipie/extractor/recorder.rb', line 6

def initialize
  @ignored_params = [:controller, :action]
end

Instance Method Details

#analyse_controller(controller) ⇒ Object 



24
25
26
27
 
# File 'lib/apipie/extractor/recorder.rb', line 24

def analyse_controller(controller)
  @controller = controller.class
  @action = controller.params[:action]
end

#analyse_env(env) ⇒ Object 



10
11
12
13
14
15
16
17
18
19
20
21
22
 
# File 'lib/apipie/extractor/recorder.rb', line 10

def analyse_env(env)
  @verb = env["REQUEST_METHOD"].to_sym
  @path = env["PATH_INFO"].sub(/^\/*/,"/")
  @query = env["QUERY_STRING"] unless env["QUERY_STRING"].blank?
  @params = Rack::Utils.parse_nested_query(@query)
  @params.merge!(env["action_dispatch.request.request_parameters"] || {})
  if data = parse_data(env["rack.input"].read)
    @request_data = data
    env["rack.input"].rewind
  elsif form_hash = env["rack.request.form_hash"]
    @request_data = reformat_multipart_data(form_hash)
  end
end

#analyse_response(response) ⇒ Object 



29
30
31
32
33
34
35
36
37
38
 
# File 'lib/apipie/extractor/recorder.rb', line 29

def analyse_response(response)
  if response.last.respond_to?(:body) && data = parse_data(response.last.body)
    @response_data = if response[1]['Content-Disposition'].to_s.start_with?('attachment')
                       '<STREAMED ATTACHMENT FILE>'
                     else
                       data
                     end
  end
  @code = response.first
end

#analyze_functional_test(test_context) ⇒ Object 



40
41
42
43
44
45
46
47
48
49
50
51
52
 
# File 'lib/apipie/extractor/recorder.rb', line 40

def analyze_functional_test(test_context)
  request, response = test_context.request, test_context.response
  @verb = request.request_method.to_sym
  @path = request.path
  @params = request.request_parameters
  if [:POST, :PUT, :PATCH, :DELETE].include?(@verb)
    @request_data = @params
  else
    @query = request.query_string
  end
  @response_data = parse_data(response.body)
  @code = response.code
end

#content_disposition(name) ⇒ Object 



88
89
90
 
# File 'lib/apipie/extractor/recorder.rb', line 88

def content_disposition(name)
  %{Content-Disposition: form-data; name="#{name}"}
end

#parse_data(data) ⇒ Object 



54
55
56
57
58
59
 
# File 'lib/apipie/extractor/recorder.rb', line 54

def parse_data(data)
  return nil if data.strip.blank?
  JSON.parse(data)
rescue StandardError
  data
end

#record ⇒ Object 



104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
 
# File 'lib/apipie/extractor/recorder.rb', line 104

def record
  if @controller
    {:controller => @controller,
     :action => @action,
     :verb => @verb,
     :path => @path,
     :params => @params,
     :query => @query,
     :request_data => @request_data,
     :response_data => @response_data,
     :code => @code}
  else
    nil
  end
end

#reformat_data(data) ⇒ Object 



92
93
94
95
96
97
98
99
100
101
102
 
# File 'lib/apipie/extractor/recorder.rb', line 92

def reformat_data(data)
  parsed = parse_data(data)
  case parsed
  when nil
    nil
  when String
    parsed
  else
    JSON.pretty_generate().gsub(/: \[\s*\]/,": []").gsub(/\{\s*\}/,"{}")
  end
end

#reformat_hash(boundary, attrs, lines) ⇒ Object 



76
77
78
79
80
81
82
83
84
85
86
 
# File 'lib/apipie/extractor/recorder.rb', line 76

def reformat_hash(boundary, attrs, lines)
  if head = attrs[:head]
    lines << boundary
    lines.concat(head.split("\r\n"))
    # To avoid large and/or binary file bodies, simply indicate the contents in the output.
    lines << '' << %{... contents of "#{attrs[:name]}" ...}
  else
    # Look for subelements that contain a part.
    attrs.each { |k,v| v.is_a?(Hash) and reformat_hash(boundary, v, lines) }
  end
end

#reformat_multipart_data(form) ⇒ Object 



61
62
63
64
65
66
67
68
69
70
71
72
73
74
 
# File 'lib/apipie/extractor/recorder.rb', line 61

def reformat_multipart_data(form)
  form.empty? and return ''
  lines = ["Content-Type: multipart/form-data; boundary=#{MULTIPART_BOUNDARY}",'']
  boundary = "--#{MULTIPART_BOUNDARY}"
  form.each do |key, attrs|
    if attrs.is_a?(String)
      lines << boundary << content_disposition(key) << "Content-Length: #{attrs.size}" << '' << attrs
    else
      reformat_hash(boundary, attrs, lines)
    end
  end
  lines << "#{boundary}--"
  lines.join("\n")
end