update test and docs

jonaslagoni · Mar 4, 2024 · 2558388 · 2558388
1 parent 0765db2
commit 2558388
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 12 deletions.
diff --git a/docs/presets.md b/docs/presets.md
@@ -416,6 +416,14 @@ This preset is a generator for the meta model `ConstrainedObjectModel` and [can
 |---|---|---|
 | `field` | A method to extend rendered given field. | `field` object as a [`ConstrainedObjectPropertyModel`](./internal-model.md#the-constrained-meta-model) instance. |
 
+#### **Enum**
+
+This preset is a generator for the meta model `ConstrainedEnumModel` and [can be accessed through the `model` argument](#presets-shape).
+
+| Method | Description | Additional arguments |
+|---|---|---|
+| `item` | A method to extend rendering the enum items. | `item` object as a [`ConstrainedEnumValueModel`](./internal-model.md#the-constrained-meta-model) instance. `index` as `number`, the current enum item being rendered. |
+
 ### C#
 
 #### **Class**

diff --git a/examples/generate-go-enums/__snapshots__/index.spec.ts.snap b/examples/generate-go-enums/__snapshots__/index.spec.ts.snap
@@ -34,8 +34,7 @@ var ValuesToCities = map[any]Cities{
   CitiesValues[CitiesLondon]: CitiesLondon,
   CitiesValues[CitiesRome]: CitiesRome,
   CitiesValues[CitiesBrussels]: CitiesBrussels,
-}
-",
+}",
 ]
 `;
 
@@ -65,7 +64,6 @@ var ValuesToOptions = map[any]Options{
   OptionsValues[OptionsNumber_213]: OptionsNumber_213,
   OptionsValues[OptionsTrue]: OptionsTrue,
   OptionsValues[OptionsRun]: OptionsRun,
-}
-",
+}",
 ]
 `;
diff --git a/src/generators/go/renderers/EnumRenderer.ts b/src/generators/go/renderers/EnumRenderer.ts
@@ -22,6 +22,11 @@ export class EnumRenderer extends GoRenderer<ConstrainedEnumModel> {
       return `${this.model.name}Values[${value.key}]: ${value.key},`;
     });
     const additionalContent = await this.runAdditionalContentPreset();
+    const renderedAdditionalContent = additionalContent
+      ? `
+    ${this.indent(additionalContent)}    
+`
+      : '';
     const values = this.model.values
       .map((value) => {
         return value.value;
@@ -46,10 +51,7 @@ func (op ${this.model.name}) Value() any {
 var ${this.model.name}Values = []any{${values}}
 var ValuesTo${this.model.name} = map[any]${this.model.name}{
 ${this.indent(this.renderBlock(valuesToEnumMap))}
-}
-
-${this.indent(additionalContent)}
-`;
+}${renderedAdditionalContent}`;
   }
 
   async renderItems(): Promise<string> {

diff --git a/test/generators/go/__snapshots__/GoGenerator.spec.ts.snap b/test/generators/go/__snapshots__/GoGenerator.spec.ts.snap
@@ -89,8 +89,7 @@ var ValuesToThings = map[any]Things{
   ThingsValues[ThingsReservedNumber_1]: ThingsReservedNumber_1,
   ThingsValues[ThingsFalse]: ThingsFalse,
   ThingsValues[ThingsCurlyleftQuotationTestQuotationColonQuotationTestQuotationCurlyright]: ThingsCurlyleftQuotationTestQuotationColonQuotationTestQuotationCurlyright,
-}
-"
+}"
 `;
 
 exports[`GoGenerator should render \`struct\` type 1`] = `
@@ -131,8 +130,7 @@ var ValuesToCustomEnum = map[any]CustomEnum{
   CustomEnumValues[CustomEnumTexas]: CustomEnumTexas,
   CustomEnumValues[CustomEnumAlabama]: CustomEnumAlabama,
   CustomEnumValues[CustomEnumCalifornia]: CustomEnumCalifornia,
-}
-"
+}"
 `;
 
 exports[`GoGenerator should work custom preset for \`struct\` type 1`] = `