Skip to content
Snippets Groups Projects
Commit dcefe482 authored by Sarah Yasonik's avatar Sarah Yasonik :two:
Browse files

Assert against expected unique user counter api

parent e9c9f268
No related branches found
No related tags found
1 merge request!148348Assert against expected unique user counter api
......@@ -15,9 +15,6 @@ module HLLRedisCounter
i_analytics_dev_ops_adoption
i_analytics_dev_ops_score
g_project_management_issue_cross_referenced
k8s_api_proxy_requests_unique_users_via_ci_access
k8s_api_proxy_requests_unique_users_via_pat_access
k8s_api_proxy_requests_unique_users_via_user_access
].freeze
EventError = Class.new(StandardError)
......
......@@ -124,15 +124,17 @@ def send_request(headers: {}, params: {})
}
users = create_list(:user, 3)
user_ids = users.map(&:id) << users[0].id
unique_counters = {
k8s_api_proxy_requests_unique_users_via_ci_access: user_ids,
unique_value_counters = {
k8s_api_proxy_requests_unique_agents_via_ci_access: user_ids,
k8s_api_proxy_requests_unique_users_via_user_access: user_ids,
k8s_api_proxy_requests_unique_agents_via_user_access: user_ids,
k8s_api_proxy_requests_unique_users_via_pat_access: user_ids,
k8s_api_proxy_requests_unique_agents_via_pat_access: user_ids,
flux_git_push_notified_unique_projects: user_ids
}
unique_user_counters = {
k8s_api_proxy_requests_unique_users_via_ci_access: user_ids,
k8s_api_proxy_requests_unique_users_via_user_access: user_ids,
k8s_api_proxy_requests_unique_users_via_pat_access: user_ids
}
expected_counters = {
kubernetes_agent_gitops_sync: request_count * counters[:gitops_sync],
kubernetes_agent_k8s_api_proxy_request: request_count * counters[:k8s_api_proxy_request],
......@@ -143,12 +145,17 @@ def send_request(headers: {}, params: {})
}
request_count.times do
send_request(params: { counters: counters, unique_counters: unique_counters })
send_request(
params: {
counters: counters,
unique_counters: { **unique_value_counters, **unique_user_counters }
}
)
end
expect(Gitlab::UsageDataCounters::KubernetesAgentCounter.totals).to eq(expected_counters)
unique_counters.each do |c, xs|
unique_value_counters.each do |c, xs|
expect(
Gitlab::UsageDataCounters::HLLRedisCounter
.unique_events(
......@@ -157,6 +164,17 @@ def send_request(headers: {}, params: {})
)
).to eq(xs.uniq.count)
end
unique_user_counters.each do |c, xs|
expect(
Gitlab::UsageDataCounters::HLLRedisCounter
.unique_events(
event_names: c.to_s,
start_date: Date.current, end_date: Date.current + 10,
property_name: :user
)
).to eq(xs.uniq.count)
end
end
end
end
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment