[hir] Add invariant to check if useMemo callback accepts args

compiler/forget/src/Inference/InlineUseMemo.ts

@@ -100,6 +100,14 @@ export function inlineUseMemo(fn: HIRFunction): void {
                 lambda.loc
               );
             }
+
+            if (body.loweredFunc.params.length > 0) {
+              CompilerError.invariant(
+                "Did not expect any arguments to useMemo callback",
+                body.loc
+              );
+            }
+
             // We know this function is used for useMemo and can prune it later
             useMemoFunctions.add(lambda.identifier.id);
 

compiler/forget/src/__tests__/fixtures/compiler/error.useMemo-callback-args.expect.md

@@ -0,0 +1,20 @@
+
+## Input
+
+```javascript
+// @inlineUseMemo
+function component(a, b) {
+  let x = useMemo((c) => a, []);
+  return x;
+}
+
+```
+
+
+## Error
+
+```
+[ReactForget] Invariant: Did not expect any arguments to useMemo callback (3:3)
+```
+          
+      

compiler/forget/src/__tests__/fixtures/compiler/error.useMemo-callback-args.js

@@ -0,0 +1,5 @@
+// @inlineUseMemo
+function component(a, b) {
+  let x = useMemo((c) => a, []);
+  return x;
+}