import * as path from 'path';

import { instance, mock } from 'ts-mockito';

import * as TypeMoq from 'typemoq';

import { ConfigurationTarget, Uri } from 'vscode';

import { IApplicationShell, ICommandManager, IWorkspaceService } from '../../client/common/application/types';

import { WorkspaceService } from '../../client/common/application/workspace';

import { ConfigurationService } from '../../client/common/configuration/service';

import { InstallationChannelManager } from '../../client/common/installer/channelManager';

import { ProductInstaller } from '../../client/common/installer/productInstaller';

import {

CTagsProductPathService,

DataScienceProductPathService,

FormatterProductPathService,

LinterProductPathService,

RefactoringLibraryProductPathService,

TestFrameworkProductPathService

} from '../../client/common/installer/productPath';

import { ProductService } from '../../client/common/installer/productService';

import { IInstallationChannelManager, IModuleInstaller, IProductPathService, IProductService } from '../../client/common/installer/types';

import { Logger } from '../../client/common/logger';

import { PersistentStateFactory } from '../../client/common/persistentState';

import { PathUtils } from '../../client/common/platform/pathUtils';

import { CurrentProcess } from '../../client/common/process/currentProcess';

import { ProcessLogger } from '../../client/common/process/logger';

import { IProcessLogger, IProcessServiceFactory } from '../../client/common/process/types';

import { TerminalHelper } from '../../client/common/terminal/helper';

import { ITerminalHelper } from '../../client/common/terminal/types';

import {

IConfigurationService,

ICurrentProcess,

IInstaller,

ILogger,

IPathUtils,

IPersistentStateFactory,

IsWindows,

ModuleNamePurpose,

Product,

ProductType

} from '../../client/common/types';

import { createDeferred } from '../../client/common/utils/async';

import { getNamesAndValues } from '../../client/common/utils/enum';

import { ICondaService } from '../../client/interpreter/contracts';

import { CondaService } from '../../client/interpreter/locators/services/condaService';

import { InterpreterHashProvider } from '../../client/interpreter/locators/services/hashProvider';

import { InterpeterHashProviderFactory } from '../../client/interpreter/locators/services/hashProviderFactory';

import { InterpreterFilter } from '../../client/interpreter/locators/services/interpreterFilter';

import { WindowsStoreInterpreter } from '../../client/interpreter/locators/services/windowsStoreInterpreter';

import { rootWorkspaceUri, updateSetting } from '../common';

import { MockModuleInstaller } from '../mocks/moduleInstaller';

import { MockProcessService } from '../mocks/proc';

import { UnitTestIocContainer } from '../testing/serviceRegistry';

import { closeActiveWindows, initializeTest, IS_MULTI_ROOT_TEST } from './../initialize';

// tslint:disable-next-line:max-func-body-length

