ckaczor/azuredatastudio
1
0
Fork 0
mirror of https://github.com/ckaczor/azuredatastudio.git synced 2026-02-16 10:58:30 -05:00
Code Issues Packages Projects Releases Wiki Activity

Test fix try finding text documents multiple times

Browse Source
This commit is contained in:
chlafreniere
2020-07-16 18:47:31 -07:00
parent 9744d7796c
commit b96ea07d8f
1 changed files with 17 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
extensions/notebook/src/test/model/completionItemProvider.test.ts
View File

@@ -15,6 +15,7 @@ import { JupyterNotebookManager } from '../../jupyter/jupyterNotebookManager';
import { JupyterSessionManager, JupyterSession } from '../../jupyter/jupyterSessionManager'; import { JupyterSessionManager, JupyterSession } from '../../jupyter/jupyterSessionManager';
import { LocalJupyterServerManager } from '../../jupyter/jupyterServerManager'; import { LocalJupyterServerManager } from '../../jupyter/jupyterServerManager';
import { TestKernel } from '../common'; import { TestKernel } from '../common';
import { sleep } from '../common/testUtils';
describe('Completion Item Provider', function () { describe('Completion Item Provider', function () {
let completionItemProvider: NotebookCompletionItemProvider; let completionItemProvider: NotebookCompletionItemProvider;
@@ -64,7 +65,7 @@ describe('Completion Item Provider', function () {
it('should not provide items when session does not exist in notebook provider', async () => { it('should not provide items when session does not exist in notebook provider', async () => {
let notebook = await notebookUtils.newNotebook(); let notebook = await notebookUtils.newNotebook();
await notebookUtils.addCell('code'); await notebookUtils.addCell('code');
let document = vscode.workspace.textDocuments.find(d => d.uri.path === notebook.document.cells[0].uri.path); let document = await tryFindTextDocument(notebook);
should(document).not.equal(undefined, 'Could not find text document that matched cell uri path'); should(document).not.equal(undefined, 'Could not find text document that matched cell uri path');
let completionItems = await completionItemProvider.provideCompletionItems(document, undefined, undefined, undefined); let completionItems = await completionItemProvider.provideCompletionItems(document, undefined, undefined, undefined);
@@ -76,7 +77,7 @@ describe('Completion Item Provider', function () {
let notebook = await notebookUtils.newNotebook(); let notebook = await notebookUtils.newNotebook();
await notebookUtils.addCell('code'); await notebookUtils.addCell('code');
let document = vscode.workspace.textDocuments.find(d => d.uri.path === notebook.document.cells[0].uri.path); let document = await tryFindTextDocument(notebook);
let completionItems = await completionItemProvider.provideCompletionItems(document, undefined, undefined, undefined); let completionItems = await completionItemProvider.provideCompletionItems(document, undefined, undefined, undefined);
should(completionItems).deepEqual([]); should(completionItems).deepEqual([]);
@@ -87,7 +88,7 @@ describe('Completion Item Provider', function () {
let notebook = await notebookUtils.newNotebook(); let notebook = await notebookUtils.newNotebook();
await notebookUtils.addCell('code'); await notebookUtils.addCell('code');
let document = vscode.workspace.textDocuments.find(d => d.uri.path === notebook.document.cells[0].uri.path); let document = await tryFindTextDocument(notebook);
mockJupyterSession.setup(s => s.path).returns(() => document.uri.path); mockJupyterSession.setup(s => s.path).returns(() => document.uri.path);
@@ -103,7 +104,7 @@ describe('Completion Item Provider', function () {
let notebook = await notebookUtils.newNotebook(); let notebook = await notebookUtils.newNotebook();
await notebookUtils.addCell('code'); await notebookUtils.addCell('code');
let document = vscode.workspace.textDocuments.find(d => d.uri.path === notebook.document.cells[0].uri.path); let document = await tryFindTextDocument(notebook);
mockJupyterSession.setup(s => s.path).returns(() => document.uri.path); mockJupyterSession.setup(s => s.path).returns(() => document.uri.path);
@@ -151,7 +152,7 @@ describe('Completion Item Provider', function () {
source: 'sample text' source: 'sample text'
}); });
}); });
let document = vscode.workspace.textDocuments.find(d => d.uri.path === notebook.document.cells[0].uri.path); let document = await tryFindTextDocument(notebook);
let completionItems = await completionItemProvider.provideCompletionItems(document, new vscode.Position(1, 1), token, undefined); let completionItems = await completionItemProvider.provideCompletionItems(document, new vscode.Position(1, 1), token, undefined);
should(completionItems).deepEqual([]); should(completionItems).deepEqual([]);
@@ -192,7 +193,18 @@ describe('Completion Item Provider', function () {
} else { } else {
await notebookUtils.addCell('code'); await notebookUtils.addCell('code');
} }
let document = await tryFindTextDocument(notebook);
return document;
}
async function tryFindTextDocument(notebook: azdata.nb.NotebookEditor): Promise<vscode.TextDocument> {
let document = vscode.workspace.textDocuments.find(d => d.uri.path === notebook.document.cells[0].uri.path); let document = vscode.workspace.textDocuments.find(d => d.uri.path === notebook.document.cells[0].uri.path);
let triesRemaining = 10;
while (!document && triesRemaining > 0) {
await sleep(500);
document = vscode.workspace.textDocuments.find(d => d.uri.path === notebook.document.cells[0].uri.path);
triesRemaining--;
}
return document; return document;
} }
}); });