/
test_app_status.rb
91 lines (66 loc) · 1.6 KB
/
test_app_status.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
# frozen_string_literal: true
require_relative "helper"
require "puma/app/status"
require "rack"
class TestAppStatus < Minitest::Test
parallelize_me!
class FakeServer
def initialize
@status = :running
end
attr_reader :status
def stop
@status = :stop
end
def halt
@status = :halt
end
def stats_json
"{}"
end
end
def setup
@server = FakeServer.new
@app = Puma::App::Status.new(@server)
end
def lint(uri)
app = Rack::Lint.new @app
mock_env = Rack::MockRequest.env_for uri
app.call mock_env
end
def test_bad_token
@app.instance_variable_set(:@auth_token, "abcdef")
status, _, _ = lint('/whatever')
assert_equal 403, status
end
def test_good_token
@app.instance_variable_set(:@auth_token, "abcdef")
status, _, _ = lint('/whatever?token=abcdef')
assert_equal 404, status
end
def test_unsupported
status, _, _ = lint('/not-real')
assert_equal 404, status
end
def test_stop
status, _ , app = lint('/stop')
assert_equal :stop, @server.status
assert_equal 200, status
assert_equal ['{ "status": "ok" }'], app.enum_for.to_a
end
def test_halt
status, _ , app = lint('/halt')
assert_equal :halt, @server.status
assert_equal 200, status
assert_equal ['{ "status": "ok" }'], app.enum_for.to_a
end
def test_stats
status, _ , app = lint('/stats')
assert_equal 200, status
assert_equal ['{}'], app.enum_for.to_a
end
def test_alternate_location
status, _ , _ = lint('__alternatE_location_/stats')
assert_equal 200, status
end
end