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
|
# encoding: UTF-8
require 'rack/test'
require 'prometheus/client/rack/exporter'
xdescribe Prometheus::Client::Rack::Exporter do
include Rack::Test::Methods
let(:registry) do
Prometheus::Client::Registry.new
end
let(:app) do
app = ->(_) { [200, { 'Content-Type' => 'text/html' }, ['OK']] }
Prometheus::Client::Rack::Exporter.new(app, registry: registry)
end
context 'when requesting app endpoints' do
it 'returns the app response' do
get '/foo'
expect(last_response).to be_ok
expect(last_response.body).to eql('OK')
end
end
context 'when requesting /metrics' do
text = Prometheus::Client::Formats::Text
shared_examples 'ok' do |headers, fmt|
it "responds with 200 OK and Content-Type #{fmt::CONTENT_TYPE}" do
registry.counter(:foo, 'foo counter').increment({}, 9)
get '/metrics', nil, headers
expect(last_response.status).to eql(200)
expect(last_response.header['Content-Type']).to eql(fmt::CONTENT_TYPE)
expect(last_response.body).to eql(fmt.marshal(registry))
end
end
shared_examples 'not acceptable' do |headers|
it 'responds with 406 Not Acceptable' do
message = 'Supported media types: text/plain'
get '/metrics', nil, headers
expect(last_response.status).to eql(406)
expect(last_response.header['Content-Type']).to eql('text/plain')
expect(last_response.body).to eql(message)
end
end
context 'when client does not send a Accept header' do
include_examples 'ok', {}, text
end
context 'when client accpets any media type' do
include_examples 'ok', { 'HTTP_ACCEPT' => '*/*' }, text
end
context 'when client requests application/json' do
include_examples 'not acceptable', 'HTTP_ACCEPT' => 'application/json'
end
context 'when client requests text/plain' do
include_examples 'ok', { 'HTTP_ACCEPT' => 'text/plain' }, text
end
context 'when client uses different white spaces in Accept header' do
accept = 'text/plain;q=1.0 ; version=0.0.4'
include_examples 'ok', { 'HTTP_ACCEPT' => accept }, text
end
context 'when client does not include quality attribute' do
accept = 'application/json;q=0.5, text/plain'
include_examples 'ok', { 'HTTP_ACCEPT' => accept }, text
end
context 'when client accepts some unknown formats' do
accept = 'text/plain;q=0.3, proto/buf;q=0.7'
include_examples 'ok', { 'HTTP_ACCEPT' => accept }, text
end
context 'when client accepts only unknown formats' do
accept = 'fancy/woo;q=0.3, proto/buf;q=0.7'
include_examples 'not acceptable', 'HTTP_ACCEPT' => accept
end
context 'when client accepts unknown formats and wildcard' do
accept = 'fancy/woo;q=0.3, proto/buf;q=0.7, */*;q=0.1'
include_examples 'ok', { 'HTTP_ACCEPT' => accept }, text
end
end
end
|
