headers ending in "id" are displayed correctly

zipmark · Feb 13, 2015 · 4b15530 · joshuaswilcox · Jun 18, 2015 · 4b15530
1 parent 93c88c5
commit 4b15530
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/lib/rspec_api_documentation/headers.rb b/lib/rspec_api_documentation/headers.rb
@@ -8,6 +8,7 @@ def env_to_headers(env)
         # HTTP_ACCEPT_CHARSET => Accept-Charset
         if key =~ /^(HTTP_|CONTENT_TYPE)/
           header = key.gsub(/^HTTP_/, '').titleize.split.join("-")
+          header.concat('-Id') if key.scan(/_ID\Z/).any?
           headers[header] = value
         end
       end

diff --git a/spec/http_test_client_spec.rb b/spec/http_test_client_spec.rb
@@ -62,13 +62,14 @@
 
   describe "#request_headers" do
     before do
-      test_client.get "/", {}, { "Accept" => "application/json", "Content-Type" => "application/json" }
+      test_client.get "/", {}, { "Accept" => "application/json", "Content-Type" => "application/json", "User-Id" => "1" }
     end
 
     it "should contain all the headers" do
       expect(test_client.request_headers).to eq({
         "Accept" => "application/json",
-        "Content-Type" => "application/json"
+        "Content-Type" => "application/json",
+        "User-Id" => "1"
       })
     end
   end