Add more tests

.gitignore

@@ -125,4 +125,5 @@ python/jupyter_suggestions_rtc/jupyter_suggestions_rtc/_version.py
 
 # Integration tests
 **/ui-tests/test-results/
-**/ui-tests/playwright-report/
+**/ui-tests/playwright-report/
+**/*/.jupyter_ystore.db

ui-tests/playwright.config.js

@@ -11,11 +11,12 @@ module.exports = {
     timeout: 120 * 1000,
     reuseExistingServer: false
   },
-  retries: 1,
+  retries: 0,
   use: {
     ...baseConfig.use,
     trace: 'off',
-
+    video: 'retain-on-failure',
+    viewport: { width: 1920, height: 1080 },
   },
   expect: {
     toMatchSnapshot: {

ui-tests/tests/notebooks/test.ipynb

@@ -0,0 +1,53 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "b9356f9a-ea42-4050-a3a9-5684edd6bf18",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "def foo():\n",
+    "    print(123)\n",
+    "    return 0"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "3ff6c3ec-e344-4262-818a-a1ced33c3cac",
+   "metadata": {},
+   "source": [
+    "## This is a markdown cell"
+   ]
+  },
+  {
+   "cell_type": "raw",
+   "id": "dadedd23-c079-4149-8db6-c8e305e55462",
+   "metadata": {},
+   "source": [
+    "This a a raw cell"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3 (ipykernel)",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.11.11"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

ui-tests/tests/ui.spec.ts

@@ -27,4 +27,72 @@ test.describe('UI Test', () => {
       });
     });
   });
+  test.describe('Panel activation test', () => {
+    test('should add the suggestion panel to the right', async ({
+      page,
+      request
+    }) => {
+      await page.goto();
+
+      await page.sidebar.open('right');
+      await page.getByTitle('Jupyter Suggestions').click();
+      await page.getByTitle('All Suggestions');
+      expect(await page.screenshot()).toMatchSnapshot({
+        name: `${BACKEND}-side-panel.png`
+      });
+    });
+  });
+});
+
+test.describe('Notebook Test', () => {
+  test.beforeEach(async ({ page, tmpPath }) => {
+    await page.contents.uploadDirectory(
+      path.resolve(__dirname, './notebooks'),
+      tmpPath
+    );
+    await page.filebrowser.openDirectory(tmpPath);
+  });
+
+  test('Should add suggestion widget', async ({ page, tmpPath }) => {
+    await page.goto();
+    const notebook = 'test.ipynb';
+    await page.sidebar.open('right');
+    await page.getByTitle('Jupyter Suggestions').click();
+    await page.getByTitle('All Suggestions');
+
+    await page.notebook.openByPath(`${tmpPath}/${notebook}`);
+    await page.notebook.activate(notebook);
+
+    await page.notebook.selectCells(0);
+    await page.getByRole('button', { name: 'Suggestion menu' }).click();
+    await page.getByText('Suggest change').click();
+    await page.waitForTimeout(500);
+    await page.notebook.selectCells(1);
+    await page.getByRole('button', { name: 'Suggestion menu' }).click();
+    await page.getByText('Suggest delete').click();
+    await page.waitForTimeout(500);
+    expect(await page.screenshot()).toMatchSnapshot({
+      name: `${BACKEND}-add-suggestion.png`
+    });
+    await page
+      .getByLabel('All Suggestions (2)', { exact: true })
+      .getByText('def foo(): print(123) return')
+      .fill('def foobar():\n    print(123)\n    return 123');
+    await page.waitForTimeout(500);
+    expect(
+      await page.getByLabel('All Suggestions (2)', { exact: true }).screenshot()
+    ).toMatchSnapshot({
+      name: `${BACKEND}-change-cell-content.png`
+    });
+    await page.getByLabel('Accept suggestion').nth(2).click();
+    await page.waitForTimeout(500);
+    expect(await page.screenshot()).toMatchSnapshot({
+      name: `${BACKEND}-accept-change-suggestion.png`
+    });
+    await page.getByLabel('Accept suggestion').nth(1).click();
+    await page.waitForTimeout(500);
+    expect(await page.screenshot()).toMatchSnapshot({
+      name: `${BACKEND}-accept-delete-suggestion.png`
+    });
+  });
 });