diff --git a/packages/embed/src/index.ts b/packages/embed/src/index.ts
index 4f1cce44fa..cbfb347dcd 100644
--- a/packages/embed/src/index.ts
+++ b/packages/embed/src/index.ts
@@ -1 +1,2 @@
 export * from './client';
+export * from './standalone';
diff --git a/packages/gitbook/e2e/internal.spec.ts b/packages/gitbook/e2e/internal.spec.ts
index 5bbc27561c..72ee4f9184 100644
--- a/packages/gitbook/e2e/internal.spec.ts
+++ b/packages/gitbook/e2e/internal.spec.ts
@@ -9,7 +9,8 @@ import {
     CustomizationSidebarListStyle,
     CustomizationThemeMode,
 } from '@gitbook/api';
-import { expect } from '@playwright/test';
+import type { GitBookStandalone } from '@gitbook/embed';
+import { type FrameLocator, type Locator, expect, test } from '@playwright/test';
 import jwt from 'jsonwebtoken';
 
 import {
@@ -40,7 +41,7 @@ import {
 const AI_PROMPT = `You're being invoked by the GitBook CI/CD pipeline. To make screenshot testing of the GitBook Assistant visually consistent, look up the title of the first page you find and respond with only EXACTLY its title. To find the page title, invoke the search tool with the query "GitBook". Before invoking the search tool, respond with the exact text: "I'm going to look up 'GitBook' and then respond with only the page title.". Do not execute any other tools or output any other text.`;
 
 const overrideAIInitialState = () => {
-    const greeting = document.querySelector('[data-testid="ai-chat-time-greeting"]');
+    const greeting = document.querySelector('[data-testid="ai-chat-greeting-title"]');
     if (greeting) {
         greeting.textContent = 'Good morning';
     }
@@ -62,6 +63,43 @@ const overrideAIResponse = () => {
     });
 };
 
+const CLOUDFLARE_TEXTS = [
+    'Checking if the site connection is secure',
+    'Please stand by, while we are checking your browser',
+    'DDoS protection by Cloudflare',
+    'Just a moment...',
+];
+
+async function isCloudflareChallenge(iframe: FrameLocator) {
+    for (const text of CLOUDFLARE_TEXTS) {
+        const visible = await iframe
+            .getByText(text, { exact: false })
+            .first()
+            .isVisible({ timeout: 500 })
+            .catch(() => false);
+        if (visible) {
+            return true;
+        }
+    }
+    return false;
+}
+
+async function waitForEmbedVisible(iframe: FrameLocator, locator: Locator, timeout = 30_000) {
+    if (await isCloudflareChallenge(iframe)) {
+        test.skip(true, 'Cloudflare challenge detected in embed iframe');
+    }
+    try {
+        await locator.waitFor({ state: 'visible', timeout });
+    } catch (error) {
+        if (await isCloudflareChallenge(iframe)) {
+            test.skip(true, 'Cloudflare challenge detected in embed iframe');
+        }
+        throw error;
+    }
+    await expect(locator).toBeVisible({ timeout });
+    return locator;
+}
+
 const searchTestCases: Test[] = [
     {
         name: 'Search - AI Mode: None - Complete flow',
@@ -1946,6 +1984,320 @@ const testCases: TestsCase[] = [
             ]),
         ],
     },
