mirror of
https://github.com/kremalicious/metamask-extension.git
synced 2024-12-05 15:35:46 +01:00
7db5d9e527
The fixture server now supports state substitutions. This allows us to embed dynamic values in our fixtures. The `custom-token` fixture has been updated to include such a fixture. The date that the seed phrase reminder was last shown has been updated to always be the current date, to prevent the reminder from showing up during e2e tests. This fixes the e2e test failure for the test "add-hide-token.spec.js" that we've been seeing on CI lately.
119 lines
3.2 KiB
JavaScript
119 lines
3.2 KiB
JavaScript
const { promises: fs } = require('fs');
|
|
const path = require('path');
|
|
const Koa = require('koa');
|
|
const { isObject, mapValues } = require('lodash');
|
|
|
|
const CURRENT_STATE_KEY = '__CURRENT__';
|
|
const DEFAULT_STATE_KEY = '__DEFAULT__';
|
|
|
|
const FIXTURE_SERVER_HOST = 'localhost';
|
|
const FIXTURE_SERVER_PORT = 12345;
|
|
|
|
const fixtureSubstitutionPrefix = '__FIXTURE_SUBSTITUTION__';
|
|
const fixtureSubstitutionCommands = {
|
|
currentDateInMilliseconds: 'currentDateInMilliseconds',
|
|
};
|
|
|
|
/**
|
|
* Perform substitutions on a single piece of state.
|
|
*
|
|
* @param {unknown} partialState - The piece of state to perform substitutions on.
|
|
* @returns {unknown} The partial state with substititions performed.
|
|
*/
|
|
function performSubstitution(partialState) {
|
|
if (Array.isArray(partialState)) {
|
|
return partialState.map(performSubstitution);
|
|
} else if (isObject(partialState)) {
|
|
return mapValues(partialState, performSubstitution);
|
|
} else if (
|
|
typeof partialState === 'string' &&
|
|
partialState.startsWith(fixtureSubstitutionPrefix)
|
|
) {
|
|
const substitutionCommand = partialState.substring(
|
|
fixtureSubstitutionPrefix.length,
|
|
);
|
|
if (
|
|
substitutionCommand ===
|
|
fixtureSubstitutionCommands.currentDateInMilliseconds
|
|
) {
|
|
return new Date().getTime();
|
|
}
|
|
throw new Error(`Unknown substitution command: ${substitutionCommand}`);
|
|
}
|
|
return partialState;
|
|
}
|
|
|
|
/**
|
|
* Substitute values in the state fixture.
|
|
*
|
|
* @param {object} rawState - The state fixture.
|
|
* @returns {object} The state fixture with substitutions performed.
|
|
*/
|
|
function performStateSubstitutions(rawState) {
|
|
return mapValues(rawState, performSubstitution);
|
|
}
|
|
|
|
class FixtureServer {
|
|
constructor() {
|
|
this._app = new Koa();
|
|
this._stateMap = new Map([[DEFAULT_STATE_KEY, Object.create(null)]]);
|
|
this._initialStateCache = new Map();
|
|
|
|
this._app.use(async (ctx) => {
|
|
// Firefox is _super_ strict about needing CORS headers
|
|
ctx.set('Access-Control-Allow-Origin', '*');
|
|
if (this._isStateRequest(ctx)) {
|
|
ctx.body = this._stateMap.get(CURRENT_STATE_KEY);
|
|
}
|
|
});
|
|
}
|
|
|
|
async start() {
|
|
const options = {
|
|
host: FIXTURE_SERVER_HOST,
|
|
port: FIXTURE_SERVER_PORT,
|
|
exclusive: true,
|
|
};
|
|
|
|
return new Promise((resolve, reject) => {
|
|
this._server = this._app.listen(options);
|
|
this._server.once('error', reject);
|
|
this._server.once('listening', resolve);
|
|
});
|
|
}
|
|
|
|
async stop() {
|
|
if (!this._server) {
|
|
return;
|
|
}
|
|
|
|
await new Promise((resolve, reject) => {
|
|
this._server.close();
|
|
this._server.once('error', reject);
|
|
this._server.once('close', resolve);
|
|
});
|
|
}
|
|
|
|
async loadState(directory) {
|
|
const statePath = path.resolve(__dirname, directory, 'state.json');
|
|
|
|
let state;
|
|
if (this._initialStateCache.has(statePath)) {
|
|
state = this._initialStateCache.get(statePath);
|
|
} else {
|
|
const data = await fs.readFile(statePath);
|
|
const rawState = JSON.parse(data.toString('utf-8'));
|
|
state = performStateSubstitutions(rawState);
|
|
this._initialStateCache.set(statePath, state);
|
|
}
|
|
|
|
this._stateMap.set(CURRENT_STATE_KEY, state);
|
|
}
|
|
|
|
_isStateRequest(ctx) {
|
|
return ctx.method === 'GET' && ctx.path === '/state.json';
|
|
}
|
|
}
|
|
|
|
module.exports = FixtureServer;
|