✅ Add tests for notebook 2

js/ui-tests/examples.spec.js

@@ -6,11 +6,9 @@ import { expect, test, galata } from "@jupyterlab/galata";
 import { setTimeout } from "timers/promises";
 import * as path from "path";
 
-// request and tmpPath are Playwright fixtures
-test("1-Getting-Started", async ({ page, request, tmpPath }) => {
-  // Import notebook 1
+async function openNotebook(page, request, tmpPath, notebookName) {
   const content = galata.newContentsHelper(request);
-  const filename = "1_Getting_Started.ipynb";
+  const filename = notebookName;
   await content.uploadFile(
     path.resolve(__dirname, `../../examples/${filename}`),
     `${tmpPath}/${filename}`,
@@ -22,15 +20,75 @@ test("1-Getting-Started", async ({ page, request, tmpPath }) => {
   await page.waitForSelector(
     "#jp-main-statusbar >> text=Python 3 (ipykernel) | Idle",
   );
+}
+
+async function uploadFitsFile(page, request, tmpPath, filePath) {
+  const content = galata.newContentsHelper(request);
+  const fitsFileName = path.basename(filePath);
+  await content.uploadFile(
+    path.resolve(__dirname, filePath),
+    `${tmpPath}/${fitsFileName}`,
+  );
+}
+
+// request and tmpPath are Playwright fixtures
+test("1_Getting_Started", async ({ page, request, tmpPath }) => {
+  // Import notebook 1
+  await openNotebook(page, request, tmpPath, "1_Getting_Started.ipynb");
   // Execute all cells
   await page.notebook.runCellByCell();
   // Wait for Aladin to pop
   await setTimeout(3000); // 3s
   // Scroll to the top of the notebook
-  page.notebook.getCellLocator(2);
-  page.mouse.wheel(0, -2000);
+  const aladinCell = await page.locator("canvas").nth(1);
   // Save
   await page.notebook.save();
   // And check snapshot (maybe we should clip to div jp-main-dock-panel)
-  expect(await page.screenshot()).toMatchSnapshot();
+  expect(await aladinCell.screenshot()).toMatchSnapshot();
+});
+
+test("2_Base_Commands", async ({ page, request, tmpPath }) => {
+  await uploadFitsFile(
+    page,
+    request,
+    tmpPath + "/images/",
+    "../../examples/images/m31.fits",
+  );
+  await openNotebook(page, request, tmpPath, "2_Base_Commands.ipynb");
+
+  const aladinCell = await page.locator("canvas").nth(1);
+
+  for (let i = 1; i < 7; i++) await page.notebook.runCell(i);
+  await page.waitForTimeout(1000); // 1s
+  for (let i = 7; i < 17; i++) await page.notebook.runCell(i);
+
+  const targetLocator = await page.notebook.getCellLocator(8);
+  // expect(await targetLocator.textContent()).toContain(
+  //   "<SkyCoord (ICRS): (ra, dec) in deg\n" +
+  //     "    (266.41681663, -29.00782497)>",
+  // );
+
+  const fovLocator = await page.notebook.getCellLocator(11);
+  expect(await fovLocator.textContent()).toContain("2∘00′00′′");
+
+  const cooFrameLocator = await page.notebook.getCellLocator(16);
+  expect(await cooFrameLocator.textContent()).toContain("'ICRSd'");
+
+  // Check first snapshot
+  // await aladinCell.press("ArrowUp");
+  // await page.notebook.save();
+  expect(await aladinCell.screenshot()).toMatchSnapshot();
+
+  for (let i = 18; i < 21; i++) await page.notebook.runCell(i);
+
+  // Check second snapshot
+  // await aladinCell.press("ArrowUp");
+  expect(await aladinCell.screenshot()).toMatchSnapshot();
+
+  // add_fits check
+  await page.notebook.runCell(22);
+
+  // Check third snapshot
+  // await aladinCell.press("ArrowUp");
+  expect(await aladinCell.screenshot()).toMatchSnapshot();
 });