-
Notifications
You must be signed in to change notification settings - Fork 760
/
agent.js
61 lines (49 loc) 路 1.37 KB
/
agent.js
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
/**
* Module dependencies.
*/
var Agent = require('superagent').agent;
var methods = require('methods');
var http = require('http');
var Test = require('./test');
/**
* Expose `Agent`.
*/
module.exports = TestAgent;
/**
* Initialize a new `TestAgent`.
*
* @param {Function|Server} app
* @param {Object} options
* @api public
*/
function TestAgent(app, options) {
if (!(this instanceof TestAgent)) return new TestAgent(app, options);
if (typeof app === 'function') app = http.createServer(app); // eslint-disable-line no-param-reassign
if (options) {
this._ca = options.ca;
this._key = options.key;
this._cert = options.cert;
}
Agent.call(this);
this.app = app;
}
/**
* Inherits from `Agent.prototype`.
*/
TestAgent.prototype.__proto__ = Agent.prototype;
// override HTTP verb methods
methods.forEach(function(method) {
TestAgent.prototype[method] = function(url, fn) { // eslint-disable-line no-unused-vars
var req = new Test(this.app, method.toUpperCase(), url);
req.ca(this._ca);
req.cert(this._cert);
req.key(this._key);
req.on('response', this._saveCookies.bind(this));
req.on('redirect', this._saveCookies.bind(this));
req.on('redirect', this._attachCookies.bind(this, req));
this._attachCookies(req);
this._setDefaults(req);
return req;
};
});
TestAgent.prototype.del = TestAgent.prototype.delete;