|
1 | 1 | import { expect, test } from '@jupyterlab/galata';
|
| 2 | +import { SchedulerHelper } from '../helpers/SchedulerHelper'; |
| 3 | + |
| 4 | +const CREATE_FROM_NOTEBOOK_SNAPSHOT_FILENAME = 'create-view-from-toolbar.png'; |
| 5 | +const CREATE_VIEW_SNAPSHOT_FILENAME = 'create-view-empty.png'; |
| 6 | +const LAUNCHER_SNAPSHOT_FILENAME = 'launcher-with-scheduler.png'; |
| 7 | +const LIST_VIEW_IN_PROGRESS_SNAPSHOT_FILENAME = 'list-view-in-progress.png'; |
| 8 | +const NOTEBOOK_SNAPSHOT_FILENAME= 'notebook-with-createjob-button.png'; |
| 9 | +const RIGHTCLICK_MENU_SNAPSHOT_FILENAME = 'filebrowser-notebook-rightclick-menu.png'; |
2 | 10 |
|
3 | 11 | /**
|
4 | 12 | * Don't load JupyterLab webpage before running the tests.
|
5 | 13 | * This is required to ensure we capture all log messages.
|
6 | 14 | */
|
7 | 15 | test.use({ autoGoto: false });
|
8 | 16 |
|
9 |
| -test('should emit an activation console message', async ({ page }) => { |
10 |
| - // TODO: Add real tests |
| 17 | +test.describe('Jupyter Scheduler integration tests for JupyterLab', () => { |
| 18 | + let schedulerHelper: SchedulerHelper; |
| 19 | + test.beforeEach(async ({ page }) => { |
| 20 | + schedulerHelper = new SchedulerHelper(page); |
| 21 | + await page.goto(); |
| 22 | + await page.sidebar.close( |
| 23 | + (await page.sidebar.getTabPosition('filebrowser')) ?? undefined |
| 24 | + ); |
| 25 | + }); |
| 26 | + |
| 27 | + test('"Notebook Jobs" card is visible in JupyterLab launcher', async ({ |
| 28 | + page |
| 29 | + }) => { |
| 30 | + const launcher = page.locator('div[role="main"] >> text=Launcher'); |
| 31 | + await launcher.waitFor(); |
| 32 | + const launcherCard = schedulerHelper.launcherCardLocator; |
| 33 | + |
| 34 | + await expect(launcherCard).toBeVisible(); |
| 35 | + expect(await page.screenshot()).toMatchSnapshot(LAUNCHER_SNAPSHOT_FILENAME); |
| 36 | + }); |
| 37 | + |
| 38 | + test('"Create a notebook job" button in notebook toolbar is visible', async ({ |
| 39 | + page |
| 40 | + }) => { |
| 41 | + await page.notebook.createNew(); |
| 42 | + await page |
| 43 | + .locator('.jp-DebuggerBugButton[aria-disabled="false"]') |
| 44 | + .waitFor(); |
| 45 | + await page |
| 46 | + .locator('.jp-Notebook-ExecutionIndicator[data-status="idle"]') |
| 47 | + .waitFor(); |
| 48 | + const createJobButton = schedulerHelper.notebookToolbarButtonLocator; |
| 49 | + |
| 50 | + await expect(createJobButton).toBeVisible(); |
| 51 | + expect(await page.screenshot()).toMatchSnapshot(NOTEBOOK_SNAPSHOT_FILENAME); |
| 52 | + await page.menu.clickMenuItem('File>Save Notebook'); |
| 53 | + await page.click('button:has-text("Rename")'); |
| 54 | + }); |
| 55 | + |
| 56 | + test('"Create a notebook job" button in notebook toolbar leads to "Create a Job" page', async ({ |
| 57 | + page |
| 58 | + }) => { |
| 59 | + await page.sidebar.openTab('filebrowser'); |
| 60 | + expect(await page.sidebar.isTabOpen('filebrowser')).toBeTruthy(); |
| 61 | + await page.filebrowser.refresh(); |
| 62 | + await page.dblclick('.jp-DirListing-item[data-file-type="notebook"]'); |
| 63 | + await page.sidebar.close( |
| 64 | + (await page.sidebar.getTabPosition('filebrowser')) ?? undefined |
| 65 | + ); |
| 66 | + const createJobButton = schedulerHelper.notebookToolbarButtonLocator; |
| 67 | + await createJobButton.click(); |
| 68 | + await page.waitForSelector('text=Loading …', { state: 'hidden' }); |
| 69 | + |
| 70 | + await page.waitForSelector('text=Saving Completed', { state: 'hidden' }); |
| 71 | + expect(await page.screenshot()).toMatchSnapshot(CREATE_FROM_NOTEBOOK_SNAPSHOT_FILENAME); |
| 72 | + }); |
| 73 | + |
| 74 | + test('"Create Notebook Job" item is visible when right clicking a notebook in File Browser', async ({ |
| 75 | + page |
| 76 | + }) => { |
| 77 | + await page.sidebar.openTab('filebrowser'); |
| 78 | + expect(await page.sidebar.isTabOpen('filebrowser')).toBeTruthy(); |
| 79 | + await page.filebrowser.refresh(); |
| 80 | + await page.click('.jp-DirListing-item[data-file-type="notebook"]', { |
| 81 | + button: 'right' |
| 82 | + }); |
| 83 | + |
| 84 | + expect(await page.menu.isAnyOpen()).toBe(true); |
| 85 | + const righClickMenu = page.locator('ul.lm-Menu-content[role="menu"]'); |
| 86 | + const createJobItem = schedulerHelper.filebrowserMenuItemLocator; |
| 87 | + await expect(createJobItem).toBeVisible(); |
| 88 | + expect(await righClickMenu.screenshot()).toMatchSnapshot( |
| 89 | + RIGHTCLICK_MENU_SNAPSHOT_FILENAME |
| 90 | + ); |
| 91 | + }); |
| 92 | + |
| 93 | + test('"Create Notebook Job" button from File Browser right-click menu leads to "Create a Job" page', async ({ |
| 94 | + page |
| 95 | + }) => { |
| 96 | + await page.sidebar.openTab('filebrowser'); |
| 97 | + expect(await page.sidebar.isTabOpen('filebrowser')).toBeTruthy(); |
| 98 | + await page.filebrowser.refresh(); |
| 99 | + await page.click('.jp-DirListing-item[data-file-type="notebook"]', { |
| 100 | + button: 'right' |
| 101 | + }); |
| 102 | + expect(await page.menu.isAnyOpen()).toBe(true); |
| 103 | + const createJobItem = schedulerHelper.filebrowserMenuItemLocator; |
| 104 | + await createJobItem.click(); |
| 105 | + await page.waitForSelector('text=Loading …', { state: 'hidden' }); |
| 106 | + await page.sidebar.close( |
| 107 | + (await page.sidebar.getTabPosition('filebrowser')) ?? undefined |
| 108 | + ); |
| 109 | + |
| 110 | + expect(await page.screenshot()).toMatchSnapshot(CREATE_VIEW_SNAPSHOT_FILENAME); |
| 111 | + }); |
| 112 | + |
| 113 | + test('Create a job and see it in the list of jobs', async ({ page }) => { |
| 114 | + await page.sidebar.openTab('filebrowser'); |
| 115 | + await page.filebrowser.refresh(); |
| 116 | + await page.click('.jp-DirListing-item[data-file-type="notebook"]', { |
| 117 | + button: 'right' |
| 118 | + }); |
| 119 | + const createJobItem = schedulerHelper.filebrowserMenuItemLocator; |
| 120 | + await createJobItem.click(); |
| 121 | + await page.waitForSelector('text=Loading', { state: 'hidden' }); |
| 122 | + |
| 123 | + await page.fill('input[name=jobName]', 'MyTestJob'); |
| 124 | + await page.click('button:has-text("Create")'); |
| 125 | + const jobNameLink = page.getByText('MyTestJob', { exact: true }); |
| 126 | + jobNameLink.waitFor(); |
| 127 | + await page.sidebar.close( |
| 128 | + (await page.sidebar.getTabPosition('filebrowser')) ?? undefined |
| 129 | + ); |
| 130 | + const timeStamp = schedulerHelper.timestampLocator; |
| 131 | + const contentPanel = page.locator('#jp-main-content-panel'); |
| 132 | + |
| 133 | + await expect(contentPanel).toHaveScreenshot(LIST_VIEW_IN_PROGRESS_SNAPSHOT_FILENAME, { |
| 134 | + mask: [timeStamp], |
| 135 | + maskColor: 'white', |
| 136 | + maxDiffPixelRatio: 0.01 |
| 137 | + }); |
| 138 | + }); |
11 | 139 | });
|
0 commit comments