diff options
Diffstat (limited to 'spec/workers/gitlab_performance_bar_stats_worker_spec.rb')
-rw-r--r-- | spec/workers/gitlab_performance_bar_stats_worker_spec.rb | 30 |
1 files changed, 30 insertions, 0 deletions
diff --git a/spec/workers/gitlab_performance_bar_stats_worker_spec.rb b/spec/workers/gitlab_performance_bar_stats_worker_spec.rb new file mode 100644 index 00000000000..367003dd1ad --- /dev/null +++ b/spec/workers/gitlab_performance_bar_stats_worker_spec.rb @@ -0,0 +1,30 @@ +# frozen_string_literal: true + +require 'spec_helper' + +RSpec.describe GitlabPerformanceBarStatsWorker do + include ExclusiveLeaseHelpers + + subject(:worker) { described_class.new } + + describe '#perform' do + let(:redis) { double(Gitlab::Redis::SharedState) } + let(:uuid) { 1 } + + before do + expect(Gitlab::Redis::SharedState).to receive(:with).and_yield(redis) + expect_to_cancel_exclusive_lease(GitlabPerformanceBarStatsWorker::LEASE_KEY, uuid) + end + + it 'fetches list of request ids and processes them' do + expect(redis).to receive(:smembers).with(GitlabPerformanceBarStatsWorker::STATS_KEY).and_return([1, 2]) + expect(redis).to receive(:del).with(GitlabPerformanceBarStatsWorker::STATS_KEY) + expect_next_instance_of(Gitlab::PerformanceBar::Stats) do |stats| + expect(stats).to receive(:process).with(1) + expect(stats).to receive(:process).with(2) + end + + worker.perform(uuid) + end + end +end |