From 13e56514eeecdf911f58959cc4323d76b5422812 Mon Sep 17 00:00:00 2001 From: Guo Huang Date: Wed, 13 Apr 2016 14:03:19 -0700 Subject: [PATCH] added import tests for Go --- .../java/io/swagger/codegen/go/GoModelTest.java | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) diff --git a/modules/swagger-codegen/src/test/java/io/swagger/codegen/go/GoModelTest.java b/modules/swagger-codegen/src/test/java/io/swagger/codegen/go/GoModelTest.java index 6f12032460c..83fa8d6437e 100644 --- a/modules/swagger-codegen/src/test/java/io/swagger/codegen/go/GoModelTest.java +++ b/modules/swagger-codegen/src/test/java/io/swagger/codegen/go/GoModelTest.java @@ -38,8 +38,7 @@ public class GoModelTest { Assert.assertEquals(cm.classname, "Sample"); Assert.assertEquals(cm.description, "a sample model"); Assert.assertEquals(cm.vars.size(), 3); - // {{imports}} is not used in template - //Assert.assertEquals(cm.imports.size(), 1); + Assert.assertEquals(cm.imports.size(), 1); final CodegenProperty property1 = cm.vars.get(0); Assert.assertEquals(property1.baseName, "id"); @@ -199,8 +198,7 @@ public class GoModelTest { Assert.assertEquals(cm.classname, "Sample"); Assert.assertEquals(cm.description, "a sample model"); Assert.assertEquals(cm.vars.size(), 1); - // {{imports}} is not used in template - //Assert.assertEquals(Sets.intersection(cm.imports, Sets.newHashSet("Children")).size(), 1); + Assert.assertEquals(Sets.intersection(cm.imports, Sets.newHashSet("Children")).size(), 1); final CodegenProperty property1 = cm.vars.get(0); Assert.assertEquals(property1.baseName, "children"); @@ -226,7 +224,7 @@ public class GoModelTest { Assert.assertEquals(cm.classname, "Sample"); Assert.assertEquals(cm.description, "an array model"); Assert.assertEquals(cm.vars.size(), 0); - // skip import test as import is not used by Go codegen + Assert.assertEquals(cm.imports.size(), 1); } @Test(description = "convert an map model") @@ -241,9 +239,8 @@ public class GoModelTest { Assert.assertEquals(cm.classname, "Sample"); Assert.assertEquals(cm.description, "a map model"); Assert.assertEquals(cm.vars.size(), 0); - // {{imports}} is not used in template - //Assert.assertEquals(cm.imports.size(), 2); - //Assert.assertEquals(Sets.intersection(cm.imports, Sets.newHashSet("Children")).size(), 1); + Assert.assertEquals(cm.imports.size(), 1); + Assert.assertEquals(Sets.intersection(cm.imports, Sets.newHashSet("Children")).size(), 1); } @DataProvider(name = "modelNames")