suite('Installer', () => {

let ioc: UnitTestIocContainer;

const workspaceUri = Uri.file(path.join(__dirname, '..', '..', '..', 'src', 'test'));

const resource = IS_MULTI_ROOT_TEST ? workspaceUri : undefined;

suiteSetup(initializeTest);

setup(async () => {

await initializeTest();

await resetSettings();

initializeDI();

});

suiteTeardown(async () => {

await closeActiveWindows();

await resetSettings();

});

teardown(async () => {

await ioc.dispose();

await closeActiveWindows();

});

function initializeDI() {

ioc = new UnitTestIocContainer();

ioc.registerUnitTestTypes();

ioc.registerFileSystemTypes();

ioc.registerVariableTypes();

ioc.registerLinterTypes();

ioc.registerFormatterTypes();

ioc.serviceManager.addSingleton<IPersistentStateFactory>(IPersistentStateFactory, PersistentStateFactory);

ioc.serviceManager.addSingleton<ILogger>(ILogger, Logger);

ioc.serviceManager.addSingleton<IInstaller>(IInstaller, ProductInstaller);

ioc.serviceManager.addSingleton<IPathUtils>(IPathUtils, PathUtils);

ioc.serviceManager.addSingleton<IProcessLogger>(IProcessLogger, ProcessLogger);

ioc.serviceManager.addSingleton<ICurrentProcess>(ICurrentProcess, CurrentProcess);

ioc.serviceManager.addSingleton<IInstallationChannelManager>(IInstallationChannelManager, InstallationChannelManager);

ioc.serviceManager.addSingletonInstance<ICommandManager>(ICommandManager, TypeMoq.Mock.ofType<ICommandManager>().object);

ioc.serviceManager.addSingletonInstance<IApplicationShell>(IApplicationShell, TypeMoq.Mock.ofType<IApplicationShell>().object);

ioc.serviceManager.addSingleton<IConfigurationService>(IConfigurationService, ConfigurationService);

ioc.serviceManager.addSingleton<IWorkspaceService>(IWorkspaceService, WorkspaceService);

ioc.serviceManager.addSingleton<ICondaService>(ICondaService, CondaService);

ioc.registerMockInterpreterTypes();

ioc.registerMockProcessTypes();

ioc.serviceManager.addSingletonInstance<boolean>(IsWindows, false);

ioc.serviceManager.addSingletonInstance<IProductService>(IProductService, new ProductService());

ioc.serviceManager.addSingleton<IProductPathService>(IProductPathService, CTagsProductPathService, ProductType.WorkspaceSymbols);

ioc.serviceManager.addSingleton<IProductPathService>(IProductPathService, FormatterProductPathService, ProductType.Formatter);

ioc.serviceManager.addSingleton<IProductPathService>(IProductPathService, LinterProductPathService, ProductType.Linter);

ioc.serviceManager.addSingleton<IProductPathService>(IProductPathService, TestFrameworkProductPathService, ProductType.TestFramework);

ioc.serviceManager.addSingleton<IProductPathService>(IProductPathService, RefactoringLibraryProductPathService, ProductType.RefactoringLibrary);

ioc.serviceManager.addSingleton<IProductPathService>(IProductPathService, DataScienceProductPathService, ProductType.DataScience);

ioc.serviceManager.addSingleton<WindowsStoreInterpreter>(WindowsStoreInterpreter, WindowsStoreInterpreter);

ioc.serviceManager.addSingleton<InterpreterHashProvider>(InterpreterHashProvider, InterpreterHashProvider);

ioc.serviceManager.addSingleton<InterpeterHashProviderFactory>(InterpeterHashProviderFactory, InterpeterHashProviderFactory);

ioc.serviceManager.addSingleton<InterpreterFilter>(InterpreterFilter, InterpreterFilter);

}

async function resetSettings() {

await updateSetting('linting.pylintEnabled', true, rootWorkspaceUri, ConfigurationTarget.Workspace);

}

async function testCheckingIfProductIsInstalled(product: Product) {

const installer = ioc.serviceContainer.get<IInstaller>(IInstaller);

const processService = (await ioc.serviceContainer.get<IProcessServiceFactory>(IProcessServiceFactory).create()) as MockProcessService;

const checkInstalledDef = createDeferred<boolean>();

processService.onExec((_file, args, _options, callback) => {

const moduleName = installer.translateProductToModuleName(product, ModuleNamePurpose.run);

if (args.length > 1 && args[0] === '-c' && args[1] === `import ${moduleName}`) {

checkInstalledDef.resolve(true);

}

callback({ stdout: '' });

});

await installer.isInstalled(product, resource);

await checkInstalledDef.promise;

}

getNamesAndValues<Product>(Product).forEach(prod => {

test(`Ensure isInstalled for Product: '${prod.name}' executes the right command`, async () => {

ioc.serviceManager.addSingletonInstance<IModuleInstaller>(IModuleInstaller, new MockModuleInstaller('one', false));

ioc.serviceManager.addSingletonInstance<IModuleInstaller>(IModuleInstaller, new MockModuleInstaller('two', true));

ioc.serviceManager.addSingletonInstance<ITerminalHelper>(ITerminalHelper, instance(mock(TerminalHelper)));

if (

prod.value === Product.ctags ||

prod.value === Product.unittest ||

prod.value === Product.isort ||

prod.value === Product.jupyter ||

prod.value === Product.ipykernel

) {

return;

}

await testCheckingIfProductIsInstalled(prod.value);

});

});

async function testInstallingProduct(product: Product) {

const installer = ioc.serviceContainer.get<IInstaller>(IInstaller);

const checkInstalledDef = createDeferred<boolean>();

const moduleInstallers = ioc.serviceContainer.getAll<MockModuleInstaller>(IModuleInstaller);

const moduleInstallerOne = moduleInstallers.find(item => item.displayName === 'two')!;

moduleInstallerOne.on('installModule', moduleName => {

const installName = installer.translateProductToModuleName(product, ModuleNamePurpose.install);

if (installName === moduleName) {

checkInstalledDef.resolve();

}

});

await installer.install(product);

await checkInstalledDef.promise;

}

getNamesAndValues<Product>(Product).forEach(prod => {

test(`Ensure install for Product: '${prod.name}' executes the right command in IModuleInstaller`, async () => {

ioc.serviceManager.addSingletonInstance<IModuleInstaller>(IModuleInstaller, new MockModuleInstaller('one', false));

ioc.serviceManager.addSingletonInstance<IModuleInstaller>(IModuleInstaller, new MockModuleInstaller('two', true));

ioc.serviceManager.addSingletonInstance<ITerminalHelper>(ITerminalHelper, instance(mock(TerminalHelper)));

if (

prod.value === Product.unittest ||

prod.value === Product.ctags ||

prod.value === Product.isort ||

prod.value === Product.jupyter ||

prod.value === Product.ipykernel

) {

return;

}

await testInstallingProduct(prod.value);

});

});

});