test: test fallback to custom codecs when arrow function inlining fails

Check if parseFunctionBody() succeeded before returning, allowing parser
to use custom codec definitions when function contains parameter references.

Author: yanxue-22

packages/openapi-generator/test/arrowFunctionFallback.test.ts

@@ -0,0 +1,77 @@
+const E = require('fp-ts/lib/Either');
+const assert = require('node:assert/strict');
+const test = require('node:test');
+const { TestProject } = require('./testProject');
+const { parsePlainInitializer } = require('../src');
+
+test('arrow function with parameters should use custom codec definition', async () => {
+  const files = {
+    '/api/v2/admin/migrateOrgShardKey.ts': `
+import * as t from 'io-ts';
+import { NonEmptyString } from './types';
+import { arrayFromArrayOrSingle } from '../../../utils';
+
+export const UpdateShardKeyRequestBody = {
+  shardKey: t.string,
+  collectionType: t.union([
+    t.literal('user'),
+    t.literal('enterprise'),
+    t.literal('enterpriseTemplate')
+  ]),
+  ids: arrayFromArrayOrSingle(NonEmptyString),
+  enterpriseUpdateUsers: t.boolean,
+} as const;
+`,
+    '/api/v2/admin/types.ts': `
+import * as t from 'io-ts';
+
+export const NonEmptyString = t.string;
+`,
+    '/utils/arrayFromArrayOrSingle.ts': `
+import * as t from 'io-ts';
+import { arrayFromSingle } from '@bitgo/public-types';
+
+export const arrayFromArrayOrSingle = <C extends t.Mixed>(codec: C) =>
+  t.union([t.array(codec), arrayFromSingle(codec)]);
+`,
+  };
+
+  const knownImports = {
+    '.': {
+      arrayFromArrayOrSingle: (_: any, innerSchema: any) =>
+        E.right({ type: 'array', items: innerSchema }),
+    },
+  };
+
+  const project = new TestProject(files, knownImports);
+  await project.parseEntryPoint('/api/v2/admin/migrateOrgShardKey.ts');
+  const sourceFile = project.get('/api/v2/admin/migrateOrgShardKey.ts');
+
+  assert.ok(sourceFile !== undefined, 'Source file should exist');
+
+  const actual: any = {};
+  const errors: string[] = [];
+
+  for (const symbol of sourceFile.symbols.declarations) {
+    if (symbol.init !== undefined) {
+      const result = parsePlainInitializer(project, sourceFile, symbol.init);
+      if (E.isLeft(result)) {
+        errors.push(result.left);
+      } else {
+        actual[symbol.name] = result.right;
+      }
+    }
+  }
+
+  assert.deepEqual(errors, [], `Expected no errors, but got "${errors[0]}"`);
+
+  assert.ok(
+    actual.UpdateShardKeyRequestBody?.type === 'object',
+    'Should parse UpdateShardKeyRequestBody as object',
+  );
+
+  assert.ok(
+    actual.UpdateShardKeyRequestBody?.properties?.ids?.type === 'array',
+    'Should parse ids field as array (from custom codec definition)',
+  );
+});