balena-cli/tests/commands/env/envs.spec.ts

/**
 * @license
 * Copyright 2019-2020 Balena Ltd.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import { expect } from 'chai';
import { stripIndent } from '../../../lib/utils/lazy';

import { isV12 } from '../../../lib/utils/version';
import { BalenaAPIMock } from '../../balena-api-mock';
import { runCommand } from '../../helpers';

describe('balena envs', function () {
	const appName = 'test';
	let fullUUID: string;
	let shortUUID: string;
	let api: BalenaAPIMock;

	beforeEach(() => {
		api = new BalenaAPIMock();
		api.expectGetWhoAmI({ optional: true, persist: true });
		api.expectGetMixpanel({ optional: true });
		// Random device UUID used to frustrate _.memoize() in utils/cloud.ts
		fullUUID = require('crypto').randomBytes(16).toString('hex');
		shortUUID = fullUUID.substring(0, 7);
	});

	afterEach(() => {
		// Check all expected api calls have been made and clean up.
		api.done();
	});

	it('should successfully list env vars for a test app', async () => {
		if (isV12()) {
			api.expectGetApplication();
			api.expectGetAppEnvVars();
			api.expectGetAppServiceVars();

			const { out, err } = await runCommand(`envs -a ${appName}`);

			expect(out.join('')).to.equal(
				stripIndent`
				ID     NAME  VALUE       APPLICATION SERVICE
				120110 svar1 svar1-value test        service1
				120111 svar2 svar2-value test        service2
				120101 var1  var1-val    test        *
				120102 var2  22          test        *
			` + '\n',
			);
			expect(err.join('')).to.equal('');
		} else {
			api.expectGetApplication();
			api.expectGetAppEnvVars();

			const { out, err } = await runCommand(`envs -a ${appName}`);

			expect(out.join('')).to.equal(
				stripIndent`
				ID     NAME VALUE
				120101 var1 var1-val
				120102 var2 22
			` + '\n',
			);
			expect(err.join('')).to.equal('');
		}
	});

	it('should successfully list config vars for a test app', async () => {
		api.expectGetApplication();
		api.expectGetAppConfigVars();

		const { out, err } = await runCommand(`envs -a ${appName} --config`);

		if (isV12()) {
			expect(out.join('')).to.equal(
				stripIndent`
				ID     NAME                           VALUE APPLICATION
				120300 RESIN_SUPERVISOR_NATIVE_LOGGER false test
			` + '\n',
			);
		} else {
			expect(out.join('')).to.equal(
				stripIndent`
				ID     NAME                           VALUE
				120300 RESIN_SUPERVISOR_NATIVE_LOGGER false
			` + '\n',
			);
		}

		expect(err.join('')).to.equal('');
	});

	it('should successfully list config vars for a test app (JSON output)', async () => {
		api.expectGetApplication();
		api.expectGetAppConfigVars();

		const { out, err } = await runCommand(`envs -cja ${appName}`);

		if (isV12()) {
			expect(JSON.parse(out.join(''))).to.deep.equal([
				{
					appName: 'test',
					id: 120300,
					name: 'RESIN_SUPERVISOR_NATIVE_LOGGER',
					value: 'false',
				},
			]);
		} else {
			expect(JSON.parse(out.join(''))).to.deep.equal([
				{
					id: 120300,
					name: 'RESIN_SUPERVISOR_NATIVE_LOGGER',
					value: 'false',
				},
			]);
		}
		expect(err.join('')).to.equal('');
	});

	it('should successfully list service variables for a test app (-s flag)', async () => {
		const serviceName = 'service2';
		api.expectGetService({ serviceName });
		api.expectGetApplication();
		if (isV12()) {
			api.expectGetAppEnvVars();
		}
		api.expectGetAppServiceVars();

		const { out, err } = await runCommand(
			`envs -a ${appName} -s ${serviceName}`,
		);

		if (isV12()) {
			expect(out.join('')).to.equal(
				stripIndent`
			ID     NAME  VALUE       APPLICATION SERVICE
			120111 svar2 svar2-value test        service2
			120101 var1  var1-val    test        *
			120102 var2  22          test        *
			` + '\n',
			);
		} else {
			expect(out.join('')).to.equal(
				stripIndent`
			ID     NAME  VALUE
			120111 svar2 svar2-value
			` + '\n',
			);
		}

		expect(err.join('')).to.equal('');
	});

	if (!isV12()) {
		it('should produce an empty JSON array when no app service variables exist', async () => {
			const serviceName = 'nono';
			api.expectGetService({ serviceName });
			api.expectGetApplication();
			api.expectGetAppServiceVars();

			const { out, err } = await runCommand(
				`envs -a ${appName} -s ${serviceName} -j`,
			);

			expect(out.join('')).to.equal('[]\n');
			expect(err.join('')).to.equal('');
		});
	}

	if (!isV12()) {
		it('should successfully list env and service vars for a test app (--all flag)', async () => {
			api.expectGetApplication();
			api.expectGetAppEnvVars();
			api.expectGetAppServiceVars();

			const { out, err } = await runCommand(`envs -a ${appName} --all`);

			expect(out.join('')).to.equal(
				stripIndent`
			ID     NAME  VALUE       APPLICATION SERVICE
			120110 svar1 svar1-value test        service1
			120111 svar2 svar2-value test        service2
			120101 var1  var1-val    test        *
			120102 var2  22          test        *
			` + '\n',
			);
			expect(err.join('')).to.equal('');
		});
	}

	it(
		isV12()
			? 'should successfully list env and service vars for a test app (-s flags)'
			: 'should successfully list env and service vars for a test app (--all -s flags)',
		async () => {
			const serviceName = 'service1';
			api.expectGetService({ serviceName });
			api.expectGetApplication();
			api.expectGetAppEnvVars();
			api.expectGetAppServiceVars();

			const { out, err } = await runCommand(
				isV12()
					? `envs -a ${appName} -s ${serviceName}`
					: `envs -a ${appName} --all -s ${serviceName}`,
			);

			expect(out.join('')).to.equal(
				stripIndent`
		ID     NAME  VALUE       APPLICATION SERVICE
		120110 svar1 svar1-value test        ${serviceName}
		120101 var1  var1-val    test        *
		120102 var2  22          test        *
		` + '\n',
			);
			expect(err.join('')).to.equal('');
		},
	);

	it('should successfully list env variables for a test device', async () => {
		api.expectGetDevice({ fullUUID });
		api.expectGetDeviceEnvVars();
		if (isV12()) {
			api.expectGetApplication();
			api.expectGetAppEnvVars();
			api.expectGetAppServiceVars();
			api.expectGetDeviceServiceVars();
		}

		const uuid = shortUUID;
		const { out, err } = await runCommand(`envs -d ${uuid}`);

		if (isV12()) {
			expect(out.join('')).to.equal(
				stripIndent`
			ID     NAME  VALUE       APPLICATION DEVICE  SERVICE
			120110 svar1 svar1-value test        *       service1
			120111 svar2 svar2-value test        *       service2
			120120 svar3 svar3-value test        ${uuid} service1
			120121 svar4 svar4-value test        ${uuid} service2
			120101 var1  var1-val    test        *       *
			120102 var2  22          test        *       *
			120203 var3  var3-val    test        ${uuid} *
			120204 var4  44          test        ${uuid} *
			` + '\n',
			);
		} else {
			expect(out.join('')).to.equal(
				stripIndent`
			ID     NAME VALUE
			120203 var3 var3-val
			120204 var4 44
			` + '\n',
			);
		}

		expect(err.join('')).to.equal('');
	});

	it('should successfully list env variables for a test device (JSON output)', async () => {
		api.expectGetDevice({ fullUUID });
		api.expectGetDeviceEnvVars();
		if (isV12()) {
			api.expectGetApplication();
			api.expectGetAppEnvVars();
			api.expectGetAppServiceVars();
			api.expectGetDeviceServiceVars();
		}

		const { out, err } = await runCommand(`envs -jd ${shortUUID}`);

		if (isV12()) {
			expect(JSON.parse(out.join(''))).to.deep.equal(
				JSON.parse(`[
				{ "id": 120101, "appName": "test", "deviceUUID": "*", "name": "var1", "value": "var1-val", "serviceName": "*" },
				{ "id": 120102, "appName": "test", "deviceUUID": "*", "name": "var2", "value": "22", "serviceName": "*" },
				{ "id": 120110, "appName": "test", "deviceUUID": "*", "name": "svar1", "value": "svar1-value", "serviceName": "service1" },
				{ "id": 120111, "appName": "test", "deviceUUID": "*", "name": "svar2", "value": "svar2-value", "serviceName": "service2" },
				{ "id": 120120, "appName": "test", "deviceUUID": "${fullUUID}", "name": "svar3", "value": "svar3-value", "serviceName": "service1" },
				{ "id": 120121, "appName": "test", "deviceUUID": "${fullUUID}", "name": "svar4", "value": "svar4-value", "serviceName": "service2" },
				{ "id": 120203, "appName": "test", "deviceUUID": "${fullUUID}", "name": "var3", "value": "var3-val", "serviceName": "*" },
				{ "id": 120204, "appName": "test", "deviceUUID": "${fullUUID}", "name": "var4", "value": "44", "serviceName": "*" }
			]`),
			);
		} else {
			expect(JSON.parse(out.join(''))).to.deep.equal([
				{
					id: 120203,
					name: 'var3',
					value: 'var3-val',
				},
				{
					id: 120204,
					name: 'var4',
					value: '44',
				},
			]);
		}

		expect(err.join('')).to.equal('');
	});

	it('should successfully list config variables for a test device', async () => {
		api.expectGetDevice({ fullUUID });
		api.expectGetDeviceConfigVars();
		if (isV12()) {
			api.expectGetApplication();
			api.expectGetAppConfigVars();
		}

		const { out, err } = await runCommand(`envs -d ${shortUUID} --config`);

		if (isV12()) {
			expect(out.join('')).to.equal(
				stripIndent`
			ID     NAME                           VALUE  APPLICATION DEVICE
			120300 RESIN_SUPERVISOR_NATIVE_LOGGER false  test        *
			120400 RESIN_SUPERVISOR_POLL_INTERVAL 900900 test        ${shortUUID}
			` + '\n',
			);
		} else {
			expect(out.join('')).to.equal(
				stripIndent`
			ID     NAME                           VALUE
			120400 RESIN_SUPERVISOR_POLL_INTERVAL 900900
			` + '\n',
			);
		}

		expect(err.join('')).to.equal('');
	});

	it('should successfully list service variables for a test device (-s flag)', async () => {
		const serviceName = 'service2';
		api.expectGetService({ serviceName });
		api.expectGetApplication();
		api.expectGetDevice({ fullUUID });
		api.expectGetDeviceServiceVars();
		if (isV12()) {
			api.expectGetAppEnvVars();
			api.expectGetAppServiceVars();
			api.expectGetDeviceEnvVars();
		}

		const uuid = shortUUID;
		const { out, err } = await runCommand(`envs -d ${uuid} -s ${serviceName}`);

		if (isV12()) {
			expect(out.join('')).to.equal(
				stripIndent`
			ID     NAME  VALUE       APPLICATION DEVICE  SERVICE
			120111 svar2 svar2-value test        *       service2
			120121 svar4 svar4-value test        ${uuid} service2
			120101 var1  var1-val    test        *       *
			120102 var2  22          test        *       *
			120203 var3  var3-val    test        ${uuid} *
			120204 var4  44          test        ${uuid} *
			` + '\n',
			);
		} else {
			expect(out.join('')).to.equal(
				stripIndent`
			ID     NAME  VALUE
			120121 svar4 svar4-value
			` + '\n',
			);
		}

		expect(err.join('')).to.equal('');
	});

	if (!isV12()) {
		it('should produce an empty JSON array when no device service variables exist', async () => {
			const serviceName = 'nono';
			api.expectGetService({ serviceName });
			api.expectGetApplication();
			api.expectGetDevice({ fullUUID });
			api.expectGetDeviceServiceVars();

			const { out, err } = await runCommand(
				`envs -d ${shortUUID} -s ${serviceName} -j`,
			);

			expect(out.join('')).to.equal('[]\n');
			expect(err.join('')).to.equal('');
		});
	}

	if (!isV12()) {
		it('should successfully list env and service variables for a test device (--all flag)', async () => {
			api.expectGetApplication();
			api.expectGetAppEnvVars();
			api.expectGetAppServiceVars();
			api.expectGetDevice({ fullUUID });
			api.expectGetDeviceEnvVars();
			api.expectGetDeviceServiceVars();

			const uuid = shortUUID;
			const { out, err } = await runCommand(`envs -d ${uuid} --all`);

			expect(out.join('')).to.equal(
				stripIndent`
			ID     NAME  VALUE       APPLICATION DEVICE  SERVICE
			120110 svar1 svar1-value test        *       service1
			120111 svar2 svar2-value test        *       service2
			120120 svar3 svar3-value test        ${uuid} service1
			120121 svar4 svar4-value test        ${uuid} service2
			120101 var1  var1-val    test        *       *
			120102 var2  22          test        *       *
			120203 var3  var3-val    test        ${uuid} *
			120204 var4  44          test        ${uuid} *
			` + '\n',
			);
			expect(err.join('')).to.equal('');
		});
	}

	it('should successfully list env and service variables for a test device (unknown app)', async () => {
		api.expectGetDevice({ fullUUID, inaccessibleApp: true });
		api.expectGetDeviceEnvVars();
		api.expectGetDeviceServiceVars();

		const uuid = shortUUID;

		const { out, err } = await runCommand(
			isV12() ? `envs -d ${uuid}` : `envs -d ${uuid} --all`,
		);

		expect(out.join('')).to.equal(
			stripIndent`
			ID     NAME  VALUE       APPLICATION DEVICE  SERVICE
			120120 svar3 svar3-value N/A         ${uuid} service1
			120121 svar4 svar4-value N/A         ${uuid} service2
			120203 var3  var3-val    N/A         ${uuid} *
			120204 var4  44          N/A         ${uuid} *
			` + '\n',
		);
		expect(err.join('')).to.equal('');
	});

	it(
		isV12()
			? 'should successfully list env and service vars for a test device (-s flags)'
			: 'should successfully list env and service vars for a test device (--all -s flags)',
		async () => {
			const serviceName = 'service1';
			api.expectGetService({ serviceName });
			api.expectGetApplication();
			api.expectGetAppEnvVars();
			api.expectGetAppServiceVars();
			api.expectGetDevice({ fullUUID });
			api.expectGetDeviceEnvVars();
			api.expectGetDeviceServiceVars();

			const uuid = shortUUID;
			const { out, err } = await runCommand(
				isV12()
					? `envs -d ${uuid} -s ${serviceName}`
					: `envs -d ${uuid} --all -s ${serviceName}`,
			);

			expect(out.join('')).to.equal(
				stripIndent`
		ID     NAME  VALUE       APPLICATION DEVICE  SERVICE
		120110 svar1 svar1-value test        *       ${serviceName}
		120120 svar3 svar3-value test        ${uuid} ${serviceName}
		120101 var1  var1-val    test        *       *
		120102 var2  22          test        *       *
		120203 var3  var3-val    test        ${uuid} *
		120204 var4  44          test        ${uuid} *
		` + '\n',
			);
			expect(err.join('')).to.equal('');
		},
	);

	it(
		isV12()
			? 'should successfully list env and service vars for a test device (-js flags)'
			: 'should successfully list env and service vars for a test device (--all -js flags)',
		async () => {
			const serviceName = 'service1';
			api.expectGetService({ serviceName });
			api.expectGetApplication();
			api.expectGetAppEnvVars();
			api.expectGetAppServiceVars();
			api.expectGetDevice({ fullUUID });
			api.expectGetDeviceEnvVars();
			api.expectGetDeviceServiceVars();

			const { out, err } = await runCommand(
				isV12()
					? `envs -d ${shortUUID} -js ${serviceName}`
					: `envs -d ${shortUUID} --all -js ${serviceName}`,
			);

			expect(JSON.parse(out.join(''))).to.deep.equal(
				JSON.parse(`[
				{ "id": 120101, "appName": "test", "deviceUUID": "*", "name": "var1", "value": "var1-val", "serviceName": "*" },
				{ "id": 120102, "appName": "test", "deviceUUID": "*", "name": "var2", "value": "22", "serviceName": "*" },
				{ "id": 120110, "appName": "test", "deviceUUID": "*", "name": "svar1", "value": "svar1-value", "serviceName": "${serviceName}" },
				{ "id": 120120, "appName": "test", "deviceUUID": "${fullUUID}", "name": "svar3", "value": "svar3-value", "serviceName": "${serviceName}" },
				{ "id": 120203, "appName": "test", "deviceUUID": "${fullUUID}", "name": "var3", "value": "var3-val", "serviceName": "*" },
				{ "id": 120204, "appName": "test", "deviceUUID": "${fullUUID}", "name": "var4", "value": "44", "serviceName": "*" }
			]`),
			);
			expect(err.join('')).to.equal('');
		},
	);
});