1 files changed, 5 insertions, 2 deletions
diff --git a/spec/support/shared_examples/controllers/unique_visits_shared_examples.rb b/spec/support/shared_examples/controllers/unique_visits_shared_examples.rb
index 3f97c031e27..30914e61df0 100644
--- a/spec/support/shared_examples/controllers/unique_visits_shared_examples.rb
+++ b/spec/support/shared_examples/controllers/unique_visits_shared_examples.rb
@@ -1,6 +1,8 @@
 # frozen_string_literal: true
 
 RSpec.shared_examples 'tracking unique visits' do |method|
+  include TrackingHelpers
+
   let(:request_params) { {} }
 
   it 'tracks unique visit if the format is HTML' do
@@ -14,14 +16,15 @@ RSpec.shared_examples 'tracking unique visits' do |method|
     expect(Gitlab::UsageDataCounters::HLLRedisCounter)
       .to receive(:track_event).with(target_id, values: kind_of(String))
 
-    request.headers['DNT'] = '0'
+    stub_do_not_track('0')
 
     get method, params: request_params, format: :html
   end
 
   it 'does not track unique visit if DNT is enabled' do
     expect(Gitlab::UsageDataCounters::HLLRedisCounter).not_to receive(:track_event)
-    request.headers['DNT'] = '1'
+
+    stub_do_not_track('1')
 
     get method, params: request_params, format: :html
   end