ckaczor/azuredatastudio
1
0
Fork 0
mirror of https://github.com/ckaczor/azuredatastudio.git synced 2026-01-13 17:22:15 -05:00
Code Issues Packages Projects Releases Wiki Activity

Fix Notebook Tree View smoke tests (#18101)

Browse Source 
* pin SQL notebook
This commit is contained in:
Barbara Valdez
2022-01-19 09:44:34 -08:00
committed by GitHub
parent 5847bd302d
commit 8378e89639
2 changed files with 48 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
test/automation/src/sql/notebook.ts
View File

@@ -15,12 +15,12 @@ export class Notebook {
public readonly notebookToolbar: NotebookToolbar;
public readonly textCellToolbar: TextCellToolbar;
public readonly view: NotebookView;
public readonly view: NotebookTreeView;
constructor(private code: Code, private quickAccess: QuickAccess, private quickInput: QuickInput, private editors: Editors) {
this.notebookToolbar = new NotebookToolbar(code);
this.textCellToolbar = new TextCellToolbar(code);
this.view = new NotebookView(code, quickAccess);
this.view = new NotebookTreeView(code, quickAccess);
}
async openFile(fileName: string): Promise<void> {
@@ -331,7 +331,7 @@ export class NotebookToolbar {
}
}
export class NotebookView {
export class NotebookTreeView {
private static readonly inputBox = '.notebookExplorer-viewlet .search-widget .input-box';
private static searchResult = '.search-view .result-messages';
private static notebookTreeItem = '.split-view-view .tree-explorer-viewlet-tree-view .monaco-list-row';
@@ -355,7 +355,7 @@ export class NotebookView {
async searchInNotebook(expr: string): Promise<IElement> {
await this.waitForSetSearchValue(expr);
await this.code.dispatchKeybinding('enter');
let selector = NotebookView.searchResult;
let selector = NotebookTreeView.searchResult;
if (expr) {
selector += ' .message';
}
@@ -363,40 +363,44 @@ export class NotebookView {
}
async waitForSetSearchValue(text: string): Promise<void> {
const textArea = `${NotebookView.inputBox} textarea`;
const textArea = `${NotebookTreeView.inputBox} textarea`;
await this.code.waitForTypeInEditor(textArea, text);
}
/**
* Helper function
* @returns tree item ids from Notebooks View
* Gets tree items from Notebooks Tree View
* @returns tree item from Notebooks View
*/
async getNotebookTreeItemIds(): Promise<string[]> {
return (await this.code.waitForElements(NotebookView.notebookTreeItem, false)).map(item => item.attributes['id']);
async getNotebookTreeItems(): Promise<IElement[]> {
return this.code.waitForElements(NotebookTreeView.notebookTreeItem, false);
}
/**
* Pin the first notebook in the Notebooks View
* Gets tree items from Pinned Notebooks View
* @returns tree item from Pinned Notebooks View
*/
async pinNotebook(): Promise<void> {
const notebookIds = await this.getNotebookTreeItemIds();
await this.code.waitAndDoubleClick(`${NotebookView.notebookTreeItem}[id="${notebookIds[0]}"]`);
await this.code.waitAndClick(`${NotebookView.notebookTreeItem}${NotebookView.selectedItem} .codicon-pinned`);
async getPinnedNotebookTreeItems(): Promise<IElement[]> {
return this.code.waitForElements(NotebookTreeView.pinnedNotebooksSelector, false);
}
/**
* Unpin the only pinned notebook.
* Previously pinned by the pinNotebook method.
*/
async unpinNotebook(): Promise<void> {
await this.code.waitAndClick(NotebookView.pinnedNotebooksSelector);
await this.code.waitAndClick(`${NotebookView.pinnedNotebooksSelector} .actions a[title="Unpin Notebook"]`);
async pinNotebook(notebookId: string): Promise<void> {
await this.code.waitAndDoubleClick(`${NotebookTreeView.notebookTreeItem}[id="${notebookId}"]`);
await this.code.waitAndClick(`${NotebookTreeView.notebookTreeItem}${NotebookTreeView.selectedItem} .codicon-pinned`);
}
async unpinNotebook(notebookId: string): Promise<void> {
await this.code.waitAndClick(NotebookTreeView.pinnedNotebooksSelector);
await this.code.waitAndClick(`${NotebookTreeView.pinnedNotebooksSelector}[id="${notebookId}"] .actions a[title="Unpin Notebook"]`);
}
/**
* When pinning a notebook, the pinned notebook view will show.
*/
async waitForPinnedNotebookView(): Promise<void> {
await this.code.waitForElement(NotebookView.pinnedNotebooksSelector);
async waitForPinnedNotebookTreeView(): Promise<void> {
await this.code.waitForElement(NotebookTreeView.pinnedNotebooksSelector);
}
async waitForPinnedNotebookTreeViewGone(): Promise<void> {
await this.code.waitForElementGone(NotebookTreeView.pinnedNotebooksSelector);
}
}

27
test/smoke/src/sql/areas/notebook/notebookView.test.ts
View File

@@ -9,21 +9,36 @@ import * as minimist from 'minimist';
import { afterSuite, beforeSuite } from '../../../utils';
export function setup(opts: minimist.ParsedArgs) {
describe('NotebookView', () => {
describe('NotebookTreeView', () => {
beforeSuite(opts);
afterSuite(opts);
it('Pin a notebook', async function () {
// Name of the SQL notebook from azuredatastudio-smoke-test-repo
const SQL_NOTEBOOK = 'collapsed';
it('Pin notebook', async function () {
const app = this.app as Application;
await app.workbench.sqlNotebook.view.focusNotebooksView();
await app.workbench.sqlNotebook.view.pinNotebook();
await app.workbench.sqlNotebook.view.waitForPinnedNotebookView();
const sqlNotebook = (await app.workbench.sqlNotebook.view.getNotebookTreeItems()).filter(n => n.textContent === SQL_NOTEBOOK);
// Pinning SQL notebook to prevent the Configure Python Wizard from showing, since Python is no longer set up when the NotebookTreeView test suite starts
await app.workbench.sqlNotebook.view.pinNotebook(sqlNotebook[0].attributes.id);
await app.workbench.sqlNotebook.view.waitForPinnedNotebookTreeView();
});
it('Unpin Notebook', async function () {
const app = this.app as Application;
await app.workbench.sqlNotebook.view.focusPinnedNotebooksView();
await app.workbench.sqlNotebook.view.unpinNotebook();
let pinnedNotebooks = await app.workbench.sqlNotebook.view.getPinnedNotebookTreeItems();
const sqlNotebook = (pinnedNotebooks).filter(n => n.textContent === SQL_NOTEBOOK)[0];
await app.workbench.sqlNotebook.view.unpinNotebook(sqlNotebook.attributes.id);
// wait a second for unpinning to complete
await new Promise(c => setTimeout(c, 1000));
if (pinnedNotebooks.length > 1) {
// if theres multiple pinned notebooks check that the SQL notebook is no longer in the Pinned Notebooks View
pinnedNotebooks = await app.workbench.sqlNotebook.view.getPinnedNotebookTreeItems();
assert(pinnedNotebooks.find(n => n.textContent === SQL_NOTEBOOK) === undefined);
} else {
// check that the Pinned Notebook View is gone
await app.workbench.sqlNotebook.view.waitForPinnedNotebookTreeViewGone();
}
});
it('No search results if search query is empty', async function () {