openethereum/js/test/mockRpc.js
2016-12-11 19:30:54 +01:00

97 lines
2.7 KiB
JavaScript

// Copyright 2015, 2016 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 '../src/api/util/types';
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) {
const scope = { requests: 0, body: {} };
let mockServer = new MockWsServer(TEST_WS_URL);
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));
});
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();
});
});
});
}