From 793fe9c1f26eb2488990cfb0d3a7ab308d2c54ca Mon Sep 17 00:00:00 2001 From: Cory Rivera Date: Thu, 26 May 2022 09:40:39 -0700 Subject: [PATCH] Also check for docs pane label in code cell completions test. (#19530) --- test/smoke/src/sql/areas/notebook/notebook.test.ts | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/test/smoke/src/sql/areas/notebook/notebook.test.ts b/test/smoke/src/sql/areas/notebook/notebook.test.ts index 2fd284af6d..856abea9cc 100644 --- a/test/smoke/src/sql/areas/notebook/notebook.test.ts +++ b/test/smoke/src/sql/areas/notebook/notebook.test.ts @@ -46,7 +46,12 @@ export function setup(opts: minimist.ParsedArgs) { // check for completion suggestions await app.workbench.sqlNotebook.waitForSuggestionWidget(); - await app.workbench.sqlNotebook.waitForSuggestionResult('SELECT'); + + // Docs pane might be visible in the completions list, so also check for a docs aria-label + await Promise.race([ + app.workbench.sqlNotebook.waitForSuggestionResult('SELECT'), + app.workbench.sqlNotebook.waitForSuggestionResult('SELECT, docs: SELECT keyword') + ]); await app.code.dispatchKeybinding('tab'); const text2: string = ' * FROM employees';