fix e2e test

jichang · Apr 26, 2024 · 9be3389 · 9be3389
1 parent 7922472
commit 9be3389
Showing 1 changed file with 15 additions and 3 deletions.
diff --git a/packages/react/tests/view.spec.ts b/packages/react/tests/view.spec.ts
@@ -17,7 +17,11 @@ test('view document', async ({ page }) => {
     )
     .click();
   await page.getByTestId('pdf__plugin__metadata__content');
-  await page.getByTestId('pdf__ui__dialog__close__btn').click();
+  await page
+    .getByTestId(
+      `pdf__toolbar__item__plugin__${PdfApplicatinPluginKey.Metadata}`,
+    )
+    .click();
 
   await page
     .getByTestId(
@@ -49,13 +53,21 @@ test('view document', async ({ page }) => {
     )
     .click();
   await page.getByTestId('pdf__plugin__attachments__content');
-  await page.getByTestId('pdf__ui__dialog__close__btn').click();
+  await page
+    .getByTestId(
+      `pdf__toolbar__item__plugin__${PdfApplicatinPluginKey.Attachments}`,
+    )
+    .click();
 
   await page
     .getByTestId(
       `pdf__toolbar__item__plugin__${PdfApplicatinPluginKey.Signatures}`,
     )
     .click();
   await page.getByTestId('pdf__plugin__signatures__content');
-  await page.getByTestId('pdf__ui__dialog__close__btn').click();
+  await page
+    .getByTestId(
+      `pdf__toolbar__item__plugin__${PdfApplicatinPluginKey.Signatures}`,
+    )
+    .click();
 });