import * as vscode from 'vscode';

import { IServiceContainer } from '../../../client/ioc/types';

import { PYTHON } from '../../common/constants';

import { ITestCollectionStorageService } from '../common/types';

import { TestFileCodeLensProvider } from './testFiles';

export function activateCodeLenses(

onDidChange: vscode.EventEmitter<void>,

symbolProvider: vscode.DocumentSymbolProvider,

testCollectionStorage: ITestCollectionStorageService,

serviceContainer: IServiceContainer

): vscode.Disposable {

const disposables: vscode.Disposable[] = [];

const codeLensProvider = new TestFileCodeLensProvider(

onDidChange,

symbolProvider,

testCollectionStorage,

serviceContainer

);

disposables.push(vscode.languages.registerCodeLensProvider(PYTHON, codeLensProvider));

return {

dispose: () => {

disposables.forEach(d => d.dispose());

}

};

}