136 lines
		
	
	
		
			2.7 KiB
		
	
	
	
		
			Ruby
		
	
	
	
			
		
		
	
	
			136 lines
		
	
	
		
			2.7 KiB
		
	
	
	
		
			Ruby
		
	
	
	
| # frozen_string_literal: true
 | |
| 
 | |
| require "spec_helper"
 | |
| 
 | |
| RSpec.describe RedisTracking do
 | |
|   include TrackingHelpers
 | |
| 
 | |
|   let(:user) { create(:user) }
 | |
| 
 | |
|   controller(ApplicationController) do
 | |
|     include RedisTracking
 | |
| 
 | |
|     skip_before_action :authenticate_user!, only: :show
 | |
|     track_redis_hll_event(:index, :show,
 | |
|       name: 'g_compliance_approval_rules',
 | |
|       if: [:custom_condition_one?, :custom_condition_two?]) { |controller| controller.get_custom_id }
 | |
| 
 | |
|     def index
 | |
|       render html: 'index'
 | |
|     end
 | |
| 
 | |
|     def new
 | |
|       render html: 'new'
 | |
|     end
 | |
| 
 | |
|     def show
 | |
|       render html: 'show'
 | |
|     end
 | |
| 
 | |
|     def get_custom_id
 | |
|       'some_custom_id'
 | |
|     end
 | |
| 
 | |
|     private
 | |
| 
 | |
|     def custom_condition_one?
 | |
|       true
 | |
|     end
 | |
| 
 | |
|     def custom_condition_two?
 | |
|       true
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def expect_tracking
 | |
|     expect(Gitlab::UsageDataCounters::HLLRedisCounter).to receive(:track_event)
 | |
|       .with('g_compliance_approval_rules', values: instance_of(String))
 | |
|   end
 | |
| 
 | |
|   def expect_no_tracking
 | |
|     expect(Gitlab::UsageDataCounters::HLLRedisCounter).not_to receive(:track_event)
 | |
|   end
 | |
| 
 | |
|   context 'when user is logged in' do
 | |
|     before do
 | |
|       sign_in(user)
 | |
|     end
 | |
| 
 | |
|     it 'tracks the event' do
 | |
|       expect_tracking
 | |
| 
 | |
|       get :index
 | |
|     end
 | |
| 
 | |
|     it 'tracks the event if DNT is not enabled' do
 | |
|       stub_do_not_track('0')
 | |
| 
 | |
|       expect_tracking
 | |
| 
 | |
|       get :index
 | |
|     end
 | |
| 
 | |
|     it 'does not track the event if DNT is enabled' do
 | |
|       stub_do_not_track('1')
 | |
| 
 | |
|       expect_no_tracking
 | |
| 
 | |
|       get :index
 | |
|     end
 | |
| 
 | |
|     it 'does not track the event if the format is not HTML' do
 | |
|       expect_no_tracking
 | |
| 
 | |
|       get :index, format: :json
 | |
|     end
 | |
| 
 | |
|     it 'does not track the event if a custom condition returns false' do
 | |
|       expect(controller).to receive(:custom_condition_two?).and_return(false)
 | |
| 
 | |
|       expect_no_tracking
 | |
| 
 | |
|       get :index
 | |
|     end
 | |
| 
 | |
|     it 'does not track the event for untracked actions' do
 | |
|       expect_no_tracking
 | |
| 
 | |
|       get :new
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   context 'when user is not logged in' do
 | |
|     let(:visitor_id) { SecureRandom.uuid }
 | |
| 
 | |
|     it 'tracks the event when there is a visitor id' do
 | |
|       cookies[:visitor_id] = { value: visitor_id, expires: 24.months }
 | |
| 
 | |
|       expect_tracking
 | |
| 
 | |
|       get :show, params: { id: 1 }
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   context 'when user is not logged in and there is no visitor_id' do
 | |
|     it 'does not track the event' do
 | |
|       expect_no_tracking
 | |
| 
 | |
|       get :index
 | |
|     end
 | |
| 
 | |
|     it 'tracks the event when there is custom id' do
 | |
|       expect_tracking
 | |
| 
 | |
|       get :show, params: { id: 1 }
 | |
|     end
 | |
| 
 | |
|     it 'does not track the event when there is no custom id' do
 | |
|       expect(controller).to receive(:get_custom_id).and_return(nil)
 | |
| 
 | |
|       expect_no_tracking
 | |
| 
 | |
|       get :show, params: { id: 2 }
 | |
|     end
 | |
|   end
 | |
| end
 |