test: prefer-query-by-disappearance

Author: PrashantAshok

tests/lib/rules/prefer-query-by-disappearance.test.ts

@@ -15,6 +15,14 @@ ruleTester.run(RULE_NAME, rule, {
         await waitForElementToBeRemoved(button)
       `,
     },
+    {
+      code: `
+        import { screen } from '@marko/testing-library';
+
+        const button = screen.getByRole('button')
+        await waitForElementToBeRemoved(button)
+      `,
+    },
     {
       code: `
         import { screen } from '@testing-library/react';
@@ -221,6 +229,20 @@ ruleTester.run(RULE_NAME, rule, {
         },
       ],
     },
+    {
+      code: `
+        import { screen, waitForElementToBeRemoved } from '@marko/testing-library';
+
+        await waitForElementToBeRemoved(() => screen.getByText("hello"))
+      `,
+      errors: [
+        {
+          messageId: 'preferQueryByDisappearance',
+          line: 4,
+          column: 41,
+        },
+      ],
+    },
     {
       code: `
         import { screen, waitForElementToBeRemoved } from '@testing-library/react';