'use strict';

import {PythonSettings} from '../../common/configSettings';

import {TestsToRun, Tests} from '../common/contracts';

import {runTest} from './runner';

import * as vscode from 'vscode';

import {discoverTests} from './collector';

import {BaseTestManager} from '../common/baseTestManager';

const settings = PythonSettings.getInstance();

export class TestManager extends BaseTestManager {

constructor(rootDirectory: string, outputChannel: vscode.OutputChannel) {

super('pytest', rootDirectory, outputChannel);

}

discoverTestsImpl(ignoreCache: boolean): Promise<Tests> {

let args = settings.unitTest.pyTestArgs.slice(0);

return discoverTests(this.rootDirectory, args, this.cancellationToken, ignoreCache);

}

runTestImpl(tests: Tests, testsToRun?: TestsToRun, runFailedTests?: boolean): Promise<any> {

let args = settings.unitTest.pyTestArgs.slice(0);

if (runFailedTests === true && args.indexOf('--lf') === -1 && args.indexOf('--last-failed') === -1) {

args.push('--last-failed');

}

return runTest(this.rootDirectory, tests, args, testsToRun, this.cancellationToken, this.outputChannel);

}

}