fix(services): resolve enum type correctly if enumName is provided

nestjs · Aug 28, 2023 · b517b7f · b517b7f
1 parent ae47808
commit b517b7f
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/lib/services/schema-object-factory.ts b/lib/services/schema-object-factory.ts
@@ -294,7 +294,7 @@ export class SchemaObjectFactory {
 
     if (!(enumName in schemas)) {
       schemas[enumName] = {
-        type: 'string',
+        type: metadata.type as string,
         enum:
           metadata.isArray && metadata.items
             ? metadata.items['enum']

diff --git a/test/services/schema-object-factory.spec.ts b/test/services/schema-object-factory.spec.ts
@@ -266,4 +266,20 @@ describe('SchemaObjectFactory', () => {
       });
     });
   });
+
+  describe('createEnumSchemaType', () => {
+    it('should assign schema type correctly if enumName is provided', () => {
+      const metadata = {
+        type: 'number',
+        enum: [1, 2, 3],
+        enumName: 'MyEnum',
+        isArray: false
+      };
+      const schemas = {};
+
+      schemaObjectFactory.createEnumSchemaType('field', metadata, schemas);
+
+      expect(schemas).toEqual({ MyEnum: { enum: [1, 2, 3], type: 'number' } });
+    });
+  });
 });