1 files changed, 39 insertions, 1 deletions
diff --git a/spec/lib/gitlab/request_context_spec.rb b/spec/lib/gitlab/request_context_spec.rb
index b9acfa4a841..44664be7d39 100644
--- a/spec/lib/gitlab/request_context_spec.rb
+++ b/spec/lib/gitlab/request_context_spec.rb
@@ -2,7 +2,7 @@
 
 require 'spec_helper'
 
-RSpec.describe Gitlab::RequestContext, :request_store do
+RSpec.describe Gitlab::RequestContext, :request_store, feature_category: :application_instrumentation do
   subject { described_class.instance }
 
   before do
@@ -11,6 +11,44 @@ RSpec.describe Gitlab::RequestContext, :request_store do
 
   it { is_expected.to have_attributes(client_ip: nil, start_thread_cpu_time: nil, request_start_time: nil) }
 
+  describe '.start_request_context' do
+    let(:request) { ActionDispatch::Request.new({ 'REMOTE_ADDR' => '1.2.3.4' }) }
+    let(:start_request_context) { described_class.start_request_context(request: request) }
+
+    before do
+      allow(Gitlab::Metrics::System).to receive(:real_time).and_return(123)
+    end
+
+    it 'sets the client IP' do
+      expect { start_request_context }.to change { subject.client_ip }.from(nil).to('1.2.3.4')
+    end
+
+    it 'sets the spam params' do
+      expect { start_request_context }.to change { subject.spam_params }.from(nil).to(::Spam::SpamParams)
+    end
+
+    it 'sets the request start time' do
+      expect { start_request_context }.to change { subject.request_start_time }.from(nil).to(123)
+    end
+  end
+
+  describe '.start_thread_context' do
+    let(:start_thread_context) { described_class.start_thread_context }
+
+    before do
+      allow(Gitlab::Metrics::System).to receive(:thread_cpu_time).and_return(123)
+      allow(Gitlab::Memory::Instrumentation).to receive(:start_thread_memory_allocations).and_return(456)
+    end
+
+    it 'sets the thread cpu time' do
+      expect { start_thread_context }.to change { subject.start_thread_cpu_time }.from(nil).to(123)
+    end
+
+    it 'sets the thread memory allocations' do
+      expect { start_thread_context }.to change { subject.thread_memory_allocations }.from(nil).to(456)
+    end
+  end
+
   describe '#request_deadline' do
     let(:request_start_time) { 1575982156.206008 }