+    {
+        name: 'Docs Embed - GitBook (Assistant + Docs)',
+        contentBaseURL: 'https://gitbook.com/docs/~gitbook/embed/demo/',
+        tests: [
+            {
+                name: 'Basic functionality',
+                url: '',
+                run: async (page) => {
+                    const button = page.locator('#gitbook-widget-button');
+                    await expect(button).toBeVisible();
+                    await expect(button.locator('#gitbook-widget-button-label')).toHaveText('Ask');
+                    await expect(button.locator('#gitbook-widget-button-icon')).toHaveAttribute(
+                        'data-icon',
+                        'assistant'
+                    );
+
+                    await expect(page.locator('#gitbook-widget-window')).toBeVisible();
+                    const iframe = page.frameLocator('#gitbook-widget-iframe');
+                    await expect(iframe.getByTestId('embed-tab-assistant')).toBeVisible();
+                    await expect(iframe.getByTestId('embed-tab-docs')).toBeVisible();
+
+                    await button.click(); // Toggle the window off
+                    await expect(page.locator('#gitbook-widget-window')).not.toBeVisible();
+                    await button.click(); // Toggle the window on
+                    await expect(page.locator('#gitbook-widget-window')).toBeVisible();
+
+                    await iframe.owner().evaluate(overrideAIInitialState);
+                },
+            },
+            {
+                name: 'Switch between tabs',
+                url: '',
+                run: async (page) => {
+                    await expect(page.locator('#gitbook-widget-window')).toBeVisible();
+                    const iframe = page.frameLocator('#gitbook-widget-iframe');
+                    await iframe.getByTestId('embed-tab-docs').click(); // Switch to docs tab
+                    await waitForEmbedVisible(iframe, iframe.getByTestId('embed-docs-page'));
+
+                    await iframe.getByTestId('embed-tab-assistant').click(); // Switch to assistant tab
+                    await waitForEmbedVisible(iframe, iframe.getByTestId('ai-chat'));
+
+                    await iframe.owner().evaluate(overrideAIInitialState);
+                },
+            },
+            {
+                name: 'API - navigateToPage',
+                url: '',
+                run: async (page) => {
+                    await page.evaluate(() => {
+                        const GitBook = window.GitBook as unknown as GitBookStandalone;
+                        GitBook('navigateToPage', '/getting-started/quickstart');
+                    });
+                    await expect(page.locator('#gitbook-widget-window')).toBeVisible();
+                    const iframe = page.frameLocator('#gitbook-widget-iframe');
+                    await waitForEmbedVisible(iframe, iframe.getByTestId('embed-docs-page'));
+                    await expect(iframe.owner()).toHaveAttribute(
+                        'src',
+                        expect.stringContaining('getting-started/quickstart')
+                    );
+                },
+            },
+            {
+                name: 'API - postUserMessage',
+                url: '',
+                run: async (page) => {
+                    await page.evaluate((aiPrompt) => {
+                        const GitBook = window.GitBook as unknown as GitBookStandalone;
+                        GitBook('postUserMessage', aiPrompt);
+                    }, AI_PROMPT);
+                    const iframe = page.frameLocator('#gitbook-widget-iframe');
+                    await waitForEmbedVisible(iframe, iframe.getByTestId('ai-chat'));
+                    await expect(iframe.getByTestId('ai-chat-message-user').first()).toHaveText(
+                        AI_PROMPT
+                    );
+                    await iframe.owner().evaluate(overrideAIResponse);
+                },
+            },
+            {
+                name: 'Configuration - Change standalone button label and icon',
+                url: '',
+                run: async (page) => {
+                    await page.evaluate(() => {
+                        const GitBook = window.GitBook as unknown as GitBookStandalone;
+                        GitBook('configure', {
+                            button: {
+                                label: 'Docs',
+                                icon: 'book',
+                            },
+                        });
+                    });
+                    await expect(page.locator('#gitbook-widget-button-label')).toHaveText('Docs');
+                    await expect(page.locator('#gitbook-widget-button-icon')).toHaveAttribute(
+                        'data-icon',
+                        'book'
+                    );
+                    const iframe = page.frameLocator('#gitbook-widget-iframe');
+                    await iframe.owner().evaluate(overrideAIInitialState);
+                },
+            },
+            {
+                name: 'Configuration - Suggested questions',
+                url: '',
+                run: async (page) => {
+                    await page.evaluate(() => {
+                        const GitBook = window.GitBook as unknown as GitBookStandalone;
+                        GitBook('configure', {
+                            suggestions: [
+                                'What is GitBook?',
+                                'How do I get started?',
+                                'What can you do?',
+                            ],
+                        });
+                    });
+                    const iframe = page.frameLocator('#gitbook-widget-iframe');
+                    const firstSuggestion = await waitForEmbedVisible(
+                        iframe,
+                        iframe.getByTestId('ai-chat-suggested-question').first()
+                    );
+                    await expect(firstSuggestion).toHaveText('What is GitBook?');
+                    await expect(
+                        iframe.getByTestId('ai-chat-suggested-question').nth(1)
+                    ).toHaveText('How do I get started?');
+                    await expect(
+                        iframe.getByTestId('ai-chat-suggested-question').nth(2)
+                    ).toHaveText('What can you do?');
+                    await iframe.owner().evaluate(overrideAIInitialState);
+                },
+            },
+            {
+                name: 'Configuration - Custom action buttons',
+                url: '',
+                run: async (page) => {
+                    await page.evaluate((aiPrompt) => {
+                        const GitBook = window.GitBook as unknown as GitBookStandalone;
+                        GitBook('configure', {
+                            actions: [
+                                {
+                                    label: 'Open internal link',
+                                    icon: 'bolt',
+                                    onClick: () => {
+                                        const GitBook =
+                                            window.GitBook as unknown as GitBookStandalone;
+                                        GitBook('navigateToPage', '/getting-started/quickstart');
+                                    },
+                                },
+                                {
+                                    label: 'Open external link',
+                                    icon: 'sparkle',
+                                    onClick: () => {
+                                        window.open('https://gitbook.com', '_blank');
+                                    },
+                                },
+                                {
+                                    label: 'Post message',
+                                    icon: 'message',
+                                    onClick: () => {
+                                        const GitBook =
+                                            window.GitBook as unknown as GitBookStandalone;
+                                        GitBook('postUserMessage', aiPrompt);
+                                        GitBook('navigateToAssistant');
+                                    },
+                                },
+                                {
+                                    label: 'Close',
+                                    icon: 'xmark',
+                                    onClick: () => {
+                                        const GitBook =
+                                            window.GitBook as unknown as GitBookStandalone;
+                                        GitBook('close');
+                                    },
+                                },
+                            ],
+                        });
+                    }, AI_PROMPT);
+                    const iframe = page.frameLocator('#gitbook-widget-iframe');
+                    await waitForEmbedVisible(
+                        iframe,
+                        iframe.getByTestId('embed-action').first(),
+                        30_000
+                    );
+                    await expect(iframe.getByTestId('embed-action')).toHaveCount(4, {
+                        timeout: 30_000,
+                    });
+                    const actions = iframe.getByTestId('embed-action');
+
+                    await expect(actions.nth(0)).toHaveAccessibleName('Open internal link');
+                    await actions.nth(0).click();
+                    await waitForEmbedVisible(iframe, iframe.getByTestId('embed-docs-page'));
+                    await expect(iframe.owner()).toHaveAttribute(
+                        'src',
+                        expect.stringContaining('getting-started/quickstart')
+                    );
+
+                    await expect(actions.nth(1)).toHaveAccessibleName('Open external link');
+                    // Intercept the new page event without navigating
+                    const [newPage] = await Promise.all([
+                        page.context().waitForEvent('page', { timeout: 5000 }),
+                        actions.nth(1).click(),
+                    ]);
+                    // Verify the new page would have opened with the expected URL
+                    expect(newPage.url()).toContain('gitbook.com');
+                    // Close it immediately to avoid navigation
+                    await newPage.close();
+
+                    await expect(actions.nth(2)).toHaveAccessibleName('Post message');
+                    await actions.nth(2).click();
+                    await expect(iframe.getByTestId('ai-chat')).toBeVisible();
+                    await expect(iframe.getByTestId('ai-chat-message-user').first()).toHaveText(
+                        AI_PROMPT
+                    );
+
+                    await expect(actions.nth(3)).toHaveAccessibleName('Close');
+                    await actions.nth(3).click();
+                    await expect(page.locator('#gitbook-widget-window')).not.toBeVisible();
+                    await page.locator('#gitbook-widget-button').click();
+                    await iframe.owner().evaluate(overrideAIResponse);
+                },
+            },
+            {
+                name: 'Configuration - Custom tools',
+                url: '',
+                run: async (page) => {
+                    await page.evaluate(() => {
+                        const GitBook = window.GitBook as unknown as GitBookStandalone;
+                        GitBook('configure', {
+                            tools: [
+                                {
+                                    name: 'contact_support',
+                                    description: 'Contact support on behalf of the user',
+                                    execute: async () => {
+                                        return {
+                                            output: { message: 'Support message' },
+                                            summary: {
+                                                text: 'Contacted support',
+                                                icon: 'circle-question',
+                                            },
+                                        };
+                                    },
+                                    confirmation: {
+                                        icon: 'circle-question',
+                                        label: 'Contact support',
+                                    },
+                                },
+                            ],
+                        });
+                        GitBook(
+                            'postUserMessage',
+                            'I want to contact support. Call the tool directly without a preamble. Do not respond with anything else.'
+                        );
+                    });
+                    const iframe = page.frameLocator('#gitbook-widget-iframe');
+                    await expect(iframe.getByTestId('ai-chat-message-user').first()).toHaveText(
+                        'I want to contact support. Call the tool directly without a preamble. Do not respond with anything else.'
+                    );
+                    const toolConfirmation = iframe
+                        .getByTestId('ai-chat-tool-confirmation')
+                        .first();
+                    await waitForEmbedVisible(iframe, toolConfirmation, 30_000);
+                    await page.waitForTimeout(10000);
+                    await iframe.owner().evaluate(overrideAIResponse);
+                },
+            },
+        ],
+    },
+    {
+        name: 'Docs Embed - Docs Only',
+        contentBaseURL: 'https://gitbook.gitbook.io/test-gitbook-open/~gitbook/embed/demo/',
+        tests: [
+            {
+                name: 'Docs only',
+                url: '',
+                run: async (page) => {
+                    await expect(page.locator('#gitbook-widget-window')).toBeVisible();
+                    const iframe = page.frameLocator('#gitbook-widget-iframe');
+                    await waitForEmbedVisible(iframe, iframe.getByTestId('embed-docs-page'));
+                },
+            },
+            {
+                name: 'Table of contents',
+                url: '',
+                run: async (page) => {
+                    await expect(page.locator('#gitbook-widget-window')).toBeVisible();
+                    const iframe = page.frameLocator('#gitbook-widget-iframe');
+                    await waitForEmbedVisible(iframe, iframe.getByTestId('embed-docs-page'));
+                    const tocButton = iframe.getByTestId('toc-button');
+                    await expect(tocButton).toBeVisible();
+                    await tocButton.click();
+                    await expect(iframe.getByTestId('table-of-contents')).toBeVisible();
+                },
+            },
+            {
+                name: 'Open in new tab',
+                url: '',
+                run: async (page) => {
+                    await expect(page.locator('#gitbook-widget-window')).toBeVisible();
+                    const iframe = page.frameLocator('#gitbook-widget-iframe');
+                    await waitForEmbedVisible(iframe, iframe.getByTestId('embed-docs-page'));
+                    const openInNewTabButton = iframe.getByTestId(
+                        'embed-docs-page-open-in-new-tab'
+                    );
+                    await expect(openInNewTabButton).toBeVisible();
+                    // Intercept the new page event without navigating
+                    const [newPage] = await Promise.all([
+                        page.context().waitForEvent('page', { timeout: 5000 }),
+                        openInNewTabButton.click(),
+                    ]);
+                    // Verify the new page would have opened with the expected URL
+                    expect(newPage.url()).toContain('gitbook.gitbook.io');
+                    // Close it immediately to avoid navigation
+                    await newPage.close();
+                },
+            },
+        ],
+    },
 ];
 
 runTestCases(testCases);
