From c15ac0b6d4c946b6b78822cc7f8cb25ace2cdffc Mon Sep 17 00:00:00 2001
From: Camille Chafer <cchafer@ankama.com>
Date: Wed, 25 Feb 2015 17:12:28 +0100
Subject: [PATCH] Each type used by any response is now added to the imports
 list

---
 .../com/wordnik/swagger/codegen/DefaultCodegen.java   | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/modules/swagger-codegen/src/main/java/com/wordnik/swagger/codegen/DefaultCodegen.java b/modules/swagger-codegen/src/main/java/com/wordnik/swagger/codegen/DefaultCodegen.java
index 7b600ff488d..5e60c425c3f 100644
--- a/modules/swagger-codegen/src/main/java/com/wordnik/swagger/codegen/DefaultCodegen.java
+++ b/modules/swagger-codegen/src/main/java/com/wordnik/swagger/codegen/DefaultCodegen.java
@@ -541,6 +541,7 @@ public class DefaultCodegen {
   }
 
   private Response findMethodResponse(Map<String, Response> responses) {
+
     String code = null;
     for(String responseCode : responses.keySet()) {
       if (responseCode.startsWith("2") || responseCode.equals("default")) {
@@ -627,6 +628,11 @@ public class DefaultCodegen {
         Response response = entry.getValue();
         CodegenResponse r = fromResponse(entry.getKey(), response);
         r.hasMore = true;
+        if(r.baseType != null &&
+            !defaultIncludes.contains(r.baseType) &&
+            !languageSpecificPrimitives.contains(r.baseType))
+          imports.add(r.baseType);
+
         if (response == methodResponse)
           methodCodegenResponse = r;
         op.responses.add(r);
@@ -653,11 +659,6 @@ public class DefaultCodegen {
       }
     }
 
-    if(op.returnBaseType != null &&
-      !defaultIncludes.contains(op.returnBaseType) &&
-      !languageSpecificPrimitives.contains(op.returnBaseType))
-      imports.add(op.returnBaseType);
-
     List<Parameter> parameters = operation.getParameters();
     CodegenParameter bodyParam = null;
     List<CodegenParameter> allParams = new ArrayList<CodegenParameter>();