Fix host test

Author: blake

lib/grape-swagger/openapi_3/endpoint.rb

@@ -23,13 +23,18 @@ def content_types_for(target_class)
     # openapi 3.0 related parts
     #
     # required keys for SwaggerObject
-    def swagger_object(_target_class, _request, options)
+    def swagger_object(_target_class, request, options)
+      url = GrapeSwagger::DocMethods::OptionalObject.build(:host, options, request)
+      base_path = GrapeSwagger::DocMethods::OptionalObject.build(:base_path, options, request)
+      servers = options[:servers] || [{ url: "#{request.scheme}://#{url}#{base_path}" }]
+      servers = servers.is_a?(Hash) ? [servers] : servers
+
       object = {
         info:                info_object(options[:info].merge(version: options[:doc_version])),
         openapi:             '3.0.0',
         security:            options[:security],
         authorizations:      options[:authorizations],
-        servers:             options[:servers].is_a?(Hash) ? [options[:servers]] : options[:servers]
+        servers: servers
       }
 
       if options[:security_definitions] || options[:security]

spec/openapi_3/host_spec.rb

@@ -0,0 +1,48 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+describe 'host in the swagger_doc' do
+  before :all do
+    module TheApi
+      class EmptyApi < Grape::API
+        format :json
+
+        desc 'This gets something.'
+        get '/something' do
+          { bla: 'something' }
+        end
+
+        add_swagger_documentation openapi_version: '3.0'
+      end
+    end
+  end
+
+  def app
+    TheApi::EmptyApi
+  end
+
+  describe 'host should include port' do
+    subject do
+      get 'http://example.com:8080/swagger_doc'
+      JSON.parse(last_response.body)
+    end
+
+    specify do
+      expect(subject['servers'].first['url']).to eq 'http://example.com:8080'
+    end
+  end
+
+  describe 'respect X-Forwarded-Host over Host header' do
+    subject do
+      header 'Host', 'dummy.example.com'
+      header 'X-Forwarded-Host', 'real.example.com'
+      get '/swagger_doc'
+      JSON.parse(last_response.body)
+    end
+
+    specify do
+      expect(subject['servers'].first['url']).to eq 'http://real.example.com'
+    end
+  end
+end