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

tests/blocksuite/e2e/code/utils.ts

@@ -0,0 +1,67 @@
+import type { Page } from '@playwright/test';
+
+/**
+ * @example
+ * ```ts
+ * const codeBlockController = getCodeBlock(page);
+ * const codeBlock = codeBlockController.codeBlock;
+ * ```
+ */
+export function getCodeBlock(page: Page) {
+  const codeBlock = page.locator('affine-code');
+  const languageButton = page.getByTestId('lang-button');
+
+  const clickLanguageButton = async () => {
+    await codeBlock.hover();
+    await languageButton.click({ delay: 50 });
+  };
+
+  const langList = page.locator('affine-filterable-list');
+  const langFilterInput = langList.locator('#filter-input');
+
+  const codeToolbar = page.locator('affine-code-toolbar');
+
+  const copyButton = codeToolbar.getByRole('button', { name: 'Copy code' });
+  const captionButton = codeToolbar.getByRole('button', { name: 'Caption' });
+  const moreButton = codeToolbar.getByRole('button', { name: 'More' });
+
+  const menu = page.locator('.more-popup-menu');
+
+  const openMore = async () => {
+    await moreButton.click();
+
+    const wrapButton = menu.getByRole('button', { name: 'Wrap' });
+    const cancelWrapButton = menu.getByRole('button', { name: 'Cancel wrap' });
+    const duplicateButton = menu.getByRole('button', { name: 'Duplicate' });
+    const deleteButton = menu.getByRole('button', { name: 'Delete' });
+    const lineNumberButton = menu.getByRole('button', { name: 'Line number' });
+    const cancelLineNumberButton = menu.getByRole('button', {
+      name: 'Cancel line number',
+    });
+
+    return {
+      menu,
+      wrapButton,
+      cancelWrapButton,
+      duplicateButton,
+      deleteButton,
+      lineNumberButton,
+      cancelLineNumberButton,
+    };
+  };
+
+  return {
+    codeBlock,
+    codeToolbar,
+    captionButton,
+    languageButton,
+    langList,
+    copyButton,
+    moreButton,
+    langFilterInput,
+    moreMenu: menu,
+
+    openMore,
+    clickLanguageButton,
+  };
+}