1
2
3
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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
|
# frozen_string_literal: true
require "bundler/vendored_net_http"
require_relative "../path"
CASSETTE_PATH = "#{Spec::Path.spec_dir}/support/artifice/vcr_cassettes".freeze
USED_CASSETTES_PATH = "#{Spec::Path.spec_dir}/support/artifice/used_cassettes.txt".freeze
CASSETTE_NAME = ENV.fetch("BUNDLER_SPEC_VCR_CASSETTE_NAME") { "realworld" }
class BundlerVCRHTTP < Gem::Net::HTTP
class RequestHandler
attr_reader :http, :request, :body, :response_block
def initialize(http, request, body = nil, &response_block)
@http = http
@request = request
@body = body
@response_block = response_block
end
def handle_request
handler = self
request.instance_eval do
@__vcr_request_handler = handler
end
File.open(USED_CASSETTES_PATH, "a+") do |f|
f.puts request_pair_paths.map {|path| Pathname.new(path).relative_path_from(Spec::Path.source_root).to_s }.join("\n")
end
if recorded_response?
recorded_response
else
record_response
end
end
def recorded_response?
return true if ENV["BUNDLER_SPEC_PRE_RECORDED"]
request_pair_paths.all? {|f| File.exist?(f) }
end
def recorded_response
File.open(request_pair_paths.last, "rb:ASCII-8BIT") do |response_file|
response_io = ::Gem::Net::BufferedIO.new(response_file)
::Gem::Net::HTTPResponse.read_new(response_io).tap do |response|
response.decode_content = request.decode_content if request.respond_to?(:decode_content)
response.uri = request.uri
response.reading_body(response_io, request.response_body_permitted?) do
response_block&.call(response)
end
end
end
end
def record_response
request_path, response_path = *request_pair_paths
@recording = true
response = http.request_without_vcr(request, body, &response_block)
@recording = false
unless @recording
require "fileutils"
FileUtils.mkdir_p(File.dirname(request_path))
binwrite(request_path, request_to_string(request))
binwrite(response_path, response_to_string(response))
end
response
end
def key
[request["host"] || http.address, request.path, request.method].compact
end
def file_name_for_key(key)
File.join(*key).gsub(/[\:*?"<>|]/, "-")
end
def request_pair_paths
%w[request response].map do |kind|
File.join(CASSETTE_PATH, CASSETTE_NAME, file_name_for_key(key), kind)
end
end
def request_to_string(request)
request_string = []
request_string << "> #{request.method.upcase} #{request.path}"
request.to_hash.each do |key, value|
request_string << "> #{key}: #{Array(value).first}"
end
request << "" << request.body if request.body
request_string.join("\n")
end
def response_to_string(response)
headers = response.to_hash
body = response.body
response_string = []
response_string << "HTTP/1.1 #{response.code} #{response.message}"
headers["content-length"] = [body.bytesize.to_s] if body
headers.each do |header, value|
response_string << "#{header}: #{value.join(", ")}"
end
response_string << "" << body
response_string = response_string.join("\n")
if response_string.respond_to?(:force_encoding)
response_string.force_encoding("ASCII-8BIT")
else
response_string
end
end
def binwrite(path, contents)
File.open(path, "wb:ASCII-8BIT") {|f| f.write(contents) }
end
end
def start_with_vcr
if ENV["BUNDLER_SPEC_PRE_RECORDED"]
raise IOError, "HTTP session already opened" if @started
@socket = nil
@started = true
else
start_without_vcr
end
end
alias_method :start_without_vcr, :start
alias_method :start, :start_with_vcr
def request_with_vcr(request, *args, &block)
handler = request.instance_eval do
remove_instance_variable(:@__vcr_request_handler) if defined?(@__vcr_request_handler)
end || RequestHandler.new(self, request, *args, &block)
handler.handle_request
end
alias_method :request_without_vcr, :request
alias_method :request, :request_with_vcr
end
require_relative "helpers/artifice"
# Replace Gem::Net::HTTP with our VCR subclass
Artifice.replace_net_http(BundlerVCRHTTP)
|
