Skip to content

Commit

Permalink
[JAVA - jaxrs-reasteasy-eap] Add import to models (#179)
Browse files Browse the repository at this point in the history
Call super management for managing imports in postProcessModelProperty
  • Loading branch information
Zomzog authored and jmini committed May 30, 2018
1 parent ad5d5f5 commit bd50d36
Show file tree
Hide file tree
Showing 2 changed files with 30 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -147,6 +147,7 @@ public Map<String, Object> postProcessOperations(Map<String, Object> objs) {

@Override
public void postProcessModelProperty(CodegenModel model, CodegenProperty property) {
super.postProcessModelProperty(model, property);
// Add imports for Jackson
if (!BooleanUtils.toBoolean(model.isEnum)) {
model.imports.add("JsonProperty");
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,29 @@
package org.openapitools.codegen.java.jaxrs;

import io.swagger.util.Json;
import io.swagger.v3.oas.models.media.MapSchema;
import io.swagger.v3.oas.models.media.Schema;
import org.openapitools.codegen.CodegenModel;
import org.openapitools.codegen.languages.JavaResteasyEapServerCodegen;
import org.testng.annotations.Test;

import java.util.Collections;

import static org.testng.Assert.assertEquals;
import static org.testng.Assert.assertTrue;

public class JavaResteasyEapServerCodegenModelTest {

@Test(description = "convert a simple java model with java8 types")
public void mapModelTest() {
final Schema model = new Schema()
.description("A model with a map")
.addProperties("map", new MapSchema());

final JavaResteasyEapServerCodegen codegen = new JavaResteasyEapServerCodegen();
final CodegenModel cm = codegen.fromModel("sample", model, Collections.singletonMap("sample", model));

assertEquals(cm.vars.get(0).baseType, "Map");
assertTrue(cm.imports.contains("HashMap"));
}
}

0 comments on commit bd50d36

Please # to comment.