Update tests

Author: jtpio

ui-tests/jupyter_server_config.py

@@ -2,7 +2,7 @@
 
 c.ServerApp.port = 8888
 c.ServerApp.open_browser = False
-# c.ServerApp.root_dir = mkdtemp(prefix='galata-test-')
+c.ServerApp.root_dir = mkdtemp(prefix='galata-test-')
 c.ServerApp.token = ""
 c.ServerApp.password = ""
 c.ServerApp.disable_check_xsrf = True

ui-tests/tests/widgets.test.ts

@@ -1,48 +1,59 @@
 // Copyright (c) Jupyter Development Team.
 // Distributed under the terms of the Modified BSD License.
 
-import { test } from '@jupyterlab/galata';
+import { galata, test } from '@jupyterlab/galata';
 
 import { expect } from '@playwright/test';
 
 import * as path from 'path';
 
 test.describe('Widget Visual Regression', () => {
-  test.beforeEach(async ({ page, tmpPath }) => {
-    await page.contents.uploadDirectory(
+  test.beforeAll(async ({ baseURL, tmpPath }) => {
+    const contents = galata.newContentsHelper(baseURL);
+    await contents.uploadDirectory(
       path.resolve(__dirname, './notebooks'),
       tmpPath
     );
-    await page.dblclick(`text=${tmpPath}`);
+  });
+
+  test.beforeEach(async ({ page, tmpPath }) => {
+    await page.filebrowser.openDirectory(tmpPath);
+  });
+
+  test.afterAll(async ({ baseURL, tmpPath }) => {
+    const contents = galata.newContentsHelper(baseURL);
+    await contents.deleteDirectory(tmpPath);
   });
 
   test('Run notebook widgets.ipynb and capture cell outputs', async ({
     page,
+    tmpPath,
   }) => {
     const notebook = 'widgets.ipynb';
-    await page.notebook.open(notebook);
-    expect(await page.notebook.isOpen(notebook)).toBeTruthy();
+    await page.notebook.openByPath(`${tmpPath}/${notebook}`);
     await page.notebook.activate(notebook);
-    expect(await page.notebook.isActive(notebook)).toBeTruthy();
-
-    let numCellImages = 0;
 
     const getCaptureImageName = (id: number): string => {
-      return `cell-${id}`;
+      return `cell-${id}.png`;
     };
 
+    const captures = new Array<Buffer>();
+    let numCellImages = 0;
+
     await page.notebook.runCellByCell({
       onAfterCellRun: async (cellIndex: number) => {
         const cell = await page.notebook.getCellOutput(cellIndex);
         if (cell) {
-          expect(cell.screenshot()).toMatchSnapshot(
-            getCaptureImageName(numCellImages)
-          );
-          {
-            numCellImages++;
-          }
+          captures.push(await cell.screenshot());
+          numCellImages++;
         }
       },
     });
+
+    await page.notebook.save();
+
+    for (let c = 0; c < numCellImages; ++c) {
+      expect(captures[c]).toMatchSnapshot(getCaptureImageName(c));
+    }
   });
 });