Fix handling of HTTP status codes from routes

Author: milgner

lib/grape-swagger/endpoint.rb

@@ -224,8 +224,13 @@ def response_object(route)
       end
     end
 
+    def success_code?(code)
+      status = code.is_a?(Array) ? code.first : code[:code]
+      status.between?(200, 299)
+    end
+
     def http_codes_from_route(route)
-      if route.http_codes.is_a?(Array) && route.http_codes.any? { |code| code[:code].between?(200, 299) }
+      if route.http_codes.is_a?(Array) && route.http_codes.any? { |code| success_code?(code) }
         route.http_codes.clone
       else
         success_codes_from_route(route) + (route.http_codes || route.options[:failure] || [])

spec/swagger_v2/api_swagger_v2_status_codes_spec.rb

@@ -2,15 +2,15 @@
 
 require 'spec_helper'
 
-describe 'http status code behaivours' do
+describe 'http status code behaviours' do
   include_context "#{MODEL_PARSER} swagger example"
 
   subject do
     get '/swagger_doc'
     JSON.parse(last_response.body)
   end
 
-  context 'when non-default success codes are deifined' do
+  context 'when non-default success codes are defined' do
     let(:app) do
       Class.new(Grape::API) do
         desc 'Has explicit success http_codes defined' do
@@ -31,6 +31,26 @@
     end
   end
 
+  context 'when success and failures are defined' do
+    let(:app) do
+      Class.new(Grape::API) do
+        desc 'Has explicit success http_codes defined' do
+          success code: 202, model: Entities::UseResponse, message: 'a changed status code'
+          failure [[400, 'Bad Request']]
+        end
+
+        post '/accepting_endpoint' do
+          'We got the message!'
+        end
+        add_swagger_documentation
+      end
+    end
+
+    it 'only includes the defined http codes' do
+      expect(subject['paths']['/accepting_endpoint']['post']['responses'].keys.sort).to eq(%w[202 400].sort)
+    end
+  end
+
   context 'when no success codes defined' do
     let(:app) do
       Class.new(Grape::API) do