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
|
require 'helper'
describe EventMachine::HttpRequest do
class EmptyMiddleware; end
class GlobalMiddleware
def response(resp)
resp.response_header['X-Global'] = 'middleware'
end
end
it "should accept middleware" do
EventMachine.run {
lambda {
conn = EM::HttpRequest.new('http://127.0.0.1:8090')
conn.use ResponseMiddleware
conn.use EmptyMiddleware
EM.stop
}.should_not raise_error
}
end
context "configuration" do
class ConfigurableMiddleware
def initialize(conf, &block)
@conf = conf
@block = block
end
def response(resp)
resp.response_header['X-Conf'] = @conf
resp.response_header['X-Block'] = @block.call
end
end
it "should accept middleware initialization parameters" do
EventMachine.run {
conn = EM::HttpRequest.new('http://127.0.0.1:8090')
conn.use ConfigurableMiddleware, 'conf-value' do
'block-value'
end
req = conn.get
req.callback {
req.response_header['X-Conf'].should match('conf-value')
req.response_header['X-Block'].should match('block-value')
EM.stop
}
}
end
end
context "request" do
class ResponseMiddleware
def response(resp)
resp.response_header['X-Header'] = 'middleware'
resp.response = 'Hello, Middleware!'
end
end
it "should execute response middleware before user callbacks" do
EventMachine.run {
conn = EM::HttpRequest.new('http://127.0.0.1:8090')
conn.use ResponseMiddleware
req = conn.get
req.callback {
req.response_header['X-Header'].should match('middleware')
req.response.should match('Hello, Middleware!')
EM.stop
}
}
end
it "should execute global response middleware before user callbacks" do
EventMachine.run {
EM::HttpRequest.use GlobalMiddleware
conn = EM::HttpRequest.new('http://127.0.0.1:8090')
req = conn.get
req.callback {
req.response_header['X-Global'].should match('middleware')
EM.stop
}
}
end
end
context "request" do
class RequestMiddleware
def request(client, head, body)
head['X-Middleware'] = 'middleware' # insert new header
body += ' modified' # modify post body
[head, body]
end
end
it "should execute request middleware before dispatching request" do
EventMachine.run {
conn = EventMachine::HttpRequest.new('http://127.0.0.1:8090/')
conn.use RequestMiddleware
req = conn.post :body => "data"
req.callback {
req.response_header.status.should == 200
req.response.should match(/data modified/)
EventMachine.stop
}
}
end
end
context "jsonify" do
class JSONify
def request(client, head, body)
[head, MultiJson.dump(body)]
end
def response(resp)
resp.response = MultiJson.load(resp.response)
end
end
it "should use middleware to JSON encode and JSON decode the body" do
EventMachine.run {
conn = EventMachine::HttpRequest.new('http://127.0.0.1:8090/')
conn.use JSONify
req = conn.post :body => {:ruby => :hash}
req.callback {
req.response_header.status.should == 200
req.response.should == {"ruby" => "hash"}
EventMachine.stop
}
}
end
end
end
|
