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

// Licensed under the MIT License.

'use strict';

import { Uri } from 'vscode';

import { DataScienceFileSystem } from '../../client/datascience/dataScienceFileSystem';

import { FakeVSCodeFileSystemAPI } from '../serviceRegistry';

export class MockFileSystem extends DataScienceFileSystem {

private contentOverloads = new Map<string, string>();

constructor() {

super();

this.vscfs = new FakeVSCodeFileSystemAPI();

}

public async readLocalFile(filePath: string): Promise<string> {

const contents = this.contentOverloads.get(filePath);

if (contents) {

return contents;

}

return super.readLocalFile(filePath);

}

public async readFile(filePath: Uri): Promise<string> {

return this.readLocalFile(filePath.fsPath);

}

public addFileContents(filePath: string, contents: string): void {

this.contentOverloads.set(filePath, contents);

}

}