fix(valid-expect-in-promise): allow variables assigned awaited promises

Author: G-Rath

src/rules/__tests__/valid-expect-in-promise.test.ts

@@ -272,6 +272,44 @@ ruleTester.run('valid-expect-in-promise', rule, {
         return promise;
       });
     `,
+    dedent`
+      it('is a test', async () => {
+        const value = await somePromise().then(response => {
+          expect(response).toHaveProperty('data');
+
+          return response.data;
+        });
+
+        expect(value).toBe('hello world');
+      });
+    `,
+    dedent`
+      it('is a test', async () => {
+        return await somePromise().then(response => {
+          expect(response).toHaveProperty('data');
+
+          return response.data;
+        });
+      });
+    `,
+    dedent`
+      it('is a test', async () => {
+        return somePromise().then(response => {
+          expect(response).toHaveProperty('data');
+
+          return response.data;
+        });
+      });
+    `,
+    dedent`
+      it('is a test', async () => {
+        await somePromise().then(response => {
+          expect(response).toHaveProperty('data');
+
+          return response.data;
+        });
+      });
+    `,
     dedent`
       it(
         'test function',

src/rules/valid-expect-in-promise.ts

@@ -138,6 +138,10 @@ const isVariableAwaitedOrReturned = (
   const [variable] = variables.declarations;
   const name = getVariableName(variable);
 
+  if (variable.init?.type === AST_NODE_TYPES.AwaitExpression) {
+    return true;
+  }
+
   // null means that the variable is destructured, which is pretty much impossible
   // for us to track, so we return true to bailout gracefully
   if (name === null) {