1 files changed, 7 insertions, 4 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 428389a9a01..3f97c031e27 100644
--- a/spec/support/shared_examples/controllers/unique_visits_shared_examples.rb
+++ b/spec/support/shared_examples/controllers/unique_visits_shared_examples.rb
@@ -4,27 +4,30 @@ RSpec.shared_examples 'tracking unique visits' do |method|
   let(:request_params) { {} }
 
   it 'tracks unique visit if the format is HTML' do
-    expect_any_instance_of(Gitlab::Analytics::UniqueVisits).to receive(:track_visit).with(instance_of(String), target_id)
+    expect(Gitlab::UsageDataCounters::HLLRedisCounter)
+      .to receive(:track_event).with(target_id, values: kind_of(String))
 
     get method, params: request_params, format: :html
   end
 
   it 'tracks unique visit if DNT is not enabled' do
-    expect_any_instance_of(Gitlab::Analytics::UniqueVisits).to receive(:track_visit).with(instance_of(String), target_id)
+    expect(Gitlab::UsageDataCounters::HLLRedisCounter)
+      .to receive(:track_event).with(target_id, values: kind_of(String))
+
     request.headers['DNT'] = '0'
 
     get method, params: request_params, format: :html
   end
 
   it 'does not track unique visit if DNT is enabled' do
-    expect_any_instance_of(Gitlab::Analytics::UniqueVisits).not_to receive(:track_visit)
+    expect(Gitlab::UsageDataCounters::HLLRedisCounter).not_to receive(:track_event)
     request.headers['DNT'] = '1'
 
     get method, params: request_params, format: :html
   end
 
   it 'does not track unique visit if the format is JSON' do
-    expect_any_instance_of(Gitlab::Analytics::UniqueVisits).not_to receive(:track_visit)
+    expect(Gitlab::UsageDataCounters::HLLRedisCounter).not_to receive(:track_event)
 
     get method, params: request_params, format: :json
   end