0
Fork 0
mirror of https://github.com/verdaccio/verdaccio.git synced 2024-12-23 22:27:34 -05:00
verdaccio/test/lib/request.js
Juan Picado @jotadeveloper 690b92a652
refactor: mirror and anti-loop
2018-06-23 20:59:27 +02:00

128 lines
3 KiB
JavaScript

// @flow
import _ from 'lodash';
import assert from 'assert';
import request from 'request';
import type {IRequestPromise} from '../types';
const requestData = Symbol('smart_request_data');
export class PromiseAssert extends Promise<any> implements IRequestPromise{
constructor(options: any) {
super(options);
}
status(expected: number) {
// $FlowFixMe
const selfData = this[requestData];
return injectResponse(this, this.then(function(body) {
try {
assert.equal(selfData.response.statusCode, expected);
} catch(err) {
selfData.error.message = err.message;
throw selfData.error;
}
return body;
}));
}
body_ok(expected: any) {
// $FlowFixMe
const selfData = this[requestData];
return injectResponse(this, this.then(function(body) {
try {
if (_.isRegExp(expected)) {
assert(body.ok.match(expected), '\'' + body.ok + '\' doesn\'t match ' + expected);
} else {
assert.equal(body.ok, expected);
}
assert.equal(body.error, null);
} catch(err) {
selfData.error.message = err.message;
throw selfData.error;
}
return body;
}));
}
body_error(expected: any) {
// $FlowFixMe
const selfData = this[requestData];
return injectResponse(this, this.then(function(body) {
try {
if (_.isRegExp(expected)) {
assert(body.error.match(expected), body.error + ' doesn\'t match ' + expected);
} else {
assert.equal(body.error, expected);
}
assert.equal(body.ok, null);
} catch(err) {
selfData.error.message = err.message;
throw selfData.error;
}
return body;
}));
}
request(callback: any) {
// $FlowFixMe
callback(this[requestData].request);
return this;
}
response(cb: any) {
// $FlowFixMe
const selfData = this[requestData];
return injectResponse(this, this.then(function(body) {
cb(selfData.response);
return body;
}));
}
send(data: any) {
// $FlowFixMe
this[requestData].request.end(data);
return this;
}
}
function injectResponse(smartObject: any, promise: Promise<any>): Promise<any> {
// $FlowFixMe
promise[requestData] = smartObject[requestData];
return promise;
}
function smartRequest(options: any): Promise<any> {
const smartObject: any = {};
smartObject[requestData] = {};
smartObject[requestData].error = Error();
Error.captureStackTrace(smartObject[requestData].error, smartRequest);
const promiseResult: Promise<any> = new PromiseAssert(function(resolve, reject) {
// store request reference on symbol
smartObject[requestData].request = request(options, function(err, res, body) {
if (err) {
return reject(err);
}
// store the response on symbol
smartObject[requestData].response = res;
// console.log("======>smartRequest RESPONSE: ", body);
resolve(body);
});
});
return injectResponse(smartObject, promiseResult);
}
export default smartRequest;