forked from sinatra/sinatra
-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_test.rb
152 lines (121 loc) · 3.54 KB
/
test_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
require File.dirname(__FILE__) + '/helper'
require 'yaml'
# silence deprecation warning when requiring sinatra/test
$VERBOSE, v = nil, $VERBOSE
require 'sinatra/test'
$VERBOSE = v
class TestTest < Test::Unit::TestCase
include Sinatra::Test
def request
YAML.load(body)
end
def request_body
request['test.body']
end
def request_params
YAML.load(request['test.params'])
end
setup do
mock_app {
%w[get head post put delete].each { |verb|
send(verb, '/') do
redirect '/redirected' if params[:redirect]
env.update('test.body' => request.body.read)
env.update('test.params' => params.to_yaml)
env.to_yaml
end
}
get '/redirected' do
"you've been redirected"
end
}
end
it 'allows GET/HEAD/POST/PUT/DELETE' do
get '/'
assert_equal('GET', request['REQUEST_METHOD'])
post '/'
assert_equal('POST', request['REQUEST_METHOD'])
put '/'
assert_equal('PUT', request['REQUEST_METHOD'])
delete '/'
assert_equal('DELETE', request['REQUEST_METHOD'])
head '/'
assert response.headers['Content-Length'].to_i > 0
assert_equal('', body)
end
it 'allows to specify a body' do
post '/', '42'
assert_equal '42', request_body
end
it 'allows to specify params' do
get '/', :foo => 'bar'
assert_equal 'bar', request_params['foo']
end
it 'supports nested params' do
get '/', :foo => { :x => 'y', :chunky => 'bacon' }
assert_equal "y", request_params['foo']['x']
assert_equal "bacon", request_params['foo']['chunky']
end
it 'provides easy access to response status and body' do
get '/'
assert_equal 200, status
assert body =~ /^---/
end
it 'delegates methods to @response' do
get '/'
assert ok?
end
it 'follows redirect' do
get '/', :redirect => true
follow!
assert_equal "you've been redirected", body
end
it 'provides sugar for common HTTP headers' do
get '/', :env => { :accept => 'text/plain' }
assert_equal 'text/plain', request['HTTP_ACCEPT']
get '/', :env => { :agent => 'TATFT' }
assert_equal 'TATFT', request['HTTP_USER_AGENT']
get '/', :env => { :host => '1.2.3.4' }
assert_equal '1.2.3.4', request['HTTP_HOST']
get '/', :env => { :session => {'foo' => 'bar'} }
assert_equal({'foo' => 'bar'}, request['rack.session'])
get '/', :env => { :cookies => 'foo' }
assert_equal 'foo', request['HTTP_COOKIE']
get '/', :env => { :content_type => 'text/plain' }
assert_equal 'text/plain', request['CONTENT_TYPE']
end
it 'allow to test session easily' do
app = mock_app(Sinatra::Default) {
get '/' do
session['foo'] = 'bar'
200
end
post '/' do
assert_equal 'bar', session['foo']
session['foo'] || "blah"
end
}
browser = Sinatra::TestHarness.new(app)
browser.get '/'
browser.post '/', {}, :session => { 'foo' => 'bar' }
assert_equal 'bar', browser.response.body
end
it 'yields the request object to the block before invoking the application' do
called = false
get '/' do |req|
called = true
assert req.kind_of?(Rack::MockRequest)
end
assert called
end
it 'sets the environment to :test on include' do
Sinatra::Default.set(:environment, :production)
Class.new { include Sinatra::Test }
assert_equal :test, Sinatra::Default.environment
end
def test_TestHarness
session = Sinatra::TestHarness.new(@app)
response = session.get('/')
assert_equal 200, response.status
end
end