forked from airbrake/airbrake
/
sender_test.rb
261 lines (213 loc) · 8.27 KB
/
sender_test.rb
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
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
require File.expand_path( File.join(File.dirname(__FILE__), 'helper') )
class SenderTest < Test::Unit::TestCase
def setup
reset_config
end
def build_sender(opts = {})
Airbrake.configure do |conf|
opts.each {|opt, value| conf.send(:"#{opt}=", value) }
end
end
def send_exception(args = {})
notice = args.delete(:notice) || build_notice_data
sender = args.delete(:sender) || build_sender(args)
sender.send_to_airbrake(notice)
end
def stub_http(options = {})
response = stub(:body => options[:body] || 'body')
http = stub(:post => response,
:read_timeout= => nil,
:open_timeout= => nil,
:ca_file= => nil,
:verify_mode= => nil,
:use_ssl= => nil)
Net::HTTP.stubs(:new => http)
http
end
should "post to Airbrake when using an HTTP proxy" do
response = stub(:body => 'body')
http = stub(:post => response,
:read_timeout= => nil,
:open_timeout= => nil,
:use_ssl= => nil)
proxy = stub(:new => http)
Net::HTTP.stubs(:Proxy => proxy)
url = "http://airbrake.io:80#{Airbrake::Sender::NOTICES_URI}"
uri = URI.parse(url)
proxy_host = 'some.host'
proxy_port = 88
proxy_user = 'login'
proxy_pass = 'passwd'
send_exception(:proxy_host => proxy_host,
:proxy_port => proxy_port,
:proxy_user => proxy_user,
:proxy_pass => proxy_pass)
assert_received(http, :post) do |expect|
expect.with(uri.path, anything, Airbrake::HEADERS)
end
assert_received(Net::HTTP, :Proxy) do |expect|
expect.with(proxy_host, proxy_port, proxy_user, proxy_pass)
end
end
should "return the created group's id on successful posting" do
http = stub_http(:body => '<error-id type="integer">3799307</error-id>')
assert_equal "3799307", send_exception(:secure => false)
end
context "when encountering exceptions: " do
context "HTTP connection setup problems" do
should "not be rescued" do
proxy = stub()
proxy.stubs(:new).raises(NoMemoryError)
Net::HTTP.stubs(:Proxy => proxy)
assert_raise NoMemoryError do
build_sender.send(:setup_http_connection)
end
end
should "be logged" do
proxy = stub()
proxy.stubs(:new).raises(RuntimeError)
Net::HTTP.stubs(:Proxy => proxy)
sender = build_sender
sender.expects(:log).with(:error, includes('Failure initializing the HTTP connection'))
assert_raise RuntimeError do
sender.send(:setup_http_connection)
end
end
end
context "unexpected exception sending problems" do
should "be logged" do
sender = build_sender
sender.stubs(:setup_http_connection).raises(RuntimeError.new)
sender.expects(:log).with(:error, includes('Cannot send notification. Error'))
sender.send_to_airbrake("stuff")
end
should "return nil no matter what" do
sender = build_sender
sender.stubs(:setup_http_connection).raises(LocalJumpError)
assert_nothing_thrown do
assert_nil sender.send_to_airbrake("stuff")
end
end
end
should "return nil on failed posting" do
http = stub_http
http.stubs(:post).raises(Errno::ECONNREFUSED)
assert_equal nil, send_exception(:secure => false)
end
should "not fail when posting and a timeout exception occurs" do
http = stub_http
http.stubs(:post).raises(TimeoutError)
assert_nothing_thrown do
send_exception(:secure => false)
end
end
should "not fail when posting and a connection refused exception occurs" do
http = stub_http
http.stubs(:post).raises(Errno::ECONNREFUSED)
assert_nothing_thrown do
send_exception(:secure => false)
end
end
should "not fail when posting any http exception occurs" do
http = stub_http
Airbrake::Sender::HTTP_ERRORS.each do |error|
http.stubs(:post).raises(error)
assert_nothing_thrown do
send_exception(:secure => false)
end
end
end
end
context "SSL" do
should "post to the right url for non-ssl" do
http = stub_http
url = "http://airbrake.io:80#{Airbrake::Sender::NOTICES_URI}"
uri = URI.parse(url)
send_exception(:secure => false)
assert_received(http, :post) {|expect| expect.with(uri.path, anything, Airbrake::HEADERS) }
end
should "post to the right path for ssl" do
http = stub_http
send_exception(:secure => true)
assert_received(http, :post) {|expect| expect.with(Airbrake::Sender::NOTICES_URI, anything, Airbrake::HEADERS) }
end
should "verify the SSL peer when the use_ssl option is set to true" do
url = "https://airbrake.io#{Airbrake::Sender::NOTICES_URI}"
uri = URI.parse(url)
real_http = Net::HTTP.new(uri.host, uri.port)
real_http.stubs(:post => nil)
proxy = stub(:new => real_http)
Net::HTTP.stubs(:Proxy => proxy)
File.stubs(:exist?).with(OpenSSL::X509::DEFAULT_CERT_FILE).returns(false)
send_exception(:secure => true)
assert(real_http.use_ssl?)
assert_equal(OpenSSL::SSL::VERIFY_PEER, real_http.verify_mode)
assert_equal(Airbrake.configuration.local_cert_path, real_http.ca_file)
end
should "use the default DEFAULT_CERT_FILE if asked to" do
config = Airbrake::Configuration.new
config.use_system_ssl_cert_chain = true
sender = Airbrake::Sender.new(config)
assert(sender.use_system_ssl_cert_chain?)
http = sender.send(:setup_http_connection)
assert_not_equal http.ca_file, config.local_cert_path
end
should "verify the connection when the use_ssl option is set (VERIFY_PEER)" do
sender = build_sender(:secure => true)
http = sender.send(:setup_http_connection)
assert_equal(OpenSSL::SSL::VERIFY_PEER, http.verify_mode)
end
should "use the default cert (OpenSSL::X509::DEFAULT_CERT_FILE) only if explicitly told to" do
sender = build_sender(:secure => true)
http = sender.send(:setup_http_connection)
assert_equal(Airbrake.configuration.local_cert_path, http.ca_file)
File.stubs(:exist?).with(OpenSSL::X509::DEFAULT_CERT_FILE).returns(true)
sender = build_sender(:secure => true, :use_system_ssl_cert_chain => true)
http = sender.send(:setup_http_connection)
assert_not_equal(Airbrake.configuration.local_cert_path, http.ca_file)
assert_equal(OpenSSL::X509::DEFAULT_CERT_FILE, http.ca_file)
end
should "connect to the right port for ssl" do
stub_http
send_exception(:secure => true)
assert_received(Net::HTTP, :new) {|expect| expect.with("airbrake.io", 443) }
end
should "connect to the right port for non-ssl" do
stub_http
send_exception(:secure => false)
assert_received(Net::HTTP, :new) {|expect| expect.with("airbrake.io", 80) }
end
should "use ssl if secure" do
stub_http
send_exception(:secure => true, :host => 'example.org')
assert_received(Net::HTTP, :new) {|expect| expect.with('example.org', 443) }
end
should "not use ssl if not secure" do
stub_http
send_exception(:secure => false, :host => 'example.org')
assert_received(Net::HTTP, :new) {|expect| expect.with('example.org', 80) }
end
end
context "network timeouts" do
should "default the open timeout to 2 seconds" do
http = stub_http
send_exception
assert_received(http, :open_timeout=) {|expect| expect.with(2) }
end
should "default the read timeout to 5 seconds" do
http = stub_http
send_exception
assert_received(http, :read_timeout=) {|expect| expect.with(5) }
end
should "allow override of the open timeout" do
http = stub_http
send_exception(:http_open_timeout => 4)
assert_received(http, :open_timeout=) {|expect| expect.with(4) }
end
should "allow override of the read timeout" do
http = stub_http
send_exception(:http_read_timeout => 10)
assert_received(http, :read_timeout=) {|expect| expect.with(10) }
end
end
end