AFFiNE-Mirror/tests/affine-cloud-copilot/e2e/chat-with/network.spec.ts

import { test } from '../base/base-test';

test.describe('AIChatWith/Network', () => {
  test.beforeEach(async ({ loggedInPage: page, utils }) => {
    await utils.testUtils.setupTestEnvironment(page);
    await utils.chatPanel.openChatPanel(page);
  });

  test('should support chat with network if network search enabled', async ({
    loggedInPage: page,
    utils,
  }) => {
    await utils.chatPanel.enableNetworkSearch(page);
    await utils.chatPanel.makeChat(
      page,
      'What is the weather like in Shanghai today?'
    );
    await utils.chatPanel.waitForHistory(page, [
      {
        role: 'user',
        content: 'What is the weather like in Shanghai today?',
      },
      {
        role: 'assistant',
        status: 'success',
      },
    ]);
  });
});