MLS - Changed the dashboard to match the design (#9905)

* Machine Learning Extension - Changed the dashboard to match the design

extensions/machine-learning-services/src/test/packageManagement/packageManagementService.test.ts

@@ -31,20 +31,6 @@ describe('Package Management Service', () => {
 		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 PackageManagementService(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 PackageManagementService(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));

extensions/machine-learning-services/src/test/views/dashboardWidget.test.ts

@@ -0,0 +1,39 @@
+/*---------------------------------------------------------------------------------------------
+ *  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 { createViewContext } from './utils';
+import { DashboardWidget } from '../../views/widgets/dashboardWidget';
+
+interface TestContext {
+	apiWrapper: TypeMoq.IMock<ApiWrapper>;
+	view: azdata.ModelView;
+	onClick: vscode.EventEmitter<any>;
+}
+
+
+function createContext(): TestContext {
+
+	let viewTestContext = createViewContext();
+
+	return {
+		apiWrapper: viewTestContext.apiWrapper,
+		view: viewTestContext.view,
+		onClick: viewTestContext.onClick
+	};
+}
+
+describe('Dashboard widget', () => {
+	it('Should create view components successfully ', async function (): Promise<void> {
+		let testContext = createContext();
+		const dashboard = new DashboardWidget(testContext.apiWrapper.object, '');
+		dashboard.register();
+		testContext.onClick.fire();
+		testContext.apiWrapper.verify(x => x.executeCommand(TypeMoq.It.isAny()), TypeMoq.Times.atMostOnce());
+	});
+});

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

@@ -52,6 +52,9 @@ export function createViewContext(): ViewTestContext {
 	});
 	let flex: azdata.FlexContainer = Object.assign({}, componentBase, container, {
 	});
+	let div: azdata.DivContainer = Object.assign({}, componentBase, container, {
+		onDidClick: onClick.event
+	});
 
 	let buttonBuilder: azdata.ComponentBuilder<azdata.ButtonComponent> = {
 		component: () => button,
@@ -134,6 +137,13 @@ export function createViewContext(): ViewTestContext {
 		withItems: () => flexBuilder,
 		withLayout: () => flexBuilder
 	});
+	let divBuilder: azdata.DivBuilder = Object.assign({}, {
+		component: () => div,
+		withProperties: () => divBuilder,
+		withValidation: () => divBuilder,
+		withItems: () => divBuilder,
+		withLayout: () => divBuilder
+	});
 
 	let inputBoxBuilder: azdata.ComponentBuilder<azdata.InputBoxComponent> = {
 		component: () => {
@@ -180,7 +190,7 @@ export function createViewContext(): ViewTestContext {
 		modelBuilder: {
 			radioCardGroup: undefined!,
 			navContainer: undefined!,
-			divContainer: undefined!,
+			divContainer: () => divBuilder,
 			flexContainer: () => flexBuilder,
 			splitViewContainer: undefined!,
 			dom: undefined!,
@@ -295,6 +305,13 @@ export function createViewContext(): ViewTestContext {
 	apiWrapper.setup(x => x.createWizardPage(TypeMoq.It.isAny())).returns(() => wizardPage);
 	apiWrapper.setup(x => x.createModelViewDialog(TypeMoq.It.isAny())).returns(() => dialog);
 	apiWrapper.setup(x => x.openDialog(TypeMoq.It.isAny())).returns(() => { });
+	apiWrapper.setup(x => x.registerWidget(TypeMoq.It.isAny(), TypeMoq.It.isAny())).returns(async (id, handler) => {
+		if (id) {
+			return await handler(view);
+		} else {
+			Promise.reject();
+		}
+	});
 
 	return {
 		apiWrapper: apiWrapper,