Fixed TS generator not generating union enums

asyncapi · Jul 28, 2022 · 39b45af · 39b45af
1 parent ff3f5d4
commit 39b45af
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/src/generators/typescript/renderers/EnumRenderer.ts b/src/generators/typescript/renderers/EnumRenderer.ts
@@ -20,6 +20,12 @@ ${this.indent(this.renderBlock(content, 2))}
 }`;
  }
 
+ renderUnionEnum(model: ConstrainedEnumModel): string {
+ const enums = model.values || [];
+ const enumTypes = enums.map(t => t.value).join(' | ');
+ return `type ${model.name} = ${enumTypes};`;
+ }
+
  async renderItems(): Promise<string> {
  const enums = this.model.values || [];
  const items: string[] = [];
@@ -38,7 +44,10 @@ ${this.indent(this.renderBlock(content, 2))}
 }
 
 export const TS_DEFAULT_ENUM_PRESET: EnumPresetType<TypeScriptOptions> = {
- self({ renderer }) {
+ self({ renderer, options, model }) {
+ if (options.enumType === 'union' && model instanceof ConstrainedEnumModel) {
+ return renderer.renderUnionEnum(model);
+ }
  return renderer.defaultSelf();
  },
  item({ item }): string {

diff --git a/test/generators/typescript/__snapshots__/TypeScriptGenerator.spec.ts.snap b/test/generators/typescript/__snapshots__/TypeScriptGenerator.spec.ts.snap
@@ -98,7 +98,7 @@ exports[`TypeScriptGenerator should render \`enum\` type 1`] = `
 }"
 `;
 
-exports[`TypeScriptGenerator should render \`enum\` type as \`union\` if option enumType = \`union\` 1`] = `"type States = States;"`;
+exports[`TypeScriptGenerator should render \`enum\` type as \`union\` if option enumType = \`union\` 1`] = `"type States = \\"Texas\\" | \\"Alabama\\" | \\"California\\";"`;
 
 exports[`TypeScriptGenerator should render \`interface\` type 1`] = `
 "interface Address {