test(no-await-sync-query): use more realistic scenarios

Author: Belco90

tests/lib/rules/no-await-sync-query.test.ts

@@ -13,15 +13,22 @@ ruleTester.run(RULE_NAME, rule, {
     // sync queries without await are valid
     ...SYNC_QUERIES_COMBINATIONS.map((query) => ({
       code: `() => {
-        ${query}('foo')
+        const element = ${query}('foo')
+      }
+      `,
+    })),
+    // sync queries without await inside assert are valid
+    ...SYNC_QUERIES_COMBINATIONS.map((query) => ({
+      code: `() => {
+        expect(${query}('foo')).toBeEnabled()
       }
       `,
     })),
 
     // async queries with await operator are valid
     ...ASYNC_QUERIES_COMBINATIONS.map((query) => ({
       code: `async () => {
-        await ${query}('foo')
+        const element = await ${query}('foo')
       }
       `,
     })),
@@ -40,29 +47,58 @@ ruleTester.run(RULE_NAME, rule, {
     // sync queries with await operator are not valid
     ...SYNC_QUERIES_COMBINATIONS.map((query) => ({
       code: `async () => {
-        await ${query}('foo')
+        const element = await ${query}('foo')
+      }
+      `,
+      errors: [
+        {
+          messageId: 'noAwaitSyncQuery',
+          line: 2,
+          column: 31,
+        },
+      ],
+    })),
+    // sync queries with await operator inside assert are not valid
+    ...SYNC_QUERIES_COMBINATIONS.map((query) => ({
+      code: `async () => {
+        expect(await ${query}('foo')).toBeEnabled()
       }
       `,
       errors: [
         {
           messageId: 'noAwaitSyncQuery',
           line: 2,
-          column: 15,
+          column: 22,
         },
       ],
     })),
 
     // sync queries in screen with await operator are not valid
     ...SYNC_QUERIES_COMBINATIONS.map((query) => ({
       code: `async () => {
-        await screen.${query}('foo')
+        const element = await screen.${query}('foo')
       }
       `,
       errors: [
         {
           messageId: 'noAwaitSyncQuery',
           line: 2,
-          column: 22,
+          column: 38,
+        },
+      ],
+    })),
+
+    // sync queries in screen with await operator inside assert are not valid
+    ...SYNC_QUERIES_COMBINATIONS.map((query) => ({
+      code: `async () => {
+        expect(await screen.${query}('foo')).toBeEnabled()
+      }
+      `,
+      errors: [
+        {
+          messageId: 'noAwaitSyncQuery',
+          line: 2,
+          column: 29,
         },
       ],
     })),