From 1dc35d1e9912db4b73ee7dae1071a92de6245b28 Mon Sep 17 00:00:00 2001
From: Cesar Parra <cesar.parra@gmail.com>
Date: Tue, 24 Jan 2023 06:44:36 -0400
Subject: [PATCH] Unit testing - support for inline reference overrides

---
 .../openapi/parsers/ReferenceBuilder.ts       |  4 ++
 .../__tests__/ReferenceBuilder.spec.ts        | 46 ++++++++++++++++++-
 2 files changed, 49 insertions(+), 1 deletion(-)

diff --git a/src/transpiler/openapi/parsers/ReferenceBuilder.ts b/src/transpiler/openapi/parsers/ReferenceBuilder.ts
index 3a0a716e..6b1c672c 100644
--- a/src/transpiler/openapi/parsers/ReferenceBuilder.ts
+++ b/src/transpiler/openapi/parsers/ReferenceBuilder.ts
@@ -20,6 +20,8 @@ type TypeBundleWithIsCollectionAndReferenceOverrides = TypeBundle & {
 export class ReferenceBuilder {
   build(referencedTypeName: string): Reference {
     const originalTypeName = referencedTypeName;
+
+    // Checking for inline overrides of the type: [memberName:ClassOverrideName]
     const regexForSchemaOverrides = /\[(.*?)]/g;
     const schemaOverrides = referencedTypeName.match(regexForSchemaOverrides);
     let referenceOverrides: ReferenceOverride[] = [];
@@ -223,6 +225,8 @@ export class ReferenceBuilder {
         // For Maps, we treat them as objects but do not try to define their shape, because their keys can vary
         // at runtime.
         return { schema: { type: 'object' }, referenceComponents: [] };
+      case 'object':
+        return { schema: { type: 'object' }, referenceComponents: [] };
       default:
         // If we got here we are dealing with a non-primitive (most likely a custom class or an SObject).
         const referencedType = TypesRepository.getInstance().getFromAllByName(typeName);
diff --git a/src/transpiler/openapi/parsers/__tests__/ReferenceBuilder.spec.ts b/src/transpiler/openapi/parsers/__tests__/ReferenceBuilder.spec.ts
index e728ee7c..eeec6d05 100644
--- a/src/transpiler/openapi/parsers/__tests__/ReferenceBuilder.spec.ts
+++ b/src/transpiler/openapi/parsers/__tests__/ReferenceBuilder.spec.ts
@@ -623,7 +623,7 @@ describe('ReferenceBuilder', () => {
   });
 
   describe('References can override their type', () => {
-    it("should correctly parse a manually defined schema in the reference's ApexDoc", function () {
+    it('should correctly parse a reference with an overridden reference', function () {
       const classMirror = new ClassMirrorBuilder()
         .addFiled(
           new FieldMirrorBuilder()
@@ -703,5 +703,49 @@ describe('ReferenceBuilder', () => {
       expect(result.referenceComponents.some((ref) => ref.referencedClass === 'child')).toBe(true);
       expect(result.referenceComponents.some((ref) => ref.referencedClass === 'grandchild')).toBe(true);
     });
+
+    it('should correctly parse a referenced property overridden inline', function () {
+      const mainClassMirror = new ClassMirrorBuilder()
+        .withName('parent')
+        .addFiled(new FieldMirrorBuilder().withName('childClassMember').withType('Object').build())
+        .build();
+
+      const childClass = new ClassMirrorBuilder()
+        .withName('child')
+        .addFiled(
+          new FieldMirrorBuilder()
+            .withName('grandChildClassMember')
+            .withReferencedType({
+              type: 'GrandChildClass',
+              rawDeclaration: 'GrandChildClass',
+            })
+            .build(),
+        )
+        .build();
+
+      const grandChildClass = new ClassMirrorBuilder()
+        .withName('grandchild')
+        .addFiled(new FieldMirrorBuilder().withName('stringMember').withType('String').build())
+        .build();
+
+      TypesRepository.getInstance = jest.fn().mockReturnValue({
+        getFromAllByName: jest
+          .fn()
+          .mockReturnValueOnce({ type: mainClassMirror, isChild: false })
+          .mockReturnValueOnce({ type: childClass, isChild: false })
+          .mockReturnValueOnce({ type: grandChildClass, isChild: false }),
+      });
+
+      const result = new ReferenceBuilder().build('className[childClassMember:ChildClass]');
+
+      expect(result.referenceComponents).toHaveLength(3);
+      expect(
+        result.referenceComponents.some(
+          (ref) => ref.referencedClass === 'parent_className[childClassMember:ChildClass]',
+        ),
+      ).toBe(true);
+      expect(result.referenceComponents.some((ref) => ref.referencedClass === 'child')).toBe(true);
+      expect(result.referenceComponents.some((ref) => ref.referencedClass === 'grandchild')).toBe(true);
+    });
   });
 });