Fix a test.

Author: ericsnowcurrently

src/test/datascience/nativeEditor.functional.test.tsx

@@ -14,8 +14,9 @@ import { anything, when } from 'ts-mockito';
 import * as TypeMoq from 'typemoq';
 import { Disposable, TextDocument, TextEditor, Uri, WindowState } from 'vscode';
 import { IApplicationShell, IDocumentManager } from '../../client/common/application/types';
+import { FileSystem } from '../../client/common/platform/fileSystem';
+import { TemporaryFile } from '../../client/common/platform/types';
 import { createDeferred } from '../../client/common/utils/async';
-import { createTemporaryFile } from '../../client/common/utils/fs';
 import { noop } from '../../client/common/utils/misc';
 import { Identifiers } from '../../client/datascience/constants';
 import { InteractiveWindowMessages } from '../../client/datascience/interactive-common/interactiveWindowTypes';
@@ -327,10 +328,7 @@ suite('DataScience Native Editor', () => {
             { id: 'NotebookImport#1', data: { source: 'b=2\nb' } },
             { id: 'NotebookImport#2', data: { source: 'c=3\nc' } }
         ];
-        let notebookFile: {
-            filePath: string;
-            cleanupCallback: Function;
-        };
+        let notebookFile: TemporaryFile;
         function initIoc() {
             ioc = new DataScienceIocContainer();
             ioc.registerDataScienceTypes();
@@ -348,7 +346,8 @@ suite('DataScience Native Editor', () => {
                 });
                 // Use a real file so we can save notebook to a file.
                 // This is used in some tests (saving).
-                notebookFile = await createTemporaryFile('.ipynb');
+                const filesystem = new FileSystem();
+                notebookFile = await filesystem.createTemporaryFile('.ipynb');
                 const notebook = await ioc.get<INotebookExporter>(INotebookExporter).translateToNotebook(runAllCells, undefined);
                 await Promise.all([waitForUpdate(wrapper, NativeEditor, 1), openEditor(ioc, JSON.stringify(notebook), notebookFile.filePath)]);
             } else {
@@ -370,7 +369,7 @@ suite('DataScience Native Editor', () => {
             }
             await ioc.dispose();
             try {
-                notebookFile.cleanupCallback();
+                notebookFile.dispose();
             } catch {
                 noop();
             }