diff --git a/tests/quick-search.spec.ts b/tests/quick-search.spec.ts index b45e66ff94..82e5a4db78 100644 --- a/tests/quick-search.spec.ts +++ b/tests/quick-search.spec.ts @@ -8,14 +8,19 @@ loadPage(); const openQuickSearchByShortcut = async (page: Page) => await withCtrlOrMeta(page, () => page.keyboard.press('k', { delay: 50 })); -async function assertTitleTexts(page: Page, texts: string) { - const actual = await page.evaluate(() => { - const titleElement = ( - document.querySelector('.affine-default-page-block-title') - ); - return titleElement.value; - }); - expect(actual).toEqual(texts); +async function assertTitleTexts(page: Page, texts: string, isSearch?: boolean) { + if (isSearch) { + const actual = await page.evaluate(() => { + const titleElement = ( + document.querySelector('.affine-default-page-block-title') + ); + return titleElement.value; + expect(actual).toEqual(texts); + }); + } else { + const actual = await page.title(); + expect(actual).toEqual(texts); + } } async function assertResultList(page: Page, texts: string[]) { const actual = await page.locator('[cmdk-item]').allInnerTexts(); @@ -52,14 +57,13 @@ test.describe('Open quick search', () => { }); test.describe('Add new page in quick search', () => { - // FIXME: not working test('Create a new page without keyword', async ({ page }) => { await newPage(page); await openQuickSearchByShortcut(page); const addNewPage = page.locator('[data-testid=quickSearch-addNewPage]'); await addNewPage.click(); - await page.waitForTimeout(200); - await assertTitleTexts(page, ''); + await page.waitForTimeout(300); + await assertTitleTexts(page, 'Untitled'); }); test('Create a new page with keyword', async ({ page }) => { @@ -68,7 +72,7 @@ test.describe('Add new page in quick search', () => { await page.keyboard.insertText('test123456'); const addNewPage = page.locator('[data-testid=quickSearch-addNewPage]'); await addNewPage.click(); - await page.waitForTimeout(200); + await page.waitForTimeout(300); await assertTitleTexts(page, 'test123456'); }); }); @@ -80,11 +84,14 @@ test.describe('Search and select', () => { await page.keyboard.insertText('test123456'); const addNewPage = page.locator('[data-testid=quickSearch-addNewPage]'); await addNewPage.click(); - await page.waitForTimeout(200); + await page.waitForTimeout(300); + await assertTitleTexts(page, 'test123456'); await openQuickSearchByShortcut(page); await page.keyboard.insertText('test123456'); + await page.waitForTimeout(300); await assertResultList(page, ['test123456']); - await page.keyboard.press('Enter', { delay: 50 }); - await assertTitleTexts(page, 'test123456'); + await page.keyboard.press('Enter'); + await page.waitForTimeout(300); + await assertTitleTexts(page, 'test123456', true); }); });