fix(prefer-expect-assertions): report expects in loops in functions in tests

Author: G-Rath

src/rules/__tests__/prefer-expect-assertions.test.ts

@@ -374,20 +374,6 @@ ruleTester.run('prefer-expect-assertions (loops)', rule, {
       `,
       options: [{ onlyFunctionsWithExpectInLoop: true }],
     },
-    {
-      code: dedent`
-        it('returns numbers that are greater than two', function () {
-          const expectNumbersToBeGreaterThan = (numbers, value) => {
-            for (let number of numbers) {
-              expect(number).toBeGreaterThan(value);
-            }
-          };
-
-          expectNumbersToBeGreaterThan(getNumbers(), 2);
-        });
-      `,
-      options: [{ onlyFunctionsWithExpectInLoop: true }],
-    },
     {
       code: dedent`
         it("returns numbers that are greater than five", function () {
@@ -431,6 +417,27 @@ ruleTester.run('prefer-expect-assertions (loops)', rule, {
         },
       ],
     },
+    {
+      code: dedent`
+        it('returns numbers that are greater than two', function () {
+          const expectNumbersToBeGreaterThan = (numbers, value) => {
+            for (let number of numbers) {
+              expect(number).toBeGreaterThan(value);
+            }
+          };
+
+          expectNumbersToBeGreaterThan(getNumbers(), 2);
+        });
+      `,
+      options: [{ onlyFunctionsWithExpectInLoop: true }],
+      errors: [
+        {
+          messageId: 'haveExpectAssertions',
+          column: 1,
+          line: 1,
+        },
+      ],
+    },
     {
       code: dedent`
         it("only returns numbers that are greater than seven", function () {

src/rules/prefer-expect-assertions.ts

@@ -156,7 +156,7 @@ export default createRule<[RuleOptions], MessageIds>({
           return;
         }
 
-        if (isExpectCall(node) && expressionDepth === 1 && inForLoop) {
+        if (isExpectCall(node) && inTestCaseCall && inForLoop) {
           hasExpectInLoop = true;
         }
       },