openethereum/js-old/test/mockRpc.js

101 lines
2.9 KiB
JavaScript
Raw Normal View History

2017-09-05 11:21:18 +02:00
// Copyright 2015-2017 Parity Technologies (UK) Ltd.
// This file is part of Parity.
// Parity is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
// Parity is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
// You should have received a copy of the GNU General Public License
// along with Parity. If not, see <http://www.gnu.org/licenses/>.
import nock from 'nock';
import { Server as MockWsServer } from 'mock-socket';
import { isFunction } from '@parity/api/lib/util/types';
2017-09-05 11:21:18 +02:00
export const TEST_HTTP_URL = 'http://localhost:6688';
export const TEST_WS_URL = 'ws://localhost:8866';
export function mockHttp (requests) {
nock.cleanAll();
let scope = nock(TEST_HTTP_URL);
requests.forEach((request, index) => {
scope = scope
.post('/')
.reply(request.code || 200, (uri, body) => {
if (body.method !== request.method) {
return {
error: `Invalid method ${body.method}, expected ${request.method}`
};
}
scope.body = scope.body || {};
scope.body[request.method] = body;
return request.reply;
});
});
return scope;
}
export function mockWs (requests) {
let mockServer = new MockWsServer(TEST_WS_URL);
const scope = { requests: 0, body: {}, server: mockServer };
scope.isDone = () => scope.requests === requests.length;
scope.stop = () => {
if (mockServer) {
mockServer.stop();
mockServer = null;
}
};
mockServer.on('message', (_body) => {
const body = JSON.parse(_body);
const request = requests[scope.requests];
const reply = request.reply;
const response = reply.error
? { id: body.id, error: { code: reply.error.code, message: reply.error.message } }
: { id: body.id, result: reply };
scope.body[request.method] = body;
scope.requests++;
mockServer.send(JSON.stringify(response));
if (request.method.match('subscribe') && request.subscription) {
mockServer.send(JSON.stringify(request.subscription));
}
});
return scope;
}
export function endpointTest (instance, moduleId, name) {
describe(name, () => {
it(`has the ${moduleId}.${name} endpoint`, () => {
expect(isFunction(instance[moduleId][name])).to.be.ok;
});
it(`maps to ${moduleId}_${name} via RPC`, () => {
const scope = mockHttp([{ method: `${moduleId}_${name}`, reply: {} }]);
return instance[moduleId][name]()
.then(() => {
expect(scope.isDone()).to.be.true;
})
.catch(() => {
nock.cleanAll();
});
});
});
}