Machine Learning Services Dashboard (#8796)

* Added a dashboard for Machine Learning Services Extension

extensions/machine-learning-services/src/test/common/processService.test.ts

@@ -0,0 +1,69 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the Source EULA. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+'use strict';
+
+import * as vscode from 'vscode';
+import { ProcessService } from '../../common/processService';
+import * as utils from '../../common/utils';
+import should = require('should');
+
+interface TestContext {
+
+	outputChannel: vscode.OutputChannel;
+}
+
+function createContext(): TestContext {
+	return {
+		outputChannel: {
+			name: '',
+			append: () => { },
+			appendLine: () => { },
+			clear: () => { },
+			show: () => { },
+			hide: () => { },
+			dispose: () => { }
+		}
+	};
+}
+
+function execFolderListCommand(context: TestContext, service : ProcessService): Promise<void> {
+	if (utils.isWindows()) {
+		return service.execScripts('cmd', ['dir', '.'], context.outputChannel);
+	} else {
+		return service.execScripts('/bin/sh', ['-c', 'ls'], context.outputChannel);
+	}
+}
+
+function execFolderListBufferedCommand(context: TestContext, service : ProcessService): Promise<string> {
+	if (utils.isWindows()) {
+		return service.executeBufferedCommand('dir', context.outputChannel);
+	} else {
+		return service.executeBufferedCommand('ls', context.outputChannel);
+	}
+}
+
+describe('Process Service', () => {
+	it('Executing a valid script should return successfully', async function (): Promise<void> {
+		const context = createContext();
+		let service = new ProcessService();
+		await should(execFolderListCommand(context, service)).resolved();
+	});
+
+	it('execFolderListCommand should reject if command time out @UNSTABLE@', async function (): Promise<void> {
+		const context = createContext();
+		let service = new ProcessService();
+		service.timeout = 10;
+		await should(execFolderListCommand(context, service)).rejected();
+	});
+
+	it('executeBufferedCommand should resolve give valid script', async function (): Promise<void> {
+		const context = createContext();
+		let service = new ProcessService();
+		service.timeout = 2000;
+		await should(execFolderListBufferedCommand(context, service)).resolved();
+	});
+
+});

extensions/machine-learning-services/src/test/mainController.test.ts

@@ -72,8 +72,7 @@ function createContext(): TestContext {
 }
 
 function createController(testContext: TestContext): MainController {
-	let controller = new MainController(testContext.context, testContext.apiWrapper.object, testContext.queryRunner.object, testContext.processService.object);
-	controller.packageManager = testContext.packageManager.object;
+	let controller = new MainController(testContext.context, testContext.apiWrapper.object, testContext.queryRunner.object, testContext.processService.object, testContext.packageManager.object);
 	return controller;
 }
 

extensions/machine-learning-services/src/test/serverConfig/serverConfigManager.test.ts

@@ -0,0 +1,99 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the Source EULA. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+'use strict';
+
+import * as azdata from 'azdata';
+import { QueryRunner } from '../../common/queryRunner';
+import { ApiWrapper } from '../../common/apiWrapper';
+import * as TypeMoq from 'typemoq';
+import * as should from 'should';
+import { ServerConfigManager } from '../../serverConfig/serverConfigManager';
+
+interface TestContext {
+
+	apiWrapper: TypeMoq.IMock<ApiWrapper>;
+	queryRunner: TypeMoq.IMock<QueryRunner>;
+}
+
+function createContext(): TestContext {
+	return {
+		apiWrapper: TypeMoq.Mock.ofType(ApiWrapper),
+		queryRunner: TypeMoq.Mock.ofType(QueryRunner)
+	};
+}
+
+describe('Server Config Manager', () => {
+	it('openDocuments should open document in browser successfully', async function (): Promise<void> {
+		const context = createContext();
+		context.apiWrapper.setup(x => x.openExternal(TypeMoq.It.isAny())).returns(() => Promise.resolve(true));
+		let serverConfigManager = new ServerConfigManager(context.apiWrapper.object, context.queryRunner.object);
+		should.equal(await serverConfigManager.openDocuments(), true);
+	});
+
+	it('openOdbcDriverDocuments should open document in browser successfully', async function (): Promise<void> {
+		const context = createContext();
+		context.apiWrapper.setup(x => x.openExternal(TypeMoq.It.isAny())).returns(() => Promise.resolve(true));
+		let serverConfigManager = new ServerConfigManager(context.apiWrapper.object, context.queryRunner.object);
+		should.equal(await serverConfigManager.openOdbcDriverDocuments(), true);
+	});
+
+	it('openInstallDocuments should open document in browser successfully', async function (): Promise<void> {
+		const context = createContext();
+		context.apiWrapper.setup(x => x.openExternal(TypeMoq.It.isAny())).returns(() => Promise.resolve(true));
+		let serverConfigManager = new ServerConfigManager(context.apiWrapper.object, context.queryRunner.object);
+		should.equal(await serverConfigManager.openInstallDocuments(), true);
+	});
+
+	it('isMachineLearningServiceEnabled should return true if external script is enabled', async function (): Promise<void> {
+		const context = createContext();
+		context.queryRunner.setup(x => x.isMachineLearningServiceEnabled(TypeMoq.It.isAny())).returns(() => Promise.resolve(true));
+		let serverConfigManager = new ServerConfigManager(context.apiWrapper.object, context.queryRunner.object);
+		let connection  = new azdata.connection.ConnectionProfile();
+		should.equal(await serverConfigManager.isMachineLearningServiceEnabled(connection), true);
+	});
+
+	it('isRInstalled should return true if R is installed', async function (): Promise<void> {
+		const context = createContext();
+		context.queryRunner.setup(x => x.isRInstalled(TypeMoq.It.isAny())).returns(() => Promise.resolve(true));
+		let serverConfigManager = new ServerConfigManager(context.apiWrapper.object, context.queryRunner.object);
+		let connection  = new azdata.connection.ConnectionProfile();
+		should.equal(await serverConfigManager.isRInstalled(connection), true);
+	});
+
+	it('isPythonInstalled should return true if Python is installed', async function (): Promise<void> {
+		const context = createContext();
+		context.queryRunner.setup(x => x.isPythonInstalled(TypeMoq.It.isAny())).returns(() => Promise.resolve(true));
+		let serverConfigManager = new ServerConfigManager(context.apiWrapper.object, context.queryRunner.object);
+		let connection  = new azdata.connection.ConnectionProfile();
+		should.equal(await serverConfigManager.isPythonInstalled(connection), true);
+	});
+
+	it('updateExternalScriptConfig should show info message if updated successfully', async function (): Promise<void> {
+		const context = createContext();
+		context.queryRunner.setup(x => x.updateExternalScriptConfig(TypeMoq.It.isAny(), true)).returns(() => Promise.resolve());
+		context.queryRunner.setup(x => x.isMachineLearningServiceEnabled(TypeMoq.It.isAny())).returns(() => Promise.resolve(true));
+		context.apiWrapper.setup(x => x.showInfoMessage(TypeMoq.It.isAny())).returns(() => Promise.resolve(''));
+		context.apiWrapper.setup(x => x.showErrorMessage(TypeMoq.It.isAny())).returns(() => Promise.resolve(''));
+		let serverConfigManager = new ServerConfigManager(context.apiWrapper.object, context.queryRunner.object);
+		let connection  = new azdata.connection.ConnectionProfile();
+		await serverConfigManager.updateExternalScriptConfig(connection, true);
+
+		context.apiWrapper.verify(x => x.showInfoMessage(TypeMoq.It.isAny()), TypeMoq.Times.once());
+	});
+
+	it('updateExternalScriptConfig should show error message if did not updated successfully', async function (): Promise<void> {
+		const context = createContext();
+		context.queryRunner.setup(x => x.updateExternalScriptConfig(TypeMoq.It.isAny(), true)).returns(() => Promise.resolve());
+		context.queryRunner.setup(x => x.isMachineLearningServiceEnabled(TypeMoq.It.isAny())).returns(() => Promise.resolve(false));
+		context.apiWrapper.setup(x => x.showInfoMessage(TypeMoq.It.isAny())).returns(() => Promise.resolve(''));
+		context.apiWrapper.setup(x => x.showErrorMessage(TypeMoq.It.isAny())).returns(() => Promise.resolve(''));
+		let serverConfigManager = new ServerConfigManager(context.apiWrapper.object, context.queryRunner.object);
+		let connection  = new azdata.connection.ConnectionProfile();
+		await serverConfigManager.updateExternalScriptConfig(connection, true);
+
+		context.apiWrapper.verify(x => x.showErrorMessage(TypeMoq.It.isAny()), TypeMoq.Times.once());
+	});
+});