fix tests

Perryvw · Perryvw · commit 83637a3311b1 · 2024-09-15T21:34:23.000+02:00
diff --git a/test/unit/functions/validation/functionPermutations.ts b/test/unit/functions/validation/functionPermutations.ts
@@ -351,13 +351,18 @@ const noSelfInFileTestFunctions: TestFunction[] = [
     },
 ];
 
-export const anonTestFunctionExpressions: TestFunction[] = [
-    { value: "s => s" },
-    { value: "(s => s)" },
+export const anonArrowFunctionExpressions: TestFunction[] = [{ value: "s => s" }, { value: "(s => s)" }];
+
+export const anonNonArrowFunctionExpressions: TestFunction[] = [
     { value: "function(s) { return s; }" },
     { value: "(function(s) { return s; })" },
 ];
 
+export const anonTestFunctionExpressions: TestFunction[] = [
+    ...anonArrowFunctionExpressions,
+    ...anonNonArrowFunctionExpressions,
+];
+
 export const selfTestFunctionExpressions: TestFunction[] = [
     { value: "function(this: any, s) { return s; }" },
     { value: "(function(this: any, s) { return s; })" },
diff --git a/test/unit/functions/validation/validFunctionAssignments.spec.ts b/test/unit/functions/validation/validFunctionAssignments.spec.ts
@@ -1,6 +1,7 @@
 import * as util from "../../../util";
 import {
-    anonTestFunctionExpressions,
+    anonArrowFunctionExpressions,
+    anonNonArrowFunctionExpressions,
     anonTestFunctionType,
     noSelfTestFunctionExpressions,
     noSelfTestFunctions,
@@ -133,10 +134,10 @@ test.each([
 });
 
 test.each([
-    ...anonTestFunctionExpressions.map((f): [TestFunction, string[]] => [f, ["0", "'foobar'"]]),
-    ...selfTestFunctionExpressions.map((f): [TestFunction, string[]] => [f, ["0", "'foobar'"]]),
+    ...anonNonArrowFunctionExpressions.map((f): [TestFunction, string[]] => [f, ["'context'", "'foobar'"]]),
+    ...selfTestFunctionExpressions.map((f): [TestFunction, string[]] => [f, ["'context'", "'foobar'"]]),
     ...noSelfTestFunctionExpressions.map((f): [TestFunction, string[]] => [f, ["'foobar'"]]),
-])("Valid function expression argument with no signature (%p, %p)", (testFunction, args) => {
+])("Valid function expression argument with no signature have context (%p, %p)", (testFunction, args) => {
     util.testFunction`
         const takesFunction: any = (fn: (this: void, ...args: any[]) => any, ...args: any[]) => {
             return fn(...args);
@@ -147,6 +148,20 @@ test.each([
         .expectToEqual("foobar");
 });
 
+test.each([...anonArrowFunctionExpressions.map((f): [TestFunction, string[]] => [f, ["'foobar'"]])])(
+    "Valid arrow function expression argument with no signature do not have context (%p, %p)",
+    (testFunction, args) => {
+        util.testFunction`
+        const takesFunction: any = (fn: (this: void, ...args: any[]) => any, ...args: any[]) => {
+            return fn(...args);
+        }
+        return takesFunction(${testFunction.value}, ${args.join(", ")});
+    `
+            .setTsHeader(testFunction.definition ?? "")
+            .expectToEqual("foobar");
+    }
+);
+
 test.each(validTestFunctionAssignments)("Valid function return (%p)", (testFunction, functionType) => {
     util.testFunction`
         function returnsFunction(): ${functionType} {
