Files
xahau.js/test/mock-rippled.js
2015-08-21 14:24:13 -07:00

275 lines
11 KiB
JavaScript

'use strict';
const _ = require('lodash');
const assert = require('assert');
const WebSocketServer = require('ws').Server;
const EventEmitter2 = require('eventemitter2').EventEmitter2;
const fixtures = require('./fixtures/api/rippled');
const addresses = require('./fixtures/addresses');
const hashes = require('./fixtures/hashes');
const transactionsResponse = require('./fixtures/api/rippled/account-tx');
const accountLinesResponse = require('./fixtures/api/rippled/account-lines');
const fullLedger = require('./fixtures/ledger-full-38129.json');
function isUSD(json) {
return json === 'USD' || json === '0000000000000000000000005553440000000000';
}
function isBTC(json) {
return json === 'BTC' || json === '0000000000000000000000004254430000000000';
}
function createResponse(request, response, overrides = {}) {
const result = _.assign({}, response.result, overrides);
const change = response.result && !_.isEmpty(overrides) ?
{id: request.id, result: result} : {id: request.id};
return JSON.stringify(_.assign({}, response, change));
}
function createLedgerResponse(request, response) {
const newResponse = JSON.parse(createResponse(request, response));
if (newResponse.result && newResponse.result.ledger) {
if (!request.transactions) {
delete newResponse.result.ledger.transactions;
}
if (!request.accounts) {
delete newResponse.result.ledger.accountState;
}
// the following fields were not in the ledger response in the past
if (newResponse.result.ledger.close_flags === undefined) {
newResponse.result.ledger.close_flags = 0;
}
if (newResponse.result.ledger.parent_close_time === undefined) {
newResponse.result.ledger.parent_close_time =
newResponse.result.ledger.close_time - 10;
}
}
return JSON.stringify(newResponse);
}
module.exports = function(port) {
const mock = new WebSocketServer({port: port});
_.assign(mock, EventEmitter2.prototype);
const close = mock.close;
mock.close = function() {
if (mock.expectedRequests !== undefined) {
const allRequestsMade = _.every(mock.expectedRequests, function(counter) {
return counter === 0;
});
if (!allRequestsMade) {
const json = JSON.stringify(mock.expectedRequests, null, 2);
const indent = ' ';
const indented = indent + json.replace(/\n/g, '\n' + indent);
assert(false, 'Not all expected requests were made:\n' + indented);
}
}
close.call(mock);
};
mock.expect = function(expectedRequests) {
mock.expectedRequests = expectedRequests;
};
mock.once('connection', function(conn) {
conn.on('message', function(requestJSON) {
const request = JSON.parse(requestJSON);
mock.emit('request_' + request.command, request, conn);
});
});
mock.onAny(function() {
if (this.event.indexOf('request_') !== 0) {
return;
}
if (mock.listeners(this.event).length === 0) {
throw new Error('No event handler registered for ' + this.event);
}
if (mock.expectedRequests === undefined) {
return; // TODO: fail here to require expectedRequests
}
const expectedCount = mock.expectedRequests[this.event];
if (expectedCount === undefined || expectedCount === 0) {
throw new Error('Unexpected request: ' + this.event);
}
mock.expectedRequests[this.event] -= 1;
});
mock.on('request_server_info', function(request, conn) {
assert.strictEqual(request.command, 'server_info');
if (mock.returnErrorOnServerInfo) {
conn.send(createResponse(request, fixtures.server_info_error));
} else {
conn.send(createResponse(request, fixtures.server_info));
}
});
mock.on('request_subscribe', function(request, conn) {
assert.strictEqual(request.command, 'subscribe');
if (request.accounts) {
assert(_.indexOf(_.values(addresses), request.accounts[0]) !== -1);
} else {
assert.deepEqual(request.streams, ['ledger', 'server']);
}
conn.send(createResponse(request, fixtures.subscribe));
});
mock.on('request_unsubscribe', function(request, conn) {
assert.strictEqual(request.command, 'unsubscribe');
if (request.accounts) {
assert(_.indexOf(_.values(addresses), request.accounts[0]) !== -1);
} else {
assert.deepEqual(request.streams, ['ledger', 'server']);
}
conn.send(createResponse(request, fixtures.unsubscribe));
});
mock.on('request_account_info', function(request, conn) {
assert.strictEqual(request.command, 'account_info');
if (request.account === addresses.ACCOUNT) {
conn.send(createResponse(request, fixtures.account_info.normal));
} else if (request.account === addresses.NOTFOUND) {
conn.send(createResponse(request, fixtures.account_info.notfound));
} else if (request.account === addresses.THIRD_ACCOUNT) {
const response = _.assign({}, fixtures.account_info.normal);
response.Account = addresses.THIRD_ACCOUNT;
conn.send(createResponse(request, response));
} else {
assert(false, 'Unrecognized account address: ' + request.account);
}
});
mock.on('request_ledger', function(request, conn) {
assert.strictEqual(request.command, 'ledger');
if (request.ledger_index === 34) {
conn.send(createLedgerResponse(request, fixtures.ledgerNotFound));
} else if (request.ledger_index === 9038215) {
conn.send(createLedgerResponse(request, fixtures.ledgerWithoutCloseTime));
} else if (request.ledger_index === 38129) {
const response = _.assign({}, fixtures.ledger,
{result: {ledger: fullLedger}});
conn.send(createLedgerResponse(request, response));
} else {
conn.send(createLedgerResponse(request, fixtures.ledger));
}
});
mock.on('request_tx', function(request, conn) {
assert.strictEqual(request.command, 'tx');
if (request.transaction === hashes.VALID_TRANSACTION_HASH) {
conn.send(createResponse(request, fixtures.tx.Payment));
} else if (request.transaction ===
'4FB3ADF22F3C605E23FAEFAA185F3BD763C4692CAC490D9819D117CD33BFAA1B') {
conn.send(createResponse(request, fixtures.tx.AccountSet));
} else if (request.transaction ===
'8925FC8844A1E930E2CC76AD0A15E7665AFCC5425376D548BB1413F484C31B8C') {
conn.send(createResponse(request, fixtures.tx.AccountSetTrackingOn));
} else if (request.transaction ===
'C8C5E20DFB1BF533D0D81A2ED23F0A3CBD1EF2EE8A902A1D760500473CC9C582') {
conn.send(createResponse(request, fixtures.tx.AccountSetTrackingOff));
} else if (request.transaction ===
'278E6687C1C60C6873996210A6523564B63F2844FB1019576C157353B1813E60') {
conn.send(createResponse(request, fixtures.tx.RegularKey));
} else if (request.transaction ===
'10A6FB4A66EE80BED46AAE4815D7DC43B97E944984CCD5B93BCF3F8538CABC51') {
conn.send(createResponse(request, fixtures.tx.OfferCreate));
} else if (request.transaction ===
'809335DD3B0B333865096217AA2F55A4DF168E0198080B3A090D12D88880FF0E') {
conn.send(createResponse(request, fixtures.tx.OfferCancel));
} else if (request.transaction ===
'635A0769BD94710A1F6A76CDE65A3BC661B20B798807D1BBBDADCEA26420538D') {
conn.send(createResponse(request, fixtures.tx.TrustSet));
} else if (request.transaction ===
'4FB3ADF22F3C605E23FAEFAA185F3BD763C4692CAC490D9819D117CD33BFAA11') {
conn.send(createResponse(request, fixtures.tx.NoLedgerIndex));
} else if (request.transaction ===
'4FB3ADF22F3C605E23FAEFAA185F3BD763C4692CAC490D9819D117CD33BFAA12') {
conn.send(createResponse(request, fixtures.tx.NoLedgerFound));
} else if (request.transaction ===
'0F7ED9F40742D8A513AE86029462B7A6768325583DF8EE21B7EC663019DD6A04') {
conn.send(createResponse(request, fixtures.tx.LedgerWithoutTime));
} else if (request.transaction ===
'FE72FAD0FA7CA904FB6C633A1666EDF0B9C73B2F5A4555D37EEF2739A78A531B') {
conn.send(createResponse(request, fixtures.tx.TrustSetFrozenOff));
} else if (request.transaction ===
'BAF1C678323C37CCB7735550C379287667D8288C30F83148AD3C1CB019FC9002') {
conn.send(createResponse(request, fixtures.tx.TrustSetNoQuality));
} else if (request.transaction ===
'4FB3ADF22F3C605E23FAEFAA185F3BD763C4692CAC490D9819D117CD33BFAA10') {
conn.send(createResponse(request, fixtures.tx.NotValidated));
} else if (request.transaction === hashes.NOTFOUND_TRANSACTION_HASH) {
conn.send(createResponse(request, fixtures.tx.NotFound));
} else {
assert(false, 'Unrecognized transaction hash: ' + request.transaction);
}
});
mock.on('request_submit', function(request, conn) {
assert.strictEqual(request.command, 'submit');
conn.send(createResponse(request, fixtures.submit));
});
mock.on('request_account_lines', function(request, conn) {
if (request.account === addresses.ACCOUNT) {
conn.send(accountLinesResponse.normal(request));
} else if (request.account === addresses.OTHER_ACCOUNT) {
conn.send(accountLinesResponse.counterparty(request));
} else if (request.account === addresses.NOTFOUND) {
conn.send(createResponse(request, fixtures.account_info.notfound));
} else {
assert(false, 'Unrecognized account address: ' + request.account);
}
});
mock.on('request_account_tx', function(request, conn) {
if (request.account === addresses.ACCOUNT) {
conn.send(transactionsResponse(request));
} else {
assert(false, 'Unrecognized account address: ' + request.account);
}
});
mock.on('request_account_offers', function(request, conn) {
if (request.account === addresses.ACCOUNT) {
conn.send(fixtures.account_offers(request));
} else {
assert(false, 'Unrecognized account address: ' + request.account);
}
});
mock.on('request_book_offers', function(request, conn) {
if (isBTC(request.taker_gets.currency)
&& isUSD(request.taker_pays.currency)) {
conn.send(fixtures.book_offers.requestBookOffersBidsResponse(request));
} else if (isUSD(request.taker_gets.currency)
&& isBTC(request.taker_pays.currency)) {
conn.send(fixtures.book_offers.requestBookOffersAsksResponse(request));
} else {
assert(false, 'Unrecognized order book: ' + JSON.stringify(request));
}
});
mock.on('request_path_find', function(request, conn) {
let response = null;
if (request.subcommand === 'close') {
return;
}
if (request.source_account === addresses.NOTFOUND) {
response = createResponse(request, fixtures.path_find.srcActNotFound);
} else if (request.source_account === addresses.OTHER_ACCOUNT) {
response = createResponse(request, fixtures.path_find.sendUSD);
} else if (request.source_account === addresses.THIRD_ACCOUNT) {
response = createResponse(request, fixtures.path_find.XrpToXrp, {
destination_amount: request.destination_amount,
destination_address: request.destination_address
});
} else {
response = fixtures.path_find.generate.generateIOUPaymentPaths(
request.id, request.source_account, request.destination_account,
request.destination_amount);
}
conn.send(response);
});
return mock;
};