From a7f167968f70b3d77c75bb49d063f6a9aebf2467 Mon Sep 17 00:00:00 2001
From: JimmFly <yangjinfei001@gmail.com>
Date: Tue, 20 Dec 2022 09:59:20 +0800
Subject: [PATCH] test: update test case for quick search

---
 tests/quick-search.spec.ts | 44 ++++++++++++++++++++++++++++++++++----
 1 file changed, 40 insertions(+), 4 deletions(-)

diff --git a/tests/quick-search.spec.ts b/tests/quick-search.spec.ts
index 5aff46f67c..61e694ce1a 100644
--- a/tests/quick-search.spec.ts
+++ b/tests/quick-search.spec.ts
@@ -15,7 +15,7 @@ async function assertTitleTexts(
   texts: string[],
   option?: { delay: number }
 ) {
-  await page.mouse.move(100, 100); // move mouse for focus
+  await page.mouse.move(100, 100);
   const actual = await page
     .locator('[class=affine-default-page-block-title]')
     .allInnerTexts();
@@ -23,6 +23,17 @@ async function assertTitleTexts(
     expect(actual).toEqual(texts);
   }, option?.delay);
 }
+async function assertResultList(
+  page: Page,
+  texts: string[],
+  option?: { delay: number }
+) {
+  await page.mouse.move(100, 100);
+  const actual = await page.locator('[cmdk-item]').allInnerTexts();
+  setTimeout(() => {
+    expect(actual).toEqual(texts);
+  }, option?.delay);
+}
 
 test.describe('Open quick search', () => {
   test('Click slider bar button', async ({ page }) => {
@@ -31,7 +42,7 @@ test.describe('Open quick search', () => {
     );
     await quickSearchButton.click();
     const quickSearch = page.locator('[data-testid=quickSearch]');
-    expect(quickSearch.isVisible()).toEqual(true);
+    await expect(quickSearch).toBeVisible();
   });
   test('Click arrowDown icon after title', async ({ page }) => {
     //header-quickSearchButton
@@ -40,12 +51,12 @@ test.describe('Open quick search', () => {
     );
     await quickSearchButton.click();
     const quickSearch = page.locator('[data-testid=quickSearch]');
-    expect(quickSearch.isVisible()).toEqual(true);
+    await expect(quickSearch).toBeVisible();
   });
   test('Press the shortcut key cmd+k', async ({ page }) => {
     await openQuickSearchByShortcut(page);
     const quickSearch = page.locator('[data-testid=quickSearch]');
-    expect(quickSearch.isVisible()).toEqual(true);
+    await expect(quickSearch).toBeVisible();
   });
 });
 
@@ -64,3 +75,28 @@ test.describe('Add new page in quick search', () => {
     await assertTitleTexts(page, ['test'], { delay: 50 });
   });
 });
+
+test.describe('Search and select', () => {
+  test('Search and get results', async ({ page }) => {
+    await openQuickSearchByShortcut(page);
+    await page.keyboard.insertText('Welcome');
+    await assertResultList(page, ['Welcome to the AFFiNE Alpha'], {
+      delay: 50,
+    });
+  });
+  test('Create a new page and search this page', async ({ page }) => {
+    await openQuickSearchByShortcut(page);
+    await page.keyboard.insertText('Welcome');
+    const addNewPage = page.locator('[data-testid=quickSearch-addNewPage]');
+    await addNewPage.click();
+    await page.waitForTimeout(500);
+    await openQuickSearchByShortcut(page);
+    await page.keyboard.insertText('Welcome');
+    await assertResultList(page, ['Welcome to the AFFiNE Alpha', 'Welcome']);
+    await page.keyboard.press('ArrowDown', { delay: 50 });
+    await page.keyboard.press('Enter', { delay: 50 });
+    await assertTitleTexts(page, ['Welcome'], {
+      delay: 50,
+    });
+  });
+});