Extracted IncorrectModuleRegistryCallArityParserError as a seperate function in error-utils.js

packages/react-native-codegen/src/parsers/__tests__/error-utils-test.js

@@ -0,0 +1,51 @@
+/**
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ *
+ * @format
+ * @flow strict-local
+ * @oncall react_native
+ */
+
+'use strict';
+
+const {throwIfWrongNumberOfCallExpressionArgs} = require('../error-utils');
+const {IncorrectModuleRegistryCallArityParserError} = require('../errors');
+
+describe('throwErrorIfWrongNumberOfCallExpressionArgs', () => {
+  it('throw error if wrong number of call expression args is used', () => {
+    const nativeModuleName = 'moduleName';
+    const flowCallExpression = {argument: []};
+    const methodName = 'methodName';
+    const numberOfCallExpressionArgs = flowCallExpression.argument.length;
+    const language = 'Flow';
+    expect(() => {
+      throwIfWrongNumberOfCallExpressionArgs(
+        nativeModuleName,
+        flowCallExpression,
+        methodName,
+        numberOfCallExpressionArgs,
+        language,
+      );
+    }).toThrow(IncorrectModuleRegistryCallArityParserError);
+  });
+
+  it("don't throw error if correct number of call expression args is used", () => {
+    const nativeModuleName = 'moduleName';
+    const flowCallExpression = {argument: ['argument']};
+    const methodName = 'methodName';
+    const numberOfCallExpressionArgs = flowCallExpression.argument.length;
+    const language = 'Flow';
+    expect(() => {
+      throwIfWrongNumberOfCallExpressionArgs(
+        nativeModuleName,
+        flowCallExpression,
+        methodName,
+        numberOfCallExpressionArgs,
+        language,
+      );
+    }).not.toThrow(IncorrectModuleRegistryCallArityParserError);
+  });
+});

packages/react-native-codegen/src/parsers/error-utils.js

@@ -0,0 +1,37 @@
+/**
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ *
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ *
+ * @flow strict-local
+ * @format
+ */
+
+('use strict');
+
+import type {ParserType} from './errors';
+
+const {IncorrectModuleRegistryCallArityParserError} = require('./errors');
+
+function throwIfWrongNumberOfCallExpressionArgs(
+  nativeModuleName: string,
+  flowCallExpression: $FlowFixMe,
+  methodName: string,
+  numberOfCallExpressionArgs: number,
+  language: ParserType,
+) {
+  if (numberOfCallExpressionArgs !== 1) {
+    throw new IncorrectModuleRegistryCallArityParserError(
+      nativeModuleName,
+      flowCallExpression,
+      methodName,
+      numberOfCallExpressionArgs,
+      language,
+    );
+  }
+}
+
+module.exports = {
+  throwIfWrongNumberOfCallExpressionArgs,
+};

packages/react-native-codegen/src/parsers/flow/modules/index.js

@@ -65,10 +65,11 @@ const {
   MoreThanOneModuleRegistryCallsParserError,
   UntypedModuleRegistryCallParserError,
   IncorrectModuleRegistryCallTypeParameterParserError,
-  IncorrectModuleRegistryCallArityParserError,
   IncorrectModuleRegistryCallArgumentTypeParserError,
 } = require('../../errors.js');
 
+const {throwIfWrongNumberOfCallExpressionArgs} = require('../../error-utils');
+
 const language = 'Flow';
 
 function nullGuard<T>(fn: () => T): ?T {
@@ -641,15 +642,13 @@ function buildModuleSchema(
     const {typeArguments} = callExpression;
     const methodName = callExpression.callee.property.name;
 
-    if (callExpression.arguments.length !== 1) {
-      throw new IncorrectModuleRegistryCallArityParserError(
-        hasteModuleName,
-        callExpression,
-        methodName,
-        callExpression.arguments.length,
-        language,
-      );
-    }
+    throwIfWrongNumberOfCallExpressionArgs(
+      hasteModuleName,
+      callExpression,
+      methodName,
+      callExpression.arguments.length,
+      language,
+    );
 
     if (callExpression.arguments[0].type !== 'Literal') {
       const {type} = callExpression.arguments[0];

packages/react-native-codegen/src/parsers/typescript/modules/index.js

@@ -65,10 +65,11 @@ const {
   MoreThanOneModuleRegistryCallsParserError,
   UntypedModuleRegistryCallParserError,
   IncorrectModuleRegistryCallTypeParameterParserError,
-  IncorrectModuleRegistryCallArityParserError,
   IncorrectModuleRegistryCallArgumentTypeParserError,
 } = require('../../errors.js');
 
+const {throwIfWrongNumberOfCallExpressionArgs} = require('../../error-utils');
+
 const language = 'TypeScript';
 
 function nullGuard<T>(fn: () => T): ?T {
@@ -675,15 +676,13 @@ function buildModuleSchema(
     const {typeParameters} = callExpression;
     const methodName = callExpression.callee.property.name;
 
-    if (callExpression.arguments.length !== 1) {
-      throw new IncorrectModuleRegistryCallArityParserError(
-        hasteModuleName,
-        callExpression,
-        methodName,
-        callExpression.arguments.length,
-        language,
-      );
-    }
+    throwIfWrongNumberOfCallExpressionArgs(
+      hasteModuleName,
+      callExpression,
+      methodName,
+      callExpression.arguments.length,
+      language,
+    );
 
     if (callExpression.arguments[0].type !== 'StringLiteral') {
       const {type} = callExpression.arguments[0];