/* * Copyright 2017 balena.io * * 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 * as os from 'os'; import { stub } from 'sinon'; import * as permissions from '../../lib/shared/permissions'; describe('Shared: permissions', function () { describe('.createLaunchScript()', function () { describe('given windows', function () { beforeEach(function () { this.osPlatformStub = stub(os, 'platform'); this.osPlatformStub.returns('win32'); }); afterEach(function () { this.osPlatformStub.restore(); }); it('should escape environment variables and arguments', function () { expect( permissions.createLaunchScript( 'C:\\Users\\Alice & Bob\'s Laptop\\"what"\\balenaEtcher', ['"a Laser"', 'arg1', "'&/ ^ \\", '" $ % *'], { key: 'value', key2: ' " \' ^ & = + $ % / \\', key3: '8', }, ), ).to.equal( `chcp 65001${os.EOL}` + `set "key=value"${os.EOL}` + `set "key2= " ' ^ & = + $ % / \\"${os.EOL}` + `set "key3=8"${os.EOL}` + `"C:\\Users\\Alice & Bob's Laptop\\\\"what\\"\\balenaEtcher" "\\"a Laser\\"" "arg1" "'&/ ^ \\" "\\" $ % *"`, ); }); }); for (const platform of ['linux', 'darwin']) { describe(`given ${platform}`, function () { beforeEach(function () { this.osPlatformStub = stub(os, 'platform'); this.osPlatformStub.returns(platform); }); afterEach(function () { this.osPlatformStub.restore(); }); it('should escape environment variables and arguments', function () { expect( permissions.createLaunchScript( '/home/Alice & Bob\'s Laptop/"what"/balenaEtcher', ['arg1', "'&/ ^ \\", '" $ % *'], { key: 'value', key2: ' " \' ^ & = + $ % / \\', key3: '8', }, ), ).to.equal( `export key='value'${os.EOL}` + `export key2=' " '\\'' ^ & = + $ % / \\'${os.EOL}` + `export key3='8'${os.EOL}` + `'/home/Alice & Bob'\\''s Laptop/"what"/balenaEtcher' 'arg1' ''\\''&/ ^ \\' '" $ % *'`, ); }); }); } }); });