snapshot of https://github.com/toeverything/AFFiNE.git (canary, 2026-09-16; blobs > 131072B omitted)

tests/affine-local/e2e/blocksuite/code/crud.spec.ts

@@ -0,0 +1,115 @@
+import { test } from '@affine-test/kit/playwright';
+import { openHomePage } from '@affine-test/kit/utils/load-page';
+import { type, waitForEditorLoad } from '@affine-test/kit/utils/page-logic';
+import { expect } from '@playwright/test';
+
+import {
+  createNewPage,
+  gotoContentFromTitle,
+  initCodeBlockByOneStep,
+} from './utils';
+
+test.describe('Code Block Autocomplete Operations', () => {
+  test('angle brackets are not supported', async ({ page }) => {
+    // open the home page and insert the code block
+    await initCodeBlockByOneStep(page);
+    await page.keyboard.type('<');
+    const codeUnit = page.locator('affine-code-unit');
+    await expect(codeUnit).toHaveText('<');
+  });
+});
+
+test.describe('Code Block Preview', () => {
+  test('enable html preview', async ({ page }) => {
+    const code = page.locator('affine-code');
+
+    await openHomePage(page);
+    await createNewPage(page);
+    await waitForEditorLoad(page);
+    await gotoContentFromTitle(page);
+    await type(page, '```html aaa');
+    await code.hover({
+      position: {
+        x: 155,
+        y: 65,
+      },
+    });
+    await page.getByText('Preview').click();
+    await expect(
+      page
+        .locator('iframe[title="HTML Preview"]')
+        .contentFrame()
+        .getByText('aaa')
+    ).toBeVisible();
+  });
+
+  test('enable mermaid preview', async ({ page }) => {
+    const code = page.locator('affine-code');
+    const mermaidSvg = page.locator('mermaid-preview .mermaid-preview-svg svg');
+
+    await openHomePage(page);
+    await createNewPage(page);
+    await waitForEditorLoad(page);
+    await gotoContentFromTitle(page);
+    await type(page, '```mermaid graph TD;A-->B');
+    await code.hover({
+      position: {
+        x: 155,
+        y: 65,
+      },
+    });
+    await page.getByText('Preview').click();
+    await expect(mermaidSvg).toBeVisible();
+  });
+
+  test('enable typst preview', async ({ page }) => {
+    const code = page.locator('affine-code');
+    const typstPreview = page.locator('typst-preview');
+
+    await openHomePage(page);
+    await createNewPage(page);
+    await waitForEditorLoad(page);
+    await gotoContentFromTitle(page);
+    await type(page, '```typst = Title');
+    await code.hover({
+      position: {
+        x: 155,
+        y: 65,
+      },
+    });
+    await page.getByText('Preview').click();
+    await expect(typstPreview).toBeVisible();
+  });
+
+  test('change lang without preview', async ({ page }) => {
+    const code = page.locator('affine-code');
+    const preview = page.locator('affine-code .affine-code-block-preview');
+
+    await openHomePage(page);
+    await createNewPage(page);
+    await waitForEditorLoad(page);
+    await gotoContentFromTitle(page);
+    await type(page, '```html aaa');
+
+    await code.hover({
+      position: {
+        x: 155,
+        y: 65,
+      },
+    });
+    await page.getByText('Preview').click();
+    await expect(preview).toBeVisible();
+
+    // change to lang without preview support
+    await page.getByTestId('lang-button').click();
+    await page.getByRole('button', { name: 'ABAP' }).click();
+
+    await expect(preview).toBeHidden();
+
+    // change back to html
+    await page.getByTestId('lang-button').click();
+    await page.getByRole('button', { name: 'HTML', exact: true }).click();
+
+    await expect(preview).toBeVisible();
+  });
+});

tests/affine-local/e2e/blocksuite/code/line-numbers.spec.ts

