Skip to content

Commit df3ee1c

Browse files
committed
refactor tests
1 parent 132c1cc commit df3ee1c

File tree

1 file changed

+41
-26
lines changed

1 file changed

+41
-26
lines changed

test/test.js

Lines changed: 41 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -14,24 +14,24 @@ var createHTTPServer = server.createHTTPServer
1414
var createHTTP2Server = server.createHTTP2Server
1515

1616
var topDescribe = function (type, createServer) {
17-
var wrapper = function wrapper (req) {
18-
if (type === 'http2') {
19-
return req.http2()
20-
}
17+
var options
2118

22-
return req
19+
if (type === 'http2') {
20+
options = { http2: true }
2321
}
2422

2523
describe('onHeaders(res, listener)', function () {
2624
it('should fire after setHeader', function (done) {
27-
wrapper(request(createServer(echoListener)).get('/'))
25+
request(createServer(echoListener), options)
26+
.get('/')
2827
.expect('X-Outgoing-Echo', 'test')
2928
.expect(200, done)
3029
})
3130
})
3231

3332
it('should fire after setHeader', function (done) {
34-
wrapper(request(createServer(echoListener)).get('/'))
33+
request(createServer(echoListener), options)
34+
.get('/')
3535
.expect('X-Outgoing-Echo', 'test')
3636
.expect(200, done)
3737
})
@@ -44,7 +44,8 @@ var topDescribe = function (type, createServer) {
4444
res.write('1')
4545
}
4646

47-
wrapper(request(server).get('/'))
47+
request(server, options)
48+
.get('/')
4849
.expect('X-Outgoing-Echo', 'test')
4950
.expect(200, '1', done)
5051
})
@@ -58,7 +59,8 @@ var topDescribe = function (type, createServer) {
5859
this.setHeader('X-Headers', getAllHeaderNames(this).join(','))
5960
}
6061

61-
wrapper(request(server).get('/'))
62+
request(server, options)
63+
.get('/')
6264
.expect('X-Headers', '')
6365
.expect(200, done)
6466
})
@@ -77,7 +79,8 @@ var topDescribe = function (type, createServer) {
7779
count++
7880
}
7981

80-
wrapper(request(server).get('/'))
82+
request(server, options)
83+
.get('/')
8184
.expect(200, function (err) {
8285
if (err) return done(err)
8386
assert.strictEqual(count, 1)
@@ -95,7 +98,8 @@ var topDescribe = function (type, createServer) {
9598
res.setHeader('X-Outgoing', 'test')
9699
}
97100

98-
wrapper(request(server).get('/'))
101+
request(server, options)
102+
.get('/')
99103
.expect('X-Outgoing-Echo', 'test,3,2,1')
100104
.expect(200, done)
101105
})
@@ -111,15 +115,16 @@ var topDescribe = function (type, createServer) {
111115
it('should be required', function (done) {
112116
var server = createServer()
113117

114-
wrapper(request(server)
115-
.get('/'))
118+
request(server, options)
119+
.get('/')
116120
.expect(500, /listener.*function/, done)
117121
})
118122

119123
it('should only accept function', function (done) {
120124
var server = createServer(42)
121125

122-
wrapper(request(server).get('/'))
126+
request(server, options)
127+
.get('/')
123128
.expect(500, /listener.*function/, done)
124129
})
125130
})
@@ -129,7 +134,8 @@ var topDescribe = function (type, createServer) {
129134
it('should be available in listener', function (done) {
130135
var server = createServer(echoListener)
131136

132-
wrapper(request(server).get('/'))
137+
request(server, options)
138+
.get('/')
133139
.expect('X-Outgoing-Echo', 'test')
134140
.expect(200, done)
135141
})
@@ -147,7 +153,8 @@ var topDescribe = function (type, createServer) {
147153
this.setHeader('X-Status', this.statusCode)
148154
}
149155

150-
wrapper(request(server).get('/'))
156+
request(server, options)
157+
.get('/')
151158
.expect('X-Status', '201')
152159
.expect(201, done)
153160
})
@@ -164,7 +171,8 @@ var topDescribe = function (type, createServer) {
164171
this.statusCode = 202
165172
}
166173

167-
wrapper(request(server).get('/'))
174+
request(server, options)
175+
.get('/')
168176
.expect('X-Status', '201')
169177
.expect(202, done)
170178
})
@@ -176,7 +184,8 @@ var topDescribe = function (type, createServer) {
176184
res.writeHead() // error
177185
}
178186

179-
wrapper(request(server).get('/'))
187+
request(server, options)
188+
.get('/')
180189
.expect(500, done)
181190
})
182191

@@ -196,12 +205,13 @@ var topDescribe = function (type, createServer) {
196205
server = http2.createServer(callbackServer)
197206
}
198207

199-
wrapper(request(server).get('/'))
208+
request(server, options)
209+
.get('/')
200210
.expect(200, function (err, res) {
201211
if (err) return done(err)
202212

203-
wrapper(request(server)
204-
.get('/attach'))
213+
request(server, options)
214+
.get('/attach')
205215
.expect(200, res.text, done)
206216
})
207217
})
@@ -216,7 +226,8 @@ var topDescribe = function (type, createServer) {
216226
res.writeHead(200, 'OK')
217227
}
218228

219-
wrapper(request(server).get('/'))
229+
request(server, options)
230+
.get('/')
220231
.expect('X-Outgoing-Echo', 'test')
221232
.expect(200, done)
222233
})
@@ -230,7 +241,8 @@ var topDescribe = function (type, createServer) {
230241
res.writeHead(200, 'OK', { 'X-Outgoing': 'test' })
231242
}
232243

233-
wrapper(request(server).get('/'))
244+
request(server, options)
245+
.get('/')
234246
.expect('X-Outgoing-Echo', 'test')
235247
.expect(200, done)
236248
})
@@ -249,7 +261,8 @@ var topDescribe = function (type, createServer) {
249261
this.setHeader('X-Outgoing-Echo', this.getHeader('X-Outgoing'))
250262
}
251263

252-
wrapper(request(server).get('/'))
264+
request(server, options)
265+
.get('/')
253266
.expect('X-Status', '201')
254267
.expect('X-Outgoing-Echo', 'test')
255268
.expect(201, done)
@@ -267,7 +280,8 @@ var topDescribe = function (type, createServer) {
267280
this.setHeader('X-Outgoing-Echo', this.getHeader('X-Outgoing'))
268281
}
269282

270-
wrapper(request(server).get('/'))
283+
request(server, options)
284+
.get('/')
271285
.expect('X-Status', '201')
272286
.expect('X-Outgoing-Echo', 'test')
273287
.expect(201, done)
@@ -287,7 +301,8 @@ var topDescribe = function (type, createServer) {
287301
this.setHeader('X-Outgoing-Echo', this.getHeader('X-Outgoing'))
288302
}
289303

290-
wrapper(request(server).get('/'))
304+
request(server, options)
305+
.get('/')
291306
.expect('X-Status', '201')
292307
.expect('X-Outgoing-Echo', 'test')
293308
.expect(201, done)

0 commit comments

Comments
 (0)