diff --git a/packages/gitbook/e2e/util.ts b/packages/gitbook/e2e/util.ts
index 6b62dea8c0..706b66c19f 100644
--- a/packages/gitbook/e2e/util.ts
+++ b/packages/gitbook/e2e/util.ts
@@ -454,7 +454,8 @@ export async function waitForIcons(page: Page) {
  */
 async function waitForTOCScrolling(page: Page) {
     const viewport = await page.viewportSize();
-    if (viewport && viewport.width >= 1024) {
+    if (viewport && viewport.width >= 1024 && !page.url().includes('~gitbook/embed/demo')) {
+        // The embed demo is an iframe, which means the viewport only a fraction of the main document. So there is no open TOC to scroll to.
         const toc = page.getByTestId('table-of-contents');
         await expect(toc).toBeVisible();
         await page.evaluate(() => {
diff --git a/packages/gitbook/src/components/AI/server-actions/AIToolCallsSummary.tsx b/packages/gitbook/src/components/AI/server-actions/AIToolCallsSummary.tsx
index 731e6480ff..b91a421817 100644
--- a/packages/gitbook/src/components/AI/server-actions/AIToolCallsSummary.tsx
+++ b/packages/gitbook/src/components/AI/server-actions/AIToolCallsSummary.tsx
@@ -39,7 +39,10 @@ function ToolCallSummary(props: { toolCall: AIToolCall; context: GitBookSiteCont
     const { toolCall, context } = props;
 
     return (
-        <div className="mt-2 flex origin-top-left animate-blur-in-slow items-start gap-2 text-sm text-tint-subtle">
+        <div
+            data-testid="ai-chat-tool-summary"
+            className="mt-2 flex origin-top-left animate-blur-in-slow items-start gap-2 text-sm text-tint-subtle"
+        >
             <Icon
                 icon={getIconForToolCall(toolCall)}
                 className="mt-1 size-3 shrink-0 text-tint-subtle/8"
diff --git a/packages/gitbook/src/components/AIChat/AIChat.tsx b/packages/gitbook/src/components/AIChat/AIChat.tsx
index 6bc5304811..cbe151841d 100644
--- a/packages/gitbook/src/components/AIChat/AIChat.tsx
+++ b/packages/gitbook/src/components/AIChat/AIChat.tsx
@@ -70,7 +70,6 @@ export function AIChat() {
 
     return (
         <div
-            data-testid="ai-chat"
             className={tcls(
                 'ai-chat inset-y-0 right-0 z-40 mx-auto flex max-w-3xl scroll-mt-36 px-4 py-4 transition-[width,opacity,margin,display] transition-discrete duration-300 sm:px-6 lg:fixed lg:w-80 lg:p-0 xl:w-96',
                 chat.opened
@@ -79,7 +78,7 @@ export function AIChat() {
             )}
         >
             <EmbeddableFrame className="relative shrink-0 border-tint-subtle border-l to-tint-base transition-all duration-300 max-lg:circular-corners:rounded-3xl max-lg:rounded-corners:rounded-md max-lg:border lg:w-80 xl:w-96">
-                <EmbeddableFrameMain>
+                <EmbeddableFrameMain data-testid="ai-chat">
                     <EmbeddableFrameHeader>
                         <AIChatDynamicIcon trademark={config.trademark} />
                         <EmbeddableFrameHeaderMain>
@@ -237,13 +236,14 @@ export function AIChatBody(props: {
                                 <h5
                                     className="animate-blur-in-slow font-bold text-lg text-tint-strong [@container(min-height:400px)]:text-center"
                                     style={{ animationDelay: '.5s' }}
-                                    data-testid="ai-chat-time-greeting"
+                                    data-testid="ai-chat-greeting-title"
                                 >
                                     {timeGreeting}
                                 </h5>
                                 <p
                                     className="animate-blur-in-slow text-tint [@container(min-height:400px)]:text-center"
                                     style={{ animationDelay: '.6s' }}
+                                    data-testid="ai-chat-greeting-subtitle"
                                 >
                                     {t(language, 'ai_chat_assistant_description')}
                                 </p>
diff --git a/packages/gitbook/src/components/AIChat/AIChatSuggestedQuestions.tsx b/packages/gitbook/src/components/AIChat/AIChatSuggestedQuestions.tsx
index cbc31bff64..afd603b18e 100644
--- a/packages/gitbook/src/components/AIChat/AIChatSuggestedQuestions.tsx
+++ b/packages/gitbook/src/components/AIChat/AIChatSuggestedQuestions.tsx
@@ -17,9 +17,13 @@ export default function AIChatSuggestedQuestions(props: {
     } = props;
 
     return (
-        <div className="flex flex-col items-start gap-2 self-start">
+        <div
+            className="flex flex-col items-start gap-2 self-start"
+            data-testid="ai-chat-suggested-questions"
+        >
             {suggestions.map((question, index) => (
                 <Button
+                    data-testid="ai-chat-suggested-question"
                     key={question}
                     variant="blank"
                     size="medium"
diff --git a/packages/gitbook/src/components/AIChat/AIChatToolConfirmations.tsx b/packages/gitbook/src/components/AIChat/AIChatToolConfirmations.tsx
index c0aeb371d6..292830f37a 100644
--- a/packages/gitbook/src/components/AIChat/AIChatToolConfirmations.tsx
+++ b/packages/gitbook/src/components/AIChat/AIChatToolConfirmations.tsx
@@ -37,6 +37,7 @@ export function AIChatToolConfirmations(props: {
                     style={{ animationDelay: `${0.5 + index * 0.1}s` }}
                 >
                     <Button
+                        data-testid="ai-chat-tool-confirmation"
                         onClick={() => {
                             tool.confirm();
                         }}
diff --git a/packages/gitbook/src/components/Embeddable/EmbeddableAIChat.tsx b/packages/gitbook/src/components/Embeddable/EmbeddableAIChat.tsx
index 09fde03a02..6bf601bd0a 100644
--- a/packages/gitbook/src/components/Embeddable/EmbeddableAIChat.tsx
+++ b/packages/gitbook/src/components/Embeddable/EmbeddableAIChat.tsx
@@ -75,7 +75,7 @@ export function EmbeddableAIChat(props: EmbeddableAIChatProps) {
                 />
                 <EmbeddableIframeButtons />
             </EmbeddableFrameSidebar>
-            <EmbeddableFrameMain>
+            <EmbeddableFrameMain data-testid="ai-chat">
                 <EmbeddableFrameHeader>
                     {!tabsRef.current ? (
                         <AIChatDynamicIcon
diff --git a/packages/gitbook/src/components/Embeddable/EmbeddableDocsPage.tsx b/packages/gitbook/src/components/Embeddable/EmbeddableDocsPage.tsx
index 08ee0b12b9..18e9d09556 100644
--- a/packages/gitbook/src/components/Embeddable/EmbeddableDocsPage.tsx
+++ b/packages/gitbook/src/components/Embeddable/EmbeddableDocsPage.tsx
@@ -47,7 +47,7 @@ export async function EmbeddableDocsPage(props: EmbeddableDocsPageProps) {
                 />
                 <EmbeddableIframeButtons />
             </EmbeddableFrameSidebar>
-            <EmbeddableFrameMain>
+            <EmbeddableFrameMain data-testid="embed-docs-page">
                 <div className="relative flex not-hydrated:animate-blur-in-slow flex-col">
                     <EmbeddableFrameHeader>
                         <HeaderMobileMenu className="-ml-2 page-no-toc:hidden" />
diff --git a/packages/gitbook/src/components/Embeddable/EmbeddableDocsPageControlButtons.tsx b/packages/gitbook/src/components/Embeddable/EmbeddableDocsPageControlButtons.tsx
index 09532b90ce..722b5018bd 100644
--- a/packages/gitbook/src/components/Embeddable/EmbeddableDocsPageControlButtons.tsx
+++ b/packages/gitbook/src/components/Embeddable/EmbeddableDocsPageControlButtons.tsx
@@ -10,6 +10,7 @@ export function EmbeddableDocsPageControlButtons(props: { href: string }) {
     return (
         <Button
             icon="arrow-up-right-from-square"
+            data-testid="embed-docs-page-open-in-new-tab"
             label={tString(language, 'open_in_new_tab')}
             href={href}
             target="_blank"
diff --git a/packages/gitbook/src/components/Embeddable/EmbeddableFrame.tsx b/packages/gitbook/src/components/Embeddable/EmbeddableFrame.tsx
index 0b312cf711..3dd1d63a73 100644
--- a/packages/gitbook/src/components/Embeddable/EmbeddableFrame.tsx
+++ b/packages/gitbook/src/components/Embeddable/EmbeddableFrame.tsx
@@ -28,10 +28,14 @@ export const EmbeddableFrame = React.forwardRef<HTMLDivElement, EmbeddableFrameP
     }
 );
 
-export function EmbeddableFrameMain(props: { children: React.ReactNode }) {
-    const { children } = props;
+export function EmbeddableFrameMain(props: React.ComponentProps<'div'>) {
+    const { children, ...rest } = props;
 
-    return <div className="flex flex-1 flex-col overflow-hidden">{children}</div>;
+    return (
+        <div className="flex flex-1 flex-col overflow-hidden" {...rest}>
+            {children}
+        </div>
+    );
 }
 
 export function EmbeddableFrameHeader(props: {
diff --git a/packages/gitbook/src/components/Embeddable/EmbeddableIframeAPI.tsx b/packages/gitbook/src/components/Embeddable/EmbeddableIframeAPI.tsx
index 8bd22b26aa..6f5c8e6c53 100644
--- a/packages/gitbook/src/components/Embeddable/EmbeddableIframeAPI.tsx
+++ b/packages/gitbook/src/components/Embeddable/EmbeddableIframeAPI.tsx
@@ -112,6 +112,7 @@ export function EmbeddableIframeButtons() {
             )}
             {actions.map((action, index) => (
                 <Button
+                    data-testid="embed-action"
                     key={action.label}
                     size="default"
                     variant="blank"
diff --git a/packages/gitbook/src/components/Header/HeaderMobileMenu.tsx b/packages/gitbook/src/components/Header/HeaderMobileMenu.tsx
index a7d0fcf0e0..dea4ff836e 100644
--- a/packages/gitbook/src/components/Header/HeaderMobileMenu.tsx
+++ b/packages/gitbook/src/components/Header/HeaderMobileMenu.tsx
@@ -45,6 +45,7 @@ export function HeaderMobileMenu(props: ButtonProps) {
 
     return (
         <Button
+            data-testid="toc-button"
             icon="bars"
             iconOnly
             variant="blank"