@@ -0,0 +1,192 @@
+import { test } from '@affine-test/kit/playwright';
+import { openHomePage } from '@affine-test/kit/utils/load-page';
+import {
+  addCodeBlock,
+  waitForEditorLoad,
+} from '@affine-test/kit/utils/page-logic';
+import {
+  closeSettingModal,
+  openEditorSetting,
+} from '@affine-test/kit/utils/setting';
+import { expect, type Page } from '@playwright/test';
+
+import { initCodeBlockByOneStep, openCodeBlockMoreMenu } from './utils';
+
+// ---------------------------------------------------------------------------
+// Helpers
+// ---------------------------------------------------------------------------
+
+/**
+ * Opens Editor Settings and returns a locator for the
+ * "Show line numbers in code blocks" toggle switch.
+ */
+async function openLineNumbersSetting(page: Page) {
+  await openEditorSetting(page);
+  return page.getByTestId('code-block-line-numbers-trigger');
+}
+
+// ---------------------------------------------------------------------------
+// Tests
+// ---------------------------------------------------------------------------
+
+test.describe('Global code block line numbers setting', () => {
+  test('line numbers are visible by default', async ({ page }) => {
+    await initCodeBlockByOneStep(page);
+
+    const lineNumber = page.locator('affine-code .line-number');
+    await expect(lineNumber).toBeVisible();
+  });
+
+  test('global toggle hides line numbers on all existing code blocks', async ({
+    page,
+  }) => {
+    await initCodeBlockByOneStep(page);
+
+    const lineNumber = page.locator('affine-code .line-number');
+    await expect(lineNumber).toBeVisible();
+
+    // Turn the global setting OFF
+    const toggle = await openLineNumbersSetting(page);
+    await expect(toggle).toBeVisible();
+    await toggle.click();
+    await closeSettingModal(page);
+
+    // Line numbers must now be hidden
+    await expect(lineNumber).toBeHidden();
+  });
+
+  test('turning global toggle back on restores line numbers', async ({
+    page,
+  }) => {
+    await initCodeBlockByOneStep(page);
+
+    const lineNumber = page.locator('affine-code .line-number');
+
+    // OFF
+    const toggle = await openLineNumbersSetting(page);
+    await toggle.click();
+    await closeSettingModal(page);
+    await expect(lineNumber).toBeHidden();
+
+    // ON again
+    const toggle2 = await openLineNumbersSetting(page);
+    await toggle2.click();
+    await closeSettingModal(page);
+    await expect(lineNumber).toBeVisible();
+  });
+
+  test('per-block toggle shows line numbers when global is off', async ({
+    page,
+  }) => {
+    await initCodeBlockByOneStep(page);
+
+    const lineNumber = page.locator('affine-code .line-number');
+
+    // Turn global setting OFF
+    const toggle = await openLineNumbersSetting(page);
+    await toggle.click();
+    await closeSettingModal(page);
+    await expect(lineNumber).toBeHidden();
+
+    // Per-block: explicitly enable for this block
+    const { lineNumberButton } = await openCodeBlockMoreMenu(page);
+    await lineNumberButton.click();
+    await expect(lineNumber).toBeVisible();
+  });
+
+  test('per-block toggle can hide line numbers when global is on', async ({
+    page,
+  }) => {
+    await initCodeBlockByOneStep(page);
+
+    const lineNumber = page.locator('affine-code .line-number');
+    // Global is ON by default
+    await expect(lineNumber).toBeVisible();
+
+    // Per-block: hide for this block
+    const { cancelLineNumberButton } = await openCodeBlockMoreMenu(page);
+    await cancelLineNumberButton.click();
+    // Click away to dismiss menu
+    await page.mouse.click(300, 300);
+    await expect(lineNumber).toBeHidden();
+
+    // Turn global setting OFF and then ON - per-block override persists
+    const toggle = await openLineNumbersSetting(page);
+    await toggle.click(); // OFF
+    await toggle.click(); // ON
+    await closeSettingModal(page);
+    // Per-block explicitly set to false - still hidden
+    await expect(lineNumber).toBeHidden();
+  });
+
+  test('global setting is reflected on newly created code blocks', async ({
+    page,
+  }) => {
+    // Start on the home page and disable line numbers globally first
+    await openHomePage(page);
+    await waitForEditorLoad(page);
+
+    const toggle = await openLineNumbersSetting(page);
+    await toggle.click(); // OFF
+    await closeSettingModal(page);
+
+    // Now create a fresh code block
+    await initCodeBlockByOneStep(page);
+
+    const lineNumber = page.locator('affine-code .line-number');
+    await expect(lineNumber).toBeHidden();
+  });
+
+  test('global toggle affects all code blocks on the page', async ({
+    page,
+  }) => {
+    await initCodeBlockByOneStep(page);
+
+    // Exit the code block (Mod+Enter creates a paragraph below and moves focus there)
+    const isMac = process.platform === 'darwin';
+    await page.keyboard.press(isMac ? 'Meta+Enter' : 'Control+Enter');
+    // Add a second code block in the new paragraph
+    await addCodeBlock(page);
+
+    // Both blocks should show line numbers by default
+    const lineNumbers = page.locator('affine-code .line-number');
+    await expect(lineNumbers).toHaveCount(2);
+    await expect(lineNumbers.first()).toBeVisible();
+    await expect(lineNumbers.last()).toBeVisible();
+
+    // Turn global OFF - both must hide
+    const toggle = await openLineNumbersSetting(page);
+    await toggle.click();
+    await closeSettingModal(page);
+
+    await expect(lineNumbers.first()).toBeHidden();
+    await expect(lineNumbers.last()).toBeHidden();
+
+    // Turn global ON - both must show again
+    const toggle2 = await openLineNumbersSetting(page);
+    await toggle2.click();
+    await closeSettingModal(page);
+
+    await expect(lineNumbers.first()).toBeVisible();
+    await expect(lineNumbers.last()).toBeVisible();
+  });
+
+  test('global setting persists across page reload', async ({ page }) => {
+    await initCodeBlockByOneStep(page);
+
+    const lineNumber = page.locator('affine-code .line-number');
+    await expect(lineNumber).toBeVisible();
+
+    // Turn global OFF
+    const toggle = await openLineNumbersSetting(page);
+    await toggle.click();
+    await closeSettingModal(page);
+    await expect(lineNumber).toBeHidden();
+
+    // Reload the page - setting must survive
+    await page.reload();
+    await waitForEditorLoad(page);
+
+    await expect(lineNumber).toBeHidden();
+  });
+});

