// Copyright (c) Microsoft Corporation. All rights reserved.

// Licensed under the MIT License.

'use strict';

import { expect } from 'chai';

import * as fs from 'fs';

import { ConfigurationTarget, Disposable } from 'vscode';

import { FileSystem } from '../client/common/platform/fileSystem';

import { Diagnostics } from '../client/common/utils/localize';

import { SourceMapSupport } from '../client/sourceMapSupport';

import { noop } from './core';

suite('Source Map Support', () => {

function createVSCStub(isEnabled: boolean = false, selectDisableButton: boolean = false) {

const stubInfo = {

configValueRetrieved: false,

configValueUpdated: false,

messageDisplayed: false,

};

const vscode = {

workspace: {

getConfiguration: (setting: string, _defaultValue: any) => {

if (setting !== 'python.diagnostics') {

return;

}

return {

get: (prop: string) => {

stubInfo.configValueRetrieved = prop === 'sourceMapsEnabled';

return isEnabled;

},

update: (prop: string, value: boolean, scope: ConfigurationTarget) => {

if (

prop === 'sourceMapsEnabled' &&

value === false &&

scope === ConfigurationTarget.Global

) {

stubInfo.configValueUpdated = true;

}

},

};

},

},

window: {

showWarningMessage: () => {

stubInfo.messageDisplayed = true;

return Promise.resolve(selectDisableButton ? Diagnostics.disableSourceMaps() : undefined);

},

},

ConfigurationTarget: ConfigurationTarget,

};

return { stubInfo, vscode };

}

const disposables: Disposable[] = [];

teardown(() => {

disposables.forEach((disposable) => {

try {

disposable.dispose();

} catch {

noop();

}

});

});

test('When disabling source maps, the map file is renamed and vice versa', async () => {

const fileSystem = new FileSystem();

const jsFile = await fileSystem.createTemporaryFile('.js');

disposables.push(jsFile);

const mapFile = `${jsFile.filePath}.map`;

disposables.push({

dispose: () => fs.unlinkSync(mapFile),

});

await fileSystem.writeFile(mapFile, 'ABC');

expect(await fileSystem.fileExists(mapFile)).to.be.true;

const stub = createVSCStub(true, true);

const instance = new (class extends SourceMapSupport {

public async enableSourceMap(enable: boolean, sourceFile: string) {

return super.enableSourceMap(enable, sourceFile);

}

})(stub.vscode as any);

await instance.enableSourceMap(false, jsFile.filePath);

expect(await fileSystem.fileExists(jsFile.filePath)).to.be.equal(true, 'Source file does not exist');

expect(await fileSystem.fileExists(mapFile)).to.be.equal(false, 'Source map file not renamed');

expect(await fileSystem.fileExists(`${mapFile}.disabled`)).to.be.equal(true, 'Expected renamed file not found');

await instance.enableSourceMap(true, jsFile.filePath);

expect(await fileSystem.fileExists(jsFile.filePath)).to.be.equal(true, 'Source file does not exist');

expect(await fileSystem.fileExists(mapFile)).to.be.equal(true, 'Source map file not found');

expect(await fileSystem.fileExists(`${mapFile}.disabled`)).to.be.equal(false, 'Source map file not renamed');

});

});