added spec for expected headers when error raised

spec/grape/headers_on_error_spec.rb

@@ -0,0 +1,47 @@
+require 'spec_helper'
+
+describe Grape::API do
+  let(:error_header) do
+    Class.new(Grape::API) do
+      before do
+        header 'X-Grape-Before-Header', '1'
+      end
+      after do
+        header 'X-Grape-After-Header', '1'
+      end
+      get '/success' do
+        header 'X-Grape-Returns-Error', '1'
+      end
+      get '/error' do
+        header 'X-Grape-Returns-Error', '1'
+        error!(success: false)
+      end
+    end
+  end
+
+  subject do
+    ErrorHeader = error_header unless defined?(ErrorHeader)
+    Class.new(Grape::API) do
+      format :json
+      mount ErrorHeader => '/'
+    end
+  end
+
+  def app
+    subject
+  end
+
+  it 'should returns all headers on success' do
+    get '/success'
+    expect(last_response.headers['X-Grape-Returns-Error']).to eq('1')
+    expect(last_response.headers['X-Grape-Before-Header']).to eq('1')
+    expect(last_response.headers['X-Grape-After-Header']).to eq('1')
+  end
+
+  it 'should returns all headers on error' do
+    get '/error'
+    expect(last_response.headers['X-Grape-Returns-Error']).to eq('1')
+    expect(last_response.headers['X-Grape-Before-Header']).to eq('1')
+    expect(last_response.headers['X-Grape-After-Header']).to eq('1')
+  end
+end