ckaczor/azuredatastudio
1
0
Fork 0
mirror of https://github.com/ckaczor/azuredatastudio.git synced 2026-01-14 01:25:37 -05:00
Code Issues Packages Projects Releases Wiki Activity

Add simple smoke test for collapse/expand cell (#15879)

Browse Source 
* simple smoke test for collapse / expand cells
This commit is contained in:
Vasu Bhog
2021-06-25 18:08:35 -07:00
committed by GitHub
parent edc319a90c
commit c3e0958f2d
2 changed files with 68 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -117,6 +117,30 @@ export class Notebook {
await this.code.waitForElementGone(Notebook.placeholderSelector);
}
async waitForCollapseIconInCells(): Promise<void> {
let cellIds = await this.getCellIds();
for (let i of cellIds) {
const editor = `.notebook-cell[id="${i}"] code-cell-component code-component collapse-component`;
await this.code.waitForElement(`${editor} [title="Collapse code cell contents"]`);
}
}
async waitForExpandIconInCells(): Promise<void> {
let cellIds = await this.getCellIds();
for (let i of cellIds) {
const editor = `.notebook-cell[id="${i}"] code-cell-component code-component collapse-component`;
await this.code.waitForElement(`${editor} [title="Expand code cell contents"]`);
}
}
/**
* Helper function
* @returns cell ids for the notebook
*/
async getCellIds(): Promise<string[]> {
return (await this.code.waitForElements('div.notebook-cell', false)).map(cell => cell.attributes['id']);
}
// Text Cell Actions
private static readonly textCellPreviewSelector = 'div.notebook-preview';
@@ -260,6 +284,10 @@ export class NotebookToolbar {
private static readonly trustedButtonSelector = `${NotebookToolbar.toolbarSelector} a[class="${NotebookToolbar.trustedButtonClass}"]`;
private static readonly notTrustedButtonClass = 'action-label codicon masked-icon icon-shield-x';
private static readonly notTrustedButtonSelector = `${NotebookToolbar.toolbarSelector} a[class="${NotebookToolbar.notTrustedButtonClass}"]`;
private static readonly collapseCellsClass = 'action-label codicon masked-icon icon-collapse-cells';
private static readonly collapseCellsButtonSelector = `${NotebookToolbar.toolbarSelector} a[class="${NotebookToolbar.collapseCellsClass}"]`;
private static readonly expandCellsClass = 'action-label codicon masked-icon icon-expand-cells';
private static readonly expandCellsButtonSelector = `${NotebookToolbar.toolbarSelector} a[class="${NotebookToolbar.expandCellsClass}"]`;
constructor(private code: Code) { }
@@ -295,6 +323,30 @@ export class NotebookToolbar {
async waitForNotTrustedIcon(): Promise<void> {
await this.code.waitForElement(NotebookToolbar.notTrustedButtonSelector);
}
async collapseCells(): Promise<void> {
let buttons: IElement[] = await this.code.waitForElements(NotebookToolbar.toolbarButtonSelector, false);
let collapseButton = buttons.find(button => button.className.includes('icon-collapse-cells'));
if (collapseButton) {
await this.code.waitAndClick(NotebookToolbar.collapseCellsButtonSelector);
}
}
async expandCells(): Promise<void> {
let buttons: IElement[] = await this.code.waitForElements(NotebookToolbar.toolbarButtonSelector, false);
let expandButton = buttons.find(button => button.className.includes('icon-expand-cells'));
if (expandButton) {
await this.code.waitAndClick(NotebookToolbar.expandCellsButtonSelector);
}
}
async waitForCollapseCellsNotebookIcon(): Promise<void> {
await this.code.waitForElement(NotebookToolbar.collapseCellsButtonSelector);
}
async waitForExpandCellsNotebookIcon(): Promise<void> {
await this.code.waitForElement(NotebookToolbar.expandCellsButtonSelector);
}
}
export class NotebookView {

16
test/smoke/src/sql/areas/notebook/notebook.test.ts
View File

@@ -94,6 +94,22 @@ export function setup() {
const app = this.app as Application;
await app.workbench.quickaccess.runCommand('workbench.action.revertAndCloseActiveEditor');
});
describe('Notebook Toolbar Actions', async () => {
it('Collapse and Expand Cell', async function () {
const app = this.app as Application;
await app.workbench.sqlNotebook.openFile('collapsed.ipynb');
await app.workbench.sqlNotebook.waitForCollapseIconInCells();
await app.workbench.sqlNotebook.notebookToolbar.waitForCollapseCellsNotebookIcon();
await app.workbench.sqlNotebook.notebookToolbar.collapseCells();
await app.workbench.sqlNotebook.waitForExpandIconInCells();
await app.workbench.sqlNotebook.notebookToolbar.waitForExpandCellsNotebookIcon();
await app.workbench.sqlNotebook.notebookToolbar.expandCells();
await app.workbench.sqlNotebook.waitForCollapseIconInCells();
await app.workbench.sqlNotebook.notebookToolbar.waitForCollapseCellsNotebookIcon();
});
});
});
}