azuredatastudio/extensions/sql-database-projects/src/test/testContext.ts

/*---------------------------------------------------------------------------------------------
 *  Copyright (c) Microsoft Corporation. All rights reserved.
 *  Licensed under the Source EULA. See License.txt in the project root for license information.
 *--------------------------------------------------------------------------------------------*/

import * as vscode from 'vscode';
import * as azdata from 'azdata';
import * as path from 'path';
import * as TypeMoq from 'typemoq';
import * as mssql from '../../../mssql/src/mssql';

export interface TestContext {
	context: vscode.ExtensionContext;
	dacFxService: TypeMoq.IMock<mssql.IDacFxService>;
}

export const mockDacFxResult = {
	operationId: '',
	success: true,
	errorMessage: '',
	report: ''
};

export const mockDacFxOptionsResult: mssql.DacFxOptionsResult = {
	success: true,
	errorMessage: '',
	deploymentOptions: {
		ignoreTableOptions: false,
		ignoreSemicolonBetweenStatements: false,
		ignoreRouteLifetime: false,
		ignoreRoleMembership: false,
		ignoreQuotedIdentifiers: false,
		ignorePermissions: false,
		ignorePartitionSchemes: false,
		ignoreObjectPlacementOnPartitionScheme: false,
		ignoreNotForReplication: false,
		ignoreLoginSids: false,
		ignoreLockHintsOnIndexes: false,
		ignoreKeywordCasing: false,
		ignoreIndexPadding: false,
		ignoreIndexOptions: false,
		ignoreIncrement: false,
		ignoreIdentitySeed: false,
		ignoreUserSettingsObjects: false,
		ignoreFullTextCatalogFilePath: false,
		ignoreWhitespace: false,
		ignoreWithNocheckOnForeignKeys: false,
		verifyCollationCompatibility: false,
		unmodifiableObjectWarnings: false,
		treatVerificationErrorsAsWarnings: false,
		scriptRefreshModule: false,
		scriptNewConstraintValidation: false,
		scriptFileSize: false,
		scriptDeployStateChecks: false,
		scriptDatabaseOptions: false,
		scriptDatabaseCompatibility: false,
		scriptDatabaseCollation: false,
		runDeploymentPlanExecutors: false,
		registerDataTierApplication: false,
		populateFilesOnFileGroups: false,
		noAlterStatementsToChangeClrTypes: false,
		includeTransactionalScripts: false,
		includeCompositeObjects: false,
		allowUnsafeRowLevelSecurityDataMovement: false,
		ignoreWithNocheckOnCheckConstraints: false,
		ignoreFillFactor: false,
		ignoreFileSize: false,
		ignoreFilegroupPlacement: false,
		doNotAlterReplicatedObjects: false,
		doNotAlterChangeDataCaptureObjects: false,
		disableAndReenableDdlTriggers: false,
		deployDatabaseInSingleUserMode: false,
		createNewDatabase: false,
		compareUsingTargetCollation: false,
		commentOutSetVarDeclarations: false,
		blockWhenDriftDetected: false,
		blockOnPossibleDataLoss: false,
		backupDatabaseBeforeChanges: false,
		allowIncompatiblePlatform: false,
		allowDropBlockingAssemblies: false,
		dropConstraintsNotInSource: false,
		dropDmlTriggersNotInSource: false,
		dropExtendedPropertiesNotInSource: false,
		dropIndexesNotInSource: false,
		ignoreFileAndLogFilePath: false,
		ignoreExtendedProperties: false,
		ignoreDmlTriggerState: false,
		ignoreDmlTriggerOrder: false,
		ignoreDefaultSchema: false,
		ignoreDdlTriggerState: false,
		ignoreDdlTriggerOrder: false,
		ignoreCryptographicProviderFilePath: false,
		verifyDeployment: false,
		ignoreComments: false,
		ignoreColumnCollation: false,
		ignoreAuthorizer: false,
		ignoreAnsiNulls: false,
		generateSmartDefaults: false,
		dropStatisticsNotInSource: false,
		dropRoleMembersNotInSource: false,
		dropPermissionsNotInSource: false,
		dropObjectsNotInSource: false,
		ignoreColumnOrder: false,
		doNotDropObjectTypes: [],
		excludeObjectTypes: []
	}
};

export class MockDacFxService implements mssql.IDacFxService {
	public exportBacpac(_: string, __: string, ___: string, ____: azdata.TaskExecutionMode): Thenable<mssql.DacFxResult> { return Promise.resolve(mockDacFxResult); }
	public importBacpac(_: string, __: string, ___: string, ____: azdata.TaskExecutionMode): Thenable<mssql.DacFxResult> { return Promise.resolve(mockDacFxResult); }
	public extractDacpac(_: string, __: string, ___: string, ____: string, _____: string, ______: azdata.TaskExecutionMode): Thenable<mssql.DacFxResult> { return Promise.resolve(mockDacFxResult); }
	public createProjectFromDatabase(_: string, __: string, ___: string, ____: string, _____: string, ______: mssql.ExtractTarget, _______: azdata.TaskExecutionMode): Thenable<mssql.DacFxResult> { return Promise.resolve(mockDacFxResult); }
	public deployDacpac(_: string, __: string, ___: boolean, ____: string, _____: azdata.TaskExecutionMode, ______?: Record<string, string>): Thenable<mssql.DacFxResult> { return Promise.resolve(mockDacFxResult); }
	public generateDeployScript(_: string, __: string, ___: string, ____: azdata.TaskExecutionMode, ______?: Record<string, string>): Thenable<mssql.DacFxResult> { return Promise.resolve(mockDacFxResult); }
	public generateDeployPlan(_: string, __: string, ___: string, ____: azdata.TaskExecutionMode): Thenable<mssql.GenerateDeployPlanResult> { return Promise.resolve(mockDacFxResult); }
	public getOptionsFromProfile(_: string): Thenable<mssql.DacFxOptionsResult> { return Promise.resolve(mockDacFxOptionsResult); }
	public validateStreamingJob(_: string, __: string): Thenable<mssql.ValidateStreamingJobResult> { return Promise.resolve(mockDacFxResult); }
}

export function createContext(): TestContext {
	let extensionPath = path.join(__dirname, '..', '..');

	return {
		context: {
			subscriptions: [],
			workspaceState: {
				get: () => { return undefined; },
				update: () => { return Promise.resolve(); }
			},
			globalState: {
				get: () => { return Promise.resolve(); },
				update: () => { return Promise.resolve(); }
			},
			extensionPath: extensionPath,
			asAbsolutePath: () => { return ''; },
			storagePath: '',
			globalStoragePath: '',
			logPath: '',
			extensionUri: vscode.Uri.parse(''),
			environmentVariableCollection: undefined as any,
			extensionMode: undefined as any,
			globalStorageUri: vscode.Uri.parse('test://'),
			logUri: vscode.Uri.parse('test://'),
			storageUri: vscode.Uri.parse('test://')
		},
		dacFxService: TypeMoq.Mock.ofType(MockDacFxService)
	};
}

// Mock test data
export const mockConnectionProfile: azdata.IConnectionProfile = {
	connectionName: 'My Connection',
	serverName: 'My Server',
	databaseName: 'My Database',
	userName: 'My User',
	password: 'My Pwd',
	authenticationType: 'SqlLogin',
	savePassword: false,
	groupFullName: 'My groupName',
	groupId: 'My GroupId',
	providerName: 'My Server',
	saveProfile: true,
	id: 'My Id',
	options: {
		server: 'My Server',
		database: 'My Database',
		user: 'My User',
		password: 'My Pwd',
		authenticationType: 'SqlLogin'
	}
};