balena-cli/tests/commands/env/add.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 { BalenaAPIMock } from '../../nock/balena-api-mock';
import { runCommand } from '../../helpers';

describe('balena env set', function () {
	let api: BalenaAPIMock;

	const fullUUID = 'f63fd7d7812c34c4c14ae023fdff05f5';

	beforeEach(() => {
		api = new BalenaAPIMock();
		api.expectGetWhoAmI({ optional: true, persist: true });
	});

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

	it('should successfully add an environment variable', async () => {
		api.expectGetDevice({ fullUUID });
		api.expectGetConfigVars();
		api.scope
			.post(/^\/v\d+\/device_environment_variable($|\?)/)
			.reply(200, 'OK');

		const { out, err } = await runCommand(`env set TEST 1 -d ${fullUUID}`);

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

	it('should reject adding an environment variable w/o specifying a value when the process environment does not have an env var with the same name defined', async () => {
		// make sure that that there is no such env var defined atm
		delete process.env.TEST_ENV_VAR_ADD_NO_VALUE_REJECTED;

		const { out, err } = await runCommand(
			`env set TEST_ENV_VAR_ADD_NO_VALUE_REJECTED -d ${fullUUID}`,
		);

		expect(out.join('')).to.equal('');
		// Depending the context that the test was running , this was emiting either 1 or 2 '\n'
		expect(err.join('')).to.be.oneOf([
			'Value not found for environment variable: TEST_ENV_VAR_ADD_NO_VALUE_REJECTED\n',
			'Value not found for environment variable: TEST_ENV_VAR_ADD_NO_VALUE_REJECTED\n\n',
		]);
	});

	it('should successfully add an environment variable w/o specifying a value when the process environment has an env var with the same name defined', async () => {
		api.expectGetDevice({ fullUUID });
		api.expectGetConfigVars();
		api.scope
			.post(/^\/v\d+\/device_environment_variable($|\?)/)
			.reply(200, 'OK');

		process.env.TEST_ENV_VAR_ADD_NO_VALUE = '4';

		const { out, err } = await runCommand(
			`env set TEST_ENV_VAR_ADD_NO_VALUE -d ${fullUUID}`,
		);

		delete process.env.TEST_ENV_VAR_ADD_NO_VALUE;

		expect(out.join('')).to.equal('');
		expect(err.join('')).to.be.oneOf([
			' ›   Warning: Using TEST_ENV_VAR_ADD_NO_VALUE=4 from CLI process environment\n',
			// emitted when running on the windows instance
			' »   Warning: Using TEST_ENV_VAR_ADD_NO_VALUE=4 from CLI process environment\n',
		]);
	});

	it('should successfully add an environment variable w/ empty string as a value', async () => {
		api.expectGetDevice({ fullUUID });
		api.expectGetConfigVars();
		api.scope
			.post(/^\/v\d+\/device_environment_variable($|\?)/)
			.reply(200, 'OK');

		const { out, err } = await runCommand(
			`env set TEST_EMPTY_STRING '' -d ${fullUUID}`,
		);

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