132 lines
2.6 KiB
Ruby
132 lines
2.6 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require "spec_helper"
|
|
|
|
RSpec.describe RedisTracking do
|
|
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
|
|
request.headers['DNT'] = '0'
|
|
|
|
expect_tracking
|
|
|
|
get :index
|
|
end
|
|
|
|
it 'does not track the event if DNT is enabled' do
|
|
request.headers['DNT'] = '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
|