test(core): chat with collection & tags (#11388)

### TL:DR

* AI chat with collection E2E
* AI chat with tag E2E

> Close BS-3007

tests/affine-cloud-copilot/e2e/utils/chat-panel-utils.ts

@@ -225,7 +225,6 @@ export class ChatPanelUtils {
       await expect(states.every(state => state === 'finished')).toBe(true);
     }).toPass({ timeout: 20000 });
 
-    await page.pause();
     await this.makeChat(page, text);
   }
 
@@ -249,6 +248,39 @@ export class ChatPanelUtils {
     await this.makeChat(page, text);
   }
 
+  public static async chatWithTags(page: Page, tags: string[]) {
+    for (const tag of tags) {
+      const withButton = await page.getByTestId('chat-panel-with-button');
+      await withButton.click();
+      const withMenu = await page.getByTestId('ai-add-popover');
+      await withMenu.getByTestId('ai-chat-with-tags').click();
+      await withMenu.getByText(tag).click();
+      await page.getByTestId('chat-panel-chips').getByText(tag);
+    }
+    await this.waitForEmbeddingProgress(page);
+  }
+
+  public static async chatWithCollections(page: Page, collections: string[]) {
+    for (const collection of collections) {
+      const withButton = await page.getByTestId('chat-panel-with-button');
+      await withButton.click();
+      const withMenu = await page.getByTestId('ai-add-popover');
+      await withMenu.getByTestId('ai-chat-with-collections').click();
+      await withMenu.getByText(collection).click();
+      await page.getByTestId('chat-panel-chips').getByText(collection);
+    }
+    await this.waitForEmbeddingProgress(page);
+  }
+
+  public static async waitForEmbeddingProgress(page: Page) {
+    await page.getByTestId('chat-panel-embedding-progress').waitFor({
+      state: 'visible',
+    });
+    await page.getByTestId('chat-panel-embedding-progress').waitFor({
+      state: 'hidden',
+    });
+  }
+
   public static async enableNetworkSearch(page: Page) {
     const networkSearch = await page.getByTestId('chat-network-search');
     if ((await networkSearch.getAttribute('data-active')) === 'false') {

tests/affine-cloud-copilot/e2e/utils/editor-utils.ts

@@ -209,6 +209,63 @@ export class EditorUtils {
     );
   }
 
+  public static async createCollectionAndDoc(
+    page: Page,
+    collectionName: string,
+    docContent: string
+  ) {
+    // Create collection
+    await page.getByTestId('explorer-bar-add-collection-button').click();
+    const input = await page.getByTestId('prompt-modal-input');
+    await input.focus();
+    await input.pressSequentially(collectionName);
+    await page.getByTestId('prompt-modal-confirm').click();
+    const collections = await page.getByTestId('collapsible-section-content');
+    const collection = await collections
+      .getByText(collectionName)
+      .locator('..');
+
+    // Create doc
+    await collection.hover();
+    await collection.getByTestId('collection-add-doc-button').click();
+    await page.getByTestId('confirm-modal-confirm').click();
+    await this.focusToEditor(page);
+    const texts = docContent.split('\n');
+    for (const [index, line] of texts.entries()) {
+      await page.keyboard.insertText(line);
+      if (index !== texts.length - 1) {
+        await page.keyboard.press('Enter');
+      }
+    }
+  }
+
+  public static async createTagAndDoc(
+    page: Page,
+    tagName: string,
+    docContent: string
+  ) {
+    // Create tag
+    const tags = await page.getByTestId('explorer-tags');
+    await tags.getByTestId('explorer-bar-add-favorite-button').click();
+    const input = await page.getByTestId('rename-modal-input');
+    await input.focus();
+    await input.pressSequentially(tagName);
+    await input.press('Enter');
+    const tag = await tags.getByText(tagName).locator('..');
+
+    // Create doc
+    await tag.hover();
+    await tag.getByTestId('tag-add-doc-button').click();
+    await this.focusToEditor(page);
+    const texts = docContent.split('\n');
+    for (const [index, line] of texts.entries()) {
+      await page.keyboard.insertText(line);
+      if (index !== texts.length - 1) {
+        await page.keyboard.press('Enter');
+      }
+    }
+  }
+
   public static async selectElementInEdgeless(page: Page, elements: string[]) {
     await page.evaluate(
       ({ elements }) => {

tests/affine-cloud-copilot/e2e/utils/test-utils.ts

@@ -1,4 +1,7 @@
-import { createRandomAIUser } from '@affine-test/kit/utils/cloud';
+import {
+  createRandomAIUser,
+  enableCloudWorkspace,
+} from '@affine-test/kit/utils/cloud';
 import { openHomePage, setCoreUrl } from '@affine-test/kit/utils/load-page';
 import {
   clickNewPageButton,
@@ -52,10 +55,13 @@ export class TestUtils {
     };
   }
 
-  public async setupTestEnvironment(page: Page) {
+  public async setupTestEnvironment(page: Page, enableCloud: boolean = true) {
     await openHomePage(page);
     await clickNewPageButton(page);
     await waitForEditorLoad(page);
+    if (enableCloud) {
+      await enableCloudWorkspace(page);
+    }
   }
 
   public async createTestWorkspace(page: Page, name: string = 'test') {