ML - dashboard icons and links (#10153)

* ML - dashboard icons and links

extensions/machine-learning/src/test/views/externalLanguages/addEditLanguageTab.test.ts

@@ -0,0 +1,120 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the Source EULA. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import * as should from 'should';
+import 'mocha';
+import { createContext, ParentDialog } from './utils';
+import { AddEditLanguageTab } from '../../../views/externalLanguages/addEditLanguageTab';
+import { LanguageUpdateModel } from '../../../views/externalLanguages/languageViewBase';
+
+describe('Add Edit External Languages Tab', () => {
+	it('Should create AddEditLanguageTab for new language successfully ', async function (): Promise<void> {
+		let testContext = createContext();
+		let parent = new ParentDialog(testContext.apiWrapper.object);
+		let languageUpdateModel: LanguageUpdateModel = {
+			content: parent.createNewContent(),
+			language: parent.createNewLanguage(),
+			newLang: true
+		};
+		let tab = new AddEditLanguageTab(testContext.apiWrapper.object, parent, languageUpdateModel);
+		should.notEqual(tab.languageView, undefined, 'Failed to create language view for add');
+	});
+
+	it('Should create AddEditLanguageTab for edit successfully ', async function (): Promise<void> {
+		let testContext = createContext();
+		let parent = new ParentDialog(testContext.apiWrapper.object);
+		let languageUpdateModel: LanguageUpdateModel = {
+			content: {
+				extensionFileName: 'filename',
+				isLocalFile: true,
+				pathToExtension: 'path',
+			},
+			language: {
+				name: 'name',
+				contents: []
+			},
+			newLang: false
+		};
+		let tab = new AddEditLanguageTab(testContext.apiWrapper.object, parent, languageUpdateModel);
+		should.notEqual(tab.languageView, undefined, 'Failed to create language view for edit');
+		should.equal(tab.saveButton, undefined);
+	});
+
+	it('Should reset AddEditLanguageTab successfully ', async function (): Promise<void> {
+		let testContext = createContext();
+		let parent = new ParentDialog(testContext.apiWrapper.object);
+		let languageUpdateModel: LanguageUpdateModel = {
+			content: {
+				extensionFileName: 'filename',
+				isLocalFile: true,
+				pathToExtension: 'path',
+			},
+			language: {
+				name: 'name',
+				contents: []
+			},
+			newLang: false
+		};
+		let tab = new AddEditLanguageTab(testContext.apiWrapper.object, parent, languageUpdateModel);
+		if (tab.languageName) {
+			tab.languageName.value = 'some value';
+		}
+		await tab.reset();
+		should.equal(tab.languageName?.value, 'name');
+	});
+
+	it('Should load content successfully ', async function (): Promise<void> {
+		let testContext = createContext();
+		let parent = new ParentDialog(testContext.apiWrapper.object);
+		let languageUpdateModel: LanguageUpdateModel = {
+			content: {
+				extensionFileName: 'filename',
+				isLocalFile: true,
+				pathToExtension: 'path',
+				environmentVariables: 'env vars',
+				parameters: 'params'
+			},
+			language: {
+				name: 'name',
+				contents: []
+			},
+			newLang: false
+		};
+		let tab = new AddEditLanguageTab(testContext.apiWrapper.object, parent, languageUpdateModel);
+		let content = tab.languageView?.updatedContent;
+		should.notEqual(content, undefined);
+		if (content) {
+			should.equal(content.extensionFileName, languageUpdateModel.content.extensionFileName);
+			should.equal(content.pathToExtension, languageUpdateModel.content.pathToExtension);
+			should.equal(content.environmentVariables, languageUpdateModel.content.environmentVariables);
+			should.equal(content.parameters, languageUpdateModel.content.parameters);
+		}
+	});
+
+	it('Should raise save event if save button clicked ', async function (): Promise<void> {
+		let testContext = createContext();
+		let parent = new ParentDialog(testContext.apiWrapper.object);
+		let languageUpdateModel: LanguageUpdateModel = {
+			content: parent.createNewContent(),
+			language: parent.createNewLanguage(),
+			newLang: true
+		};
+		let tab = new AddEditLanguageTab(testContext.apiWrapper.object, parent, languageUpdateModel);
+		should.notEqual(tab.saveButton, undefined);
+		let updateCalled = false;
+		let promise = new Promise(resolve => {
+			parent.onUpdate(() => {
+				updateCalled = true;
+				resolve();
+			});
+		});
+
+		testContext.onClick.fire(undefined);
+		parent.onUpdatedLanguage(languageUpdateModel);
+		await promise;
+		should.equal(updateCalled, true);
+		should.notEqual(tab.updatedData, undefined);
+	});
+});

extensions/machine-learning/src/test/views/externalLanguages/languageController.test.ts

@@ -0,0 +1,104 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the Source EULA. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import * as should from 'should';
+import 'mocha';
+import * as TypeMoq from 'typemoq';
+import { createContext } from './utils';
+import { LanguageController } from '../../../views/externalLanguages/languageController';
+import * as mssql from '../../../../../mssql';
+
+describe('External Languages Controller', () => {
+	it('Should open dialog for manage languages successfully ', async function (): Promise<void> {
+		let testContext = createContext();
+		let controller = new LanguageController(testContext.apiWrapper.object, '', testContext.dialogModel.object);
+		let dialog = await controller.manageLanguages();
+		testContext.apiWrapper.verify(x => x.openDialog(TypeMoq.It.isAny()), TypeMoq.Times.once());
+		should.notEqual(dialog, undefined);
+	});
+
+	it('Should list languages successfully ', async function (): Promise<void> {
+		let testContext = createContext();
+		let languages: mssql.ExternalLanguage[]  = [{
+			name: '',
+			contents: [{
+				extensionFileName: '',
+				isLocalFile: true,
+				pathToExtension: '',
+			}]
+		}];
+
+		testContext.dialogModel.setup( x=> x.getLanguageList()).returns(() => Promise.resolve(languages));
+		let controller = new LanguageController(testContext.apiWrapper.object, '', testContext.dialogModel.object);
+		let dialog = await controller.manageLanguages();
+		let actual = await dialog.listLanguages();
+		should.deepEqual(actual, languages);
+	});
+
+	it('Should update languages successfully ', async function (): Promise<void> {
+		let testContext = createContext();
+		let language: mssql.ExternalLanguage = {
+			name: '',
+			contents: [{
+				extensionFileName: '',
+				isLocalFile: true,
+				pathToExtension: '',
+			}]
+		};
+
+		testContext.dialogModel.setup( x=> x.updateLanguage(language)).returns(() => Promise.resolve());
+		let controller = new LanguageController(testContext.apiWrapper.object, '', testContext.dialogModel.object);
+		let dialog = await controller.manageLanguages();
+		await dialog.updateLanguage({
+			language: language,
+			content: language.contents[0],
+			newLang: false
+		});
+		testContext.dialogModel.verify(x => x.updateLanguage(TypeMoq.It.isAny()), TypeMoq.Times.once());
+	});
+
+	it('Should delete language successfully ', async function (): Promise<void> {
+		let testContext = createContext();
+		let language: mssql.ExternalLanguage = {
+			name: '',
+			contents: [{
+				extensionFileName: '',
+				isLocalFile: true,
+				pathToExtension: '',
+			}]
+		};
+
+		testContext.dialogModel.setup( x=> x.deleteLanguage(language.name)).returns(() => Promise.resolve());
+		let controller = new LanguageController(testContext.apiWrapper.object, '', testContext.dialogModel.object);
+		let dialog = await controller.manageLanguages();
+		await dialog.deleteLanguage({
+			language: language,
+			content: language.contents[0],
+			newLang: false
+		});
+		testContext.dialogModel.verify(x => x.deleteLanguage(TypeMoq.It.isAny()), TypeMoq.Times.once());
+	});
+
+	it('Should open edit dialog for edit language', async function (): Promise<void> {
+		let testContext = createContext();
+		let language: mssql.ExternalLanguage = {
+			name: '',
+			contents: [{
+				extensionFileName: '',
+				isLocalFile: true,
+				pathToExtension: '',
+			}]
+		};
+		let controller = new LanguageController(testContext.apiWrapper.object, '', testContext.dialogModel.object);
+		let dialog = await controller.manageLanguages();
+		dialog.onEditLanguage({
+			language: language,
+			content: language.contents[0],
+			newLang: false
+		});
+		testContext.apiWrapper.verify(x => x.openDialog(TypeMoq.It.isAny()), TypeMoq.Times.exactly(2));
+		should.notEqual(dialog, undefined);
+	});
+});

extensions/machine-learning/src/test/views/externalLanguages/languageEditDialog.test.ts

@@ -0,0 +1,50 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the Source EULA. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import * as should from 'should';
+import 'mocha';
+import { createContext, ParentDialog } from './utils';
+import { LanguageEditDialog } from '../../../views/externalLanguages/languageEditDialog';
+import { LanguageUpdateModel } from '../../../views/externalLanguages/languageViewBase';
+
+describe('Edit External Languages Dialog', () => {
+	it('Should open dialog successfully ', async function (): Promise<void> {
+		let testContext = createContext();
+		let parent = new ParentDialog(testContext.apiWrapper.object);
+		let languageUpdateModel: LanguageUpdateModel = {
+			content: parent.createNewContent(),
+			language: parent.createNewLanguage(),
+			newLang: true
+		};
+		let dialog = new LanguageEditDialog(testContext.apiWrapper.object, parent, languageUpdateModel);
+		dialog.showDialog();
+		should.notEqual(dialog.addNewLanguageTab, undefined);
+	});
+
+	it('Should raise save event if save button clicked ', async function (): Promise<void> {
+		let testContext = createContext();
+		let parent = new ParentDialog(testContext.apiWrapper.object);
+		let languageUpdateModel: LanguageUpdateModel = {
+			content: parent.createNewContent(),
+			language: parent.createNewLanguage(),
+			newLang: true
+		};
+		let dialog = new LanguageEditDialog(testContext.apiWrapper.object, parent, languageUpdateModel);
+		dialog.showDialog();
+
+		let updateCalled = false;
+		let promise = new Promise(resolve => {
+			parent.onUpdate(() => {
+				updateCalled = true;
+				parent.onUpdatedLanguage(languageUpdateModel);
+				resolve();
+			});
+		});
+
+		dialog.onSave();
+		await promise;
+		should.equal(updateCalled, true);
+	});
+});

extensions/machine-learning/src/test/views/externalLanguages/languagesDialog.test.ts

@@ -0,0 +1,19 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the Source EULA. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import * as should from 'should';
+import 'mocha';
+import { createContext } from './utils';
+import { LanguagesDialog } from '../../../views/externalLanguages/languagesDialog';
+
+describe('External Languages Dialog', () => {
+	it('Should open dialog successfully ', async function (): Promise<void> {
+		let testContext = createContext();
+		let dialog = new LanguagesDialog(testContext.apiWrapper.object, '');
+		dialog.showDialog();
+		should.notEqual(dialog.addNewLanguageTab, undefined);
+		should.notEqual(dialog.currentLanguagesTab, undefined);
+	});
+});

extensions/machine-learning/src/test/views/externalLanguages/languagesDialogModel.test.ts

@@ -0,0 +1,61 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the Source EULA. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import * as should from 'should';
+import 'mocha';
+import { createContext } from './utils';
+import * as mssql from '../../../../../mssql';
+import { LanguageService } from '../../../externalLanguage/languageService';
+
+describe('External Languages Dialog Model', () => {
+	it('Should list languages successfully ', async function (): Promise<void> {
+		let testContext = createContext();
+		let languages: mssql.ExternalLanguage[]  = [{
+			name: '',
+			contents: [{
+				extensionFileName: '',
+				isLocalFile: true,
+				pathToExtension: '',
+			}]
+		}];
+		testContext.languageExtensionService.listLanguages = () => {return Promise.resolve(languages);};
+		let model = new LanguageService(testContext.apiWrapper.object, testContext.languageExtensionService);
+		await model.load();
+		let actual = await model.getLanguageList();
+		should.deepEqual(actual, languages);
+	});
+
+	it('Should update language successfully ', async function (): Promise<void> {
+		let testContext = createContext();
+		let language: mssql.ExternalLanguage  = {
+			name: '',
+			contents: [{
+				extensionFileName: '',
+				isLocalFile: true,
+				pathToExtension: '',
+			}]
+		};
+
+		let model = new LanguageService(testContext.apiWrapper.object, testContext.languageExtensionService);
+		await model.load();
+		await should(model.updateLanguage(language)).resolved();
+	});
+
+	it('Should delete language successfully ', async function (): Promise<void> {
+		let testContext = createContext();
+		let language: mssql.ExternalLanguage  = {
+			name: '',
+			contents: [{
+				extensionFileName: '',
+				isLocalFile: true,
+				pathToExtension: '',
+			}]
+		};
+
+		let model = new LanguageService(testContext.apiWrapper.object, testContext.languageExtensionService);
+		await model.load();
+		await should(model.deleteLanguage(language.name)).resolved();
+	});
+});

extensions/machine-learning/src/test/views/externalLanguages/utils.ts

@@ -0,0 +1,53 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the Source EULA. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import * as azdata from 'azdata';
+import * as vscode from 'vscode';
+import * as TypeMoq from 'typemoq';
+import { ApiWrapper } from '../../../common/apiWrapper';
+import { LanguageViewBase } from '../../../views/externalLanguages/languageViewBase';
+import * as mssql from '../../../../../mssql';
+import { LanguageService } from '../../../externalLanguage/languageService';
+import { createViewContext } from '../utils';
+
+export interface TestContext {
+	apiWrapper: TypeMoq.IMock<ApiWrapper>;
+	view: azdata.ModelView;
+	languageExtensionService: mssql.ILanguageExtensionService;
+	onClick: vscode.EventEmitter<any>;
+	dialogModel: TypeMoq.IMock<LanguageService>;
+}
+
+export class ParentDialog extends LanguageViewBase {
+	public reset(): Promise<void> {
+		return Promise.resolve();
+	}
+	constructor(
+		apiWrapper: ApiWrapper) {
+		super(apiWrapper, '');
+	}
+}
+
+export function createContext(): TestContext {
+
+	let viewTestContext = createViewContext();
+	let connection = new azdata.connection.ConnectionProfile();
+	viewTestContext.apiWrapper.setup(x => x.getCurrentConnection()).returns(() => { return Promise.resolve(connection); });
+	viewTestContext.apiWrapper.setup(x => x.getUriForConnection(TypeMoq.It.isAny())).returns(() => { return Promise.resolve('connectionUrl'); });
+
+	let languageExtensionService: mssql.ILanguageExtensionService = {
+		listLanguages: () => { return Promise.resolve([]); },
+		deleteLanguage: () => { return Promise.resolve(); },
+		updateLanguage: () => { return Promise.resolve(); }
+	};
+
+	return {
+		apiWrapper: viewTestContext.apiWrapper,
+		view: viewTestContext.view,
+		languageExtensionService: languageExtensionService,
+		onClick: viewTestContext.onClick,
+		dialogModel: TypeMoq.Mock.ofType(LanguageService)
+	};
+}