From 35dc8d4e5dbc04dd877f4cbf459e0d4f60354e8a Mon Sep 17 00:00:00 2001 From: Fleque Date: Tue, 27 Jan 2015 22:47:49 +0100 Subject: [PATCH] Fix for the selection of the methodResponse from an operation during code generation. Now all the smallest 2xx code is used --- .../java/com/wordnik/swagger/codegen/DefaultCodegen.java | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/src/main/java/com/wordnik/swagger/codegen/DefaultCodegen.java b/src/main/java/com/wordnik/swagger/codegen/DefaultCodegen.java index 480e092b74e..98dcb7bcee3 100644 --- a/src/main/java/com/wordnik/swagger/codegen/DefaultCodegen.java +++ b/src/main/java/com/wordnik/swagger/codegen/DefaultCodegen.java @@ -4,6 +4,7 @@ import com.wordnik.swagger.models.*; import com.wordnik.swagger.models.parameters.*; import com.wordnik.swagger.models.properties.*; import com.wordnik.swagger.util.Json; + import org.apache.commons.lang.StringUtils; import java.util.*; @@ -577,10 +578,12 @@ public class DefaultCodegen { } if(operation.getResponses() != null) { - for(String responseCode: operation.getResponses().keySet()) { + for(String responseCode: new TreeSet(operation.getResponses().keySet())) { Response response = operation.getResponses().get(responseCode); - if("200".equals(responseCode)) { + if (responseCode.startsWith("2")) { + // use the first, i.e. the smallest 2xx response status as methodResponse methodResponse = response; + break; } } if(methodResponse == null && operation.getResponses().keySet().contains("default")) {