Skip to content

Commit bb94b6c

Browse files
homeyerhomeyer
committedJul 12, 2015
use nock to mock requests
1 parent 04947cf commit bb94b6c

File tree

1 file changed

+119
-77
lines changed

1 file changed

+119
-77
lines changed
 

‎test/index-spec.js

+119-77
Original file line numberDiff line numberDiff line change
@@ -1,92 +1,88 @@
11
var assert = require('assert')
22
, sinon = require('sinon')
3+
, nock = require('nock')
34
, prerender = require('../index')
45
, request = require('request')
56
, zlib = require('zlib')
67
, bot = 'Baiduspider+(+http://www.baidu.com/search/spider.htm)'
78
, user = 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_8_5) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/29.0.1547.76 Safari/537.36';
89

9-
function mockRequest(statusCode, body, headers) {
10-
return {
11-
on: function (name, f) {
12-
if (name === 'response') {
13-
f({
14-
statusCode: statusCode,
15-
headers: headers || {},
16-
on: function (name, f) {
17-
if (name === 'data') {
18-
f(body);
19-
} else if (name === 'end') {
20-
f();
21-
}
22-
},
23-
pipe: function (stream) {
24-
stream.write(body);
25-
stream.end();
26-
}
27-
});
28-
}
29-
return this;
30-
}
31-
};
32-
}
33-
34-
describe('Prerender', function(){
10+
describe ('Prerender', function(){
3511

3612
describe('#prerender', function(){
3713

3814
var sandbox, res, next;
3915

4016
beforeEach(function () {
17+
4118
sandbox = sinon.sandbox.create();
4219

4320
prerender.prerenderToken = 'MY_TOKEN';
4421
res = { writeHead: sandbox.stub(), end: sandbox.stub() };
4522
next = sandbox.stub();
46-
sandbox.stub(prerender, 'buildApiUrl').returns('http://google.com');
4723
});
4824

4925
afterEach(function () {
5026
sandbox.restore();
5127
});
5228

53-
it('should return a prerendered response with the returned status code and headers', function(){
54-
var req = { method: 'GET', url: '/', headers: { 'user-agent': bot } };
29+
it('should return a prerendered response with the returned status code and headers', function(done){
30+
var req = { method: 'GET', url: '/', headers: { 'user-agent': bot, host: 'google.com' } };
5531

56-
sandbox.stub(request, 'get').returns(mockRequest(301, '<html></html>', { 'Location': 'http://google.com'}));
32+
nock('http://service.prerender.io', {
33+
reqheaders: {
34+
'x-prerender-token': 'MY_TOKEN',
35+
'Accept-Encoding': 'gzip'
36+
}
37+
}).get('/http://google.com/')
38+
.reply(301, '<html></html>', {
39+
Location: 'http://google.com'
40+
});
41+
42+
res.end = sandbox.spy(function(){
43+
assert.equal(next.callCount, 0);
44+
assert.equal(res.writeHead.callCount, 1);
45+
assert.equal(res.end.callCount, 1);
46+
assert.deepEqual(res.writeHead.getCall(0).args[1], { 'location': 'http://google.com'});
47+
assert.equal(res.end.getCall(0).args[0], '<html></html>');
48+
assert.equal(res.writeHead.getCall(0).args[0], 301);
49+
done();
50+
});
5751

5852
prerender(req, res, next);
5953

60-
assert.equal(request.get.getCall(0).args[0].uri.href, 'http://google.com/');
61-
assert.equal(request.get.getCall(0).args[0].headers['X-Prerender-Token'], 'MY_TOKEN');
62-
assert.equal(request.get.getCall(0).args[0].headers['Accept-Encoding'], 'gzip');
63-
assert.equal(next.callCount, 0);
64-
assert.equal(res.writeHead.callCount, 1);
65-
assert.equal(res.end.callCount, 1);
66-
assert.deepEqual(res.writeHead.getCall(0).args[1], { 'Location': 'http://google.com'});
67-
assert.equal(res.end.getCall(0).args[0], '<html></html>');
68-
assert.equal(res.writeHead.getCall(0).args[0], 301);
6954
});
7055

7156

72-
it('should return a prerendered response if user is a bot by checking for _escaped_fragment_', function(){
73-
var req = { method: 'GET', url: '/path?_escaped_fragment_=', headers: { 'user-agent': user } };
57+
it('should return a prerendered response if user is a bot by checking for _escaped_fragment_', function(done){
58+
var req = { method: 'GET', url: '/path?_escaped_fragment_=', headers: { 'user-agent': user, host: 'google.com' } };
7459

75-
sandbox.stub(request, 'get').returns(mockRequest(200, '<html></html>'));
60+
nock('http://service.prerender.io', {
61+
reqheaders: {
62+
'x-prerender-token': 'MY_TOKEN',
63+
'Accept-Encoding': 'gzip'
64+
}
65+
})
66+
.get('/http://google.com/path')
67+
.query({_escaped_fragment_: ''})
68+
.reply(200, '<html></html>');
69+
70+
res.end = sandbox.spy(function(){
71+
assert.equal(next.callCount, 0);
72+
assert.equal(res.writeHead.callCount, 1);
73+
assert.equal(res.writeHead.getCall(0).args[0], 200);
74+
assert.equal(res.end.callCount, 1);
75+
assert.equal(res.end.getCall(0).args[0], '<html></html>');
76+
done();
77+
});
7678

7779
prerender(req, res, next);
7880

79-
assert.equal(next.callCount, 0);
80-
assert.equal(res.writeHead.callCount, 1);
81-
assert.equal(res.writeHead.getCall(0).args[0], 200);
82-
assert.equal(res.end.callCount, 1);
83-
assert.equal(res.end.getCall(0).args[0], '<html></html>');
8481
});
8582

8683
it('should return a prerendered gzipped response', function(done){
84+
var req = { method: 'GET', url: '/path?_escaped_fragment_=', headers: { 'user-agent': user, host: 'google.com' } };
8785

88-
var req = { method: 'GET', url: '/path?_escaped_fragment_=', headers: { 'user-agent': user } };
89-
// we're dealing with asynchonous gzip so we can only assert on res.end. If it's not called, the default mocha timeout of 2s will fail the test
9086
res.end = function (content) {
9187
assert.equal(res.writeHead.callCount, 1);
9288
assert.equal(res.writeHead.getCall(0).args[0], 200);
@@ -96,7 +92,15 @@ describe('Prerender', function(){
9692
};
9793

9894
zlib.gzip(new Buffer('<html></html>', 'utf-8'), function (err, zipped) {
99-
sandbox.stub(request, 'get').returns(mockRequest(200, zipped, {'content-encoding': 'gzip'}));
95+
nock('http://service.prerender.io', {
96+
reqheaders: {
97+
'x-prerender-token': 'MY_TOKEN',
98+
'Accept-Encoding': 'gzip'
99+
}
100+
})
101+
.get('/http://google.com/path')
102+
.query({_escaped_fragment_: ''})
103+
.reply(200, [zipped], {'content-encoding': 'gzip'});
100104

101105
prerender(req, res, next);
102106
});
@@ -154,19 +158,34 @@ describe('Prerender', function(){
154158
assert.equal(res.end.callCount, 0);
155159
});
156160

157-
it('should return a prerendered response if the url is part of the regex specific whitelist', function(){
158-
var req = { method: 'GET', url: '/search/things?query=blah&_escaped_fragment_=', headers: { 'user-agent': bot } };
161+
it('should return a prerendered response if the url is part of the regex specific whitelist', function(done){
162+
var req = { method: 'GET', url: '/search/things?query=blah&_escaped_fragment_=', headers: { 'user-agent': bot, host: 'google.com' } };
159163

160-
sandbox.stub(request, 'get').returns(mockRequest(200, '<html></html>'));
164+
nock('http://service.prerender.io', {
165+
reqheaders: {
166+
'x-prerender-token': 'MY_TOKEN',
167+
'Accept-Encoding': 'gzip'
168+
}
169+
})
170+
.get('/http://google.com/search/things')
171+
.query({
172+
_escaped_fragment_: '',
173+
query: 'blah'
174+
})
175+
.reply(200, '<html></html>');
176+
177+
res.end = sandbox.spy(function(){
178+
assert.equal(next.callCount, 0);
179+
assert.equal(res.writeHead.callCount, 1);
180+
assert.equal(res.writeHead.getCall(0).args[0], 200);
181+
assert.equal(res.end.callCount, 1);
182+
assert.equal(res.end.getCall(0).args[0], '<html></html>');
183+
done();
184+
});
161185

162186
prerender.whitelisted(['^/search.*query', '/help'])(req, res, next);
163-
164187
delete prerender.whitelist;
165-
assert.equal(next.callCount, 0);
166-
assert.equal(res.writeHead.callCount, 1);
167-
assert.equal(res.writeHead.getCall(0).args[0], 200);
168-
assert.equal(res.end.callCount, 1);
169-
assert.equal(res.end.getCall(0).args[0], '<html></html>');
188+
170189
});
171190

172191
it('should call next() if the url is part of the regex specific blacklist', function(){
@@ -180,19 +199,31 @@ describe('Prerender', function(){
180199
assert.equal(res.end.callCount, 0);
181200
});
182201

183-
it('should return a prerendered response if the url is not part of the regex specific blacklist', function(){
184-
var req = { method: 'GET', url: '/profile/search/blah', headers: { 'user-agent': bot } };
202+
it('should return a prerendered response if the url is not part of the regex specific blacklist', function(done){
203+
var req = { method: 'GET', url: '/profile/search/blah', headers: { 'user-agent': bot, host: 'google.com' } };
185204

186-
sandbox.stub(request, 'get').returns(mockRequest(200, '<html></html>'));
187205

188-
prerender.blacklisted(['^/search', '/help'])(req, res, next);
206+
nock('http://service.prerender.io', {
207+
reqheaders: {
208+
'x-prerender-token': 'MY_TOKEN',
209+
'Accept-Encoding': 'gzip'
210+
}
211+
})
212+
.get('/http://google.com/profile/search/blah')
213+
.reply(200, '<html></html>');
189214

215+
res.end = sandbox.spy(function(){
216+
assert.equal(next.callCount, 0);
217+
assert.equal(res.writeHead.callCount, 1);
218+
assert.equal(res.writeHead.getCall(0).args[0], 200);
219+
assert.equal(res.end.callCount, 1);
220+
assert.equal(res.end.getCall(0).args[0], '<html></html>');
221+
done();
222+
});
223+
224+
prerender.blacklisted(['^/search', '/help'])(req, res, next);
190225
delete prerender.blacklist;
191-
assert.equal(next.callCount, 0);
192-
assert.equal(res.writeHead.callCount, 1);
193-
assert.equal(res.writeHead.getCall(0).args[0], 200);
194-
assert.equal(res.end.callCount, 1);
195-
assert.equal(res.end.getCall(0).args[0], '<html></html>');
226+
196227
});
197228

198229
it('should call next() if the referer is part of the regex specific blacklist', function(){
@@ -206,19 +237,30 @@ describe('Prerender', function(){
206237
assert.equal(res.end.callCount, 0);
207238
});
208239

209-
it('should return a prerendered response if the referer is not part of the regex specific blacklist', function(){
210-
var req = { method: 'GET', url: '/api/results', headers: { referer: '/profile/search', 'user-agent': bot } };
240+
it('should return a prerendered response if the referer is not part of the regex specific blacklist', function(done){
241+
var req = { method: 'GET', url: '/api/results', headers: { referer: '/profile/search', 'user-agent': bot, host: 'google.com' } };
211242

212-
sandbox.stub(request, 'get').returns(mockRequest(200, '<html></html>'));
243+
nock('http://service.prerender.io', {
244+
reqheaders: {
245+
'x-prerender-token': 'MY_TOKEN',
246+
'Accept-Encoding': 'gzip'
247+
}
248+
})
249+
.get('/http://google.com/api/results')
250+
.reply(200, '<html></html>');
213251

214-
prerender.blacklisted(['^/search', '/help'])(req, res, next);
252+
res.end = sandbox.spy(function(){
253+
assert.equal(next.callCount, 0);
254+
assert.equal(res.writeHead.callCount, 1);
255+
assert.equal(res.writeHead.getCall(0).args[0], 200);
256+
assert.equal(res.end.callCount, 1);
257+
assert.equal(res.end.getCall(0).args[0], '<html></html>');
258+
done();
259+
});
215260

261+
prerender.blacklisted(['^/search', '/help'])(req, res, next);
216262
delete prerender.blacklist;
217-
assert.equal(next.callCount, 0);
218-
assert.equal(res.writeHead.callCount, 1);
219-
assert.equal(res.writeHead.getCall(0).args[0], 200);
220-
assert.equal(res.end.callCount, 1);
221-
assert.equal(res.end.getCall(0).args[0], '<html></html>');
263+
222264
});
223265

224266
it('should return a prerendered response if a string is returned from beforeRender', function(){

0 commit comments

Comments
 (0)
Please sign in to comment.