diff options
Diffstat (limited to 'spec/lib/gitlab/usage_data_counters/kubernetes_agent_counter_spec.rb')
-rw-r--r-- | spec/lib/gitlab/usage_data_counters/kubernetes_agent_counter_spec.rb | 56 |
1 files changed, 46 insertions, 10 deletions
diff --git a/spec/lib/gitlab/usage_data_counters/kubernetes_agent_counter_spec.rb b/spec/lib/gitlab/usage_data_counters/kubernetes_agent_counter_spec.rb index 8f9a3e0cd9e..e7edb8b9cf1 100644 --- a/spec/lib/gitlab/usage_data_counters/kubernetes_agent_counter_spec.rb +++ b/spec/lib/gitlab/usage_data_counters/kubernetes_agent_counter_spec.rb @@ -3,21 +3,57 @@ require 'spec_helper' RSpec.describe Gitlab::UsageDataCounters::KubernetesAgentCounter do - it_behaves_like 'a redis usage counter', 'Kubernetes Agent', :gitops_sync + described_class::KNOWN_EVENTS.each do |event| + it_behaves_like 'a redis usage counter', 'Kubernetes Agent', event + it_behaves_like 'a redis usage counter with totals', :kubernetes_agent, event => 1 + end + + describe '.increment_event_counts' do + let(:events) do + { + 'gitops_sync' => 1, + 'k8s_api_proxy_request' => 2 + } + end - it_behaves_like 'a redis usage counter with totals', :kubernetes_agent, gitops_sync: 1 + subject { described_class.increment_event_counts(events) } - describe '.increment_gitops_sync' do - it 'increments the gtops_sync counter by the new increment amount' do - described_class.increment_gitops_sync(7) - described_class.increment_gitops_sync(2) - described_class.increment_gitops_sync(0) + it 'increments the specified counters by the new increment amount' do + described_class.increment_event_counts(events) + described_class.increment_event_counts(events) + described_class.increment_event_counts(events) - expect(described_class.totals).to eq(kubernetes_agent_gitops_sync: 9) + expect(described_class.totals).to eq(kubernetes_agent_gitops_sync: 3, kubernetes_agent_k8s_api_proxy_request: 6) end - it 'raises for negative numbers' do - expect { described_class.increment_gitops_sync(-1) }.to raise_error(ArgumentError) + context 'event is unknown' do + let(:events) do + { + 'gitops_sync' => 1, + 'other_event' => 2 + } + end + + it 'raises an ArgumentError' do + expect(described_class).not_to receive(:increment_by) + + expect { subject }.to raise_error(ArgumentError, 'unknown event other_event') + end + end + + context 'increment is negative' do + let(:events) do + { + 'gitops_sync' => -1, + 'k8s_api_proxy_request' => 2 + } + end + + it 'raises an ArgumentError' do + expect(described_class).not_to receive(:increment_by) + + expect { subject }.to raise_error(ArgumentError, 'gitops_sync count must be greater than or equal to zero') + end end end end |