From 4e45ef954d3f31d35372e9a5a17fce9c9e21295d Mon Sep 17 00:00:00 2001 From: tao Date: Sun, 19 Jun 2016 21:29:14 -0700 Subject: [PATCH] better tests --- .../swagger/codegen/java/JavaModelTest.java | 19 ++++++++++--------- 1 file changed, 10 insertions(+), 9 deletions(-) diff --git a/modules/swagger-codegen/src/test/java/io/swagger/codegen/java/JavaModelTest.java b/modules/swagger-codegen/src/test/java/io/swagger/codegen/java/JavaModelTest.java index 7927bc2b7ea..11db4afc3ca 100644 --- a/modules/swagger-codegen/src/test/java/io/swagger/codegen/java/JavaModelTest.java +++ b/modules/swagger-codegen/src/test/java/io/swagger/codegen/java/JavaModelTest.java @@ -450,18 +450,19 @@ public class JavaModelTest { Assert.assertNull(cm.allowableValues); } - @Test(description = "model with Map> should import BigDecimal") + @Test(description = "types used by inner properties should be imported") public void mapWithAnListOfBigDecimalTest() { - final Model model = new ModelImpl() + final CodegenModel cm1 = new JavaClientCodegen().fromModel("sample", new ModelImpl() .description("model with Map>") - .property("map", new MapProperty().additionalProperties(new ArrayProperty(new DecimalProperty()))); + .property("map", new MapProperty().additionalProperties(new ArrayProperty(new DecimalProperty())))); + Assert.assertEquals(cm1.vars.get(0).datatype, "Map>"); + Assert.assertTrue(cm1.imports.contains("BigDecimal")); - final DefaultCodegen codegen = new JavaClientCodegen(); - final CodegenModel cm = codegen.fromModel("sample", model); - - final CodegenProperty property = cm.vars.get(0); - Assert.assertEquals(property.datatype, "Map>"); - Assert.assertTrue(cm.imports.contains("BigDecimal")); + final CodegenModel cm2 = new JavaClientCodegen().fromModel("sample", new ModelImpl() + .description("model with Map>>") + .property("map", new MapProperty().additionalProperties(new MapProperty().additionalProperties(new ArrayProperty(new DecimalProperty()))))); + Assert.assertEquals(cm2.vars.get(0).datatype, "Map>>"); + Assert.assertTrue(cm2.imports.contains("BigDecimal")); } @DataProvider(name = "modelNames")