1 files changed, 85 insertions, 0 deletions
diff --git a/spec/lib/gitlab/puma/error_handler_spec.rb b/spec/lib/gitlab/puma/error_handler_spec.rb
new file mode 100644
index 00000000000..5b7cdf37af1
--- /dev/null
+++ b/spec/lib/gitlab/puma/error_handler_spec.rb
@@ -0,0 +1,85 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe Gitlab::Puma::ErrorHandler, feature_category: :shared do
+  subject { described_class.new(is_production) }
+
+  let(:is_production) { true }
+  let(:ex) { StandardError.new('Sample error message') }
+  let(:env) { {} }
+  let(:status_code) { 500 }
+
+  describe '#execute' do
+    it 'captures the exception and returns a Rack response' do
+      allow(Raven.configuration).to receive(:capture_allowed?).and_return(true)
+      expect(Raven).to receive(:capture_exception).with(
+        ex,
+        tags: { handler: 'puma_low_level' },
+        extra: { puma_env: env, status_code: status_code }
+      ).and_call_original
+
+      status, headers, message = subject.execute(ex, env, status_code)
+
+      expect(status).to eq(500)
+      expect(headers).to eq({})
+      expect(message).to eq(described_class::PROD_ERROR_MESSAGE)
+    end
+
+    context 'when capture is not allowed' do
+      it 'returns a Rack response without capturing the exception' do
+        allow(Raven.configuration).to receive(:capture_allowed?).and_return(false)
+        expect(Raven).not_to receive(:capture_exception)
+
+        status, headers, message = subject.execute(ex, env, status_code)
+
+        expect(status).to eq(500)
+        expect(headers).to eq({})
+        expect(message).to eq(described_class::PROD_ERROR_MESSAGE)
+      end
+    end
+
+    context 'when not in production' do
+      let(:is_production) { false }
+
+      it 'returns a Rack response with dev error message' do
+        allow(Raven.configuration).to receive(:capture_allowed?).and_return(true)
+
+        status, headers, message = subject.execute(ex, env, status_code)
+
+        expect(status).to eq(500)
+        expect(headers).to eq({})
+        expect(message).to eq(described_class::DEV_ERROR_MESSAGE)
+      end
+    end
+
+    context 'when status code is nil' do
+      let(:status_code) { 500 }
+
+      it 'defaults to error 500' do
+        allow(Raven.configuration).to receive(:capture_allowed?).and_return(false)
+        expect(Raven).not_to receive(:capture_exception)
+
+        status, headers, message = subject.execute(ex, env, status_code)
+
+        expect(status).to eq(500)
+        expect(headers).to eq({})
+        expect(message).to eq(described_class::PROD_ERROR_MESSAGE)
+      end
+    end
+
+    context 'when status code is provided' do
+      let(:status_code) { 404 }
+
+      it 'uses the provided status code in the response' do
+        allow(Raven.configuration).to receive(:capture_allowed?).and_return(true)
+
+        status, headers, message = subject.execute(ex, env, status_code)
+
+        expect(status).to eq(404)
+        expect(headers).to eq({})
+        expect(message).to eq(described_class::PROD_ERROR_MESSAGE)
+      end
+    end
+  end
+end