tests/affine-local/e2e/blocksuite/code/utils.ts

@@ -0,0 +1,46 @@
+import { openHomePage } from '@affine-test/kit/utils/load-page';
+import {
+  addCodeBlock,
+  clickNewPageButton,
+  waitForEditorLoad,
+} from '@affine-test/kit/utils/page-logic';
+import type { Page } from '@playwright/test';
+
+export const gotoContentFromTitle = async (page: Page) => {
+  await page.keyboard.press('Enter');
+};
+
+export const createNewPage = async (page: Page) => {
+  await clickNewPageButton(page);
+};
+
+export const initCodeBlockByOneStep = async (page: Page) => {
+  await openHomePage(page);
+  await createNewPage(page);
+  await waitForEditorLoad(page);
+  await gotoContentFromTitle(page);
+  await addCodeBlock(page);
+};
+
+/**
+ * Opens the "More" menu on the first code block and returns locators for
+ * the line-number toggle buttons inside it.
+ * Uses .first() to stay deterministic when multiple code blocks are present.
+ */
+export const openCodeBlockMoreMenu = async (page: Page) => {
+  const codeBlock = page.locator('affine-code').first();
+  await codeBlock.hover();
+
+  const moreButton = page
+    .locator('affine-code-toolbar')
+    .getByRole('button', { name: 'More' });
+  await moreButton.click();
+
+  const menu = page.locator('.more-popup-menu');
+  const lineNumberButton = menu.getByRole('button', { name: 'Line number' });
+  const cancelLineNumberButton = menu.getByRole('button', {
+    name: 'Cancel line number',
+  });
+
+  return { menu, lineNumberButton, cancelLineNumberButton };
+};