balena-supervisor/test/lib/mocked-balena-api.ts

import express from 'express';
import _ from 'lodash';

const api: express.Express & {
	balenaBackend?: {
		currentId: number;
		devices: { [key: string]: any };
		registerHandler: express.RequestHandler;
		getDeviceHandler: express.RequestHandler;
		deviceKeyHandler: express.RequestHandler;
	};
} = express();

// tslint:disable-next-line
api.use(express.json());

api.balenaBackend = {
	currentId: 1,
	devices: {},
	registerHandler: (req, res) => {
		console.log('/device/register called with ', req.body);
		const device = req.body;
		switch (req.body.uuid) {
			case 'not-unique':
				return res.status(409).json(device);
			default:
				device.id = api.balenaBackend!.currentId++;
				api.balenaBackend!.devices[device.id] = device;
				return res.status(201).json(device);
		}
	},
	getDeviceHandler: (req, res) => {
		const uuid = req.params[0];
		if (uuid != null) {
			return res.json({
				d: _.filter(api.balenaBackend!.devices, (dev) => dev.uuid === uuid),
			});
		} else {
			return res.json({ d: [] });
		}
	},
	deviceKeyHandler: (req, res) => {
		return res.status(200).send(req.body.apiKey);
	},
};

api.post('/device/register', (req, res) =>
	api.balenaBackend!.registerHandler(req, res, _.noop),
);

api.get(/\/v6\/device\(uuid=%27([0-9a-f]+)%27\)/, (req, res) =>
	api.balenaBackend!.getDeviceHandler(req, res, _.noop),
);

api.get(/\/v6\/device/, (req, res) => {
	const [, uuid] =
		/uuid eq '([0-9a-f]+)'/i.exec(req.query['$filter'] as string) ?? [];
	req.params[0] = uuid;
	return api.balenaBackend!.getDeviceHandler(req, res, _.noop);
});

api.post('/api-key/device/:deviceId/device-key', (req, res) =>
	api.balenaBackend!.deviceKeyHandler(req, res, _.noop),
);

export = api;