feat: unions of enum's are rendered for properties for TS (#155)

Author: jonaslagoni

src/generators/typescript/TypeScriptRenderer.ts

@@ -23,6 +23,9 @@ export abstract class TypeScriptRenderer extends AbstractRenderer<TypeScriptOpti
     if (Array.isArray(model)) {
       return model.map(t => this.renderType(t)).join(' | ');
     }
+    if (model.enum !== undefined) {
+      return model.enum.map(value => typeof value === 'string' ? `"${value}"` : value).join(' | ');
+    }
     if (model.$ref !== undefined) {
       return model.$ref;
     }

test/generators/typescript/TypeScriptGenerator.spec.ts

@@ -5,7 +5,36 @@ describe('TypeScriptGenerator', function() {
   beforeEach(() => {
     generator = new TypeScriptGenerator();
   });
+  test('should render union property type', async function() {
+    const doc = {
+      $id: "_address",
+      type: "object",
+      properties: {
+        state:          { type: "string", enum: ["Texas", "Alabama", "California", "other"] }
+      }
+    };
+    const expected = `export class Address {
+  private _state?: "Texas" | "Alabama" | "California" | "other";
+
+  constructor(input: {
+    state?: "Texas" | "Alabama" | "California" | "other",
+  }) {
+    this._state = input.state;
+  }
 
+  get state(): "Texas" | "Alabama" | "California" | "other" | undefined { return this._state; }
+  set state(state: "Texas" | "Alabama" | "California" | "other" | undefined) { this._state = state; }
+}`;
+
+    const inputModel = await generator.process(doc);
+    const model = inputModel.models["_address"];
+
+    let classModel = await generator.renderClass(model, inputModel);
+    expect(classModel).toEqual(expected);
+
+    classModel = await generator.render(model, inputModel);
+    expect(classModel).toEqual(expected);
+  });
   test('should render `class` type', async function() {
     const doc = {
       $id: "_address",

test/generators/typescript/TypeScriptRenderer.spec.ts

@@ -34,5 +34,10 @@ describe('TypeScriptRenderer', function() {
       model2.$ref = "ref2";
       expect(renderer.renderType([model1, model2])).toEqual('ref1 | ref2');
     });
+    test('Should render enums', function() {
+      const model = new CommonModel();
+      model.enum = ["enum1", "enum2", 9];
+      expect(renderer.renderType(model)).toEqual('"enum1" | "enum2" | 9');
+    });
   });
 });