/src/nni_manager/training_service/remote_machine/test/windowsCommands.test.ts
https://github.com/microsoft/nni · TypeScript · 102 lines · 82 code · 18 blank · 2 comment · 1 complexity · 90fd910e072aabf14eb96ac65649348b MD5 · raw file
- // Copyright (c) Microsoft Corporation.
- // Licensed under the MIT license.
- 'use strict';
- import * as chai from 'chai';
- import * as chaiAsPromised from 'chai-as-promised';
- import * as component from '../../../common/component';
- import { cleanupUnitTest, prepareUnitTest } from '../../../common/utils';
- import { WindowsCommands } from '../extends/windowsCommands';
- describe('Unit Test for Windows Commands', () => {
- let windowsCommands: WindowsCommands
- before(() => {
- chai.should();
- chai.use(chaiAsPromised);
- prepareUnitTest();
- });
- after(() => {
- cleanupUnitTest();
- });
- beforeEach(() => {
- windowsCommands = component.get(WindowsCommands);
- });
- afterEach(() => {
- });
- it('joinPath', async () => {
- chai.expect(windowsCommands.joinPath("/root/", "\\first")).to.equal("\\root\\first");
- chai.expect(windowsCommands.joinPath("root/", "first")).to.equal("root\\first");
- chai.expect(windowsCommands.joinPath("\\root/", "\\first")).to.equal("\\root\\first");
- chai.expect(windowsCommands.joinPath("\\root\\", "\\first")).to.equal("\\root\\first");
- chai.expect(windowsCommands.joinPath("\\root", "first")).to.equal("\\root\\first");
- chai.expect(windowsCommands.joinPath("\\root\\", "first")).to.equal("\\root\\first");
- chai.expect(windowsCommands.joinPath("root\\", "first")).to.equal("root\\first");
- chai.expect(windowsCommands.joinPath("root\\")).to.equal("root\\");
- chai.expect(windowsCommands.joinPath("root")).to.equal("root");
- chai.expect(windowsCommands.joinPath(".\\root")).to.equal(".\\root");
- chai.expect(windowsCommands.joinPath("")).to.equal(".");
- chai.expect(windowsCommands.joinPath("..")).to.equal("..");
- })
- it('createFolder', async () => {
- chai.expect(windowsCommands.createFolder("test")).to.equal("mkdir \"test\"");
- chai.expect(windowsCommands.createFolder("test", true)).to.equal("mkdir \"test\"\r\nICACLS \"test\" /grant \"Users\":F");
- })
- it('allowPermission', async () => {
- chai.expect(windowsCommands.allowPermission(true, "test", "test1")).to.equal("ICACLS \"test\" /grant \"Users\":F /T\r\nICACLS \"test1\" /grant \"Users\":F /T\r\n");
- chai.expect(windowsCommands.allowPermission(false, "test")).to.equal("ICACLS \"test\" /grant \"Users\":F\r\n");
- })
- it('removeFolder', async () => {
- chai.expect(windowsCommands.removeFolder("test")).to.equal("rmdir /q \"test\"");
- chai.expect(windowsCommands.removeFolder("test", true)).to.equal("rmdir /s /q \"test\"");
- chai.expect(windowsCommands.removeFolder("test", true, false)).to.equal("rmdir /s \"test\"");
- chai.expect(windowsCommands.removeFolder("test", false, false)).to.equal("rmdir \"test\"");
- chai.expect(windowsCommands.removeFolder("test", true, true)).to.equal("rmdir /s /q \"test\"");
- })
- it('removeFiles', async () => {
- chai.expect(windowsCommands.removeFiles("test", "*.sh")).to.equal("del \"test\\*.sh\"");
- chai.expect(windowsCommands.removeFiles("test", "")).to.equal("del \"test\"");
- })
- it('readLastLines', async () => {
- chai.expect(windowsCommands.readLastLines("test", 3)).to.equal("powershell.exe Get-Content \"test\" -Tail 3");
- })
- it('isProcessAlive', async () => {
- chai.expect(windowsCommands.isProcessAliveCommand("test")).to.equal("powershell.exe Get-Process -Id (get-content \"test\") -ErrorAction SilentlyContinue");
- chai.expect(windowsCommands.isProcessAliveProcessOutput(
- {
- exitCode: 0,
- stdout: "",
- stderr: ""
- }
- )).to.equal(true);
- chai.expect(windowsCommands.isProcessAliveProcessOutput(
- {
- exitCode: 10,
- stdout: "",
- stderr: ""
- }
- )).to.equal(false);
- })
- it('extractFile', async () => {
- chai.expect(windowsCommands.extractFile("test.tar", "testfolder")).to.equal("tar -xf \"test.tar\" -C \"testfolder\"");
- })
- it('executeScript', async () => {
- chai.expect(windowsCommands.executeScript("test.sh", true)).to.equal("test.sh");
- chai.expect(windowsCommands.executeScript("test script'\"", false)).to.equal("test script'\"");
- })
- });