fix: report query instead of var reference (#10)

Author: thomaslombart

lib/rules/await-async-query.js

@@ -62,10 +62,13 @@ module.exports = {
             });
           } else {
             references.forEach(reference => {
-              const node = reference.identifier;
-              if (!isAwaited(node.parent) && !isPromiseResolved(node)) {
+              const referenceNode = reference.identifier;
+              if (
+                !isAwaited(referenceNode.parent) &&
+                !isPromiseResolved(referenceNode)
+              ) {
                 context.report({
-                  node: reference.identifier,
+                  node,
                   messageId: 'awaitAsyncQuery',
                   data: {
                     name: node.name,

tests/lib/rules/await-async-query.js

@@ -122,16 +122,30 @@ ruleTester.run('await-async-query', rule, {
 
   invalid:
     // async queries without await operator or then method are not valid
-    ASYNC_QUERIES_COMBINATIONS.map(query => ({
-      code: `async () => {
+    [
+      ...ASYNC_QUERIES_COMBINATIONS.map(query => ({
+        code: `async () => {
         doSomething()
         const foo = ${query}('foo')
       }
       `,
-      errors: [
-        {
-          messageId: 'awaitAsyncQuery',
-        },
-      ],
-    })),
+        errors: [
+          {
+            messageId: 'awaitAsyncQuery',
+          },
+        ],
+      })),
+      ...ASYNC_QUERIES_COMBINATIONS.map(query => ({
+        code: `async () => {
+        const foo = ${query}('foo')
+        expect(foo).toBeInTheDocument()
+      }
+      `,
+        errors: [
+          {
+            message: `\`${query}\` must have \`await\` operator`,
+          },
+        ],
+      })),
+    ],
 });