36 lines
1 KiB
Ruby
36 lines
1 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
||
|
require 'spec_helper'
|
||
|
|
||
|
describe Namespaces::PruneAggregationSchedulesWorker, '#perform', :clean_gitlab_redis_shared_state do
|
||
|
include ExclusiveLeaseHelpers
|
||
|
|
||
|
let(:namespaces) { create_list(:namespace, 5, :with_aggregation_schedule) }
|
||
|
let(:timeout) { Namespace::AggregationSchedule::DEFAULT_LEASE_TIMEOUT }
|
||
|
|
||
|
subject(:worker) { described_class.new }
|
||
|
|
||
|
before do
|
||
|
allow(Namespaces::RootStatisticsWorker)
|
||
|
.to receive(:perform_async).and_return(nil)
|
||
|
|
||
|
allow(Namespaces::RootStatisticsWorker)
|
||
|
.to receive(:perform_in).and_return(nil)
|
||
|
|
||
|
namespaces.each do |namespace|
|
||
|
lease_key = "namespace:namespaces_root_statistics:#{namespace.id}"
|
||
|
stub_exclusive_lease(lease_key, timeout: timeout)
|
||
|
end
|
||
|
end
|
||
|
|
||
|
it 'schedules a worker per pending aggregation' do
|
||
|
expect(Namespaces::RootStatisticsWorker)
|
||
|
.to receive(:perform_async).exactly(5).times
|
||
|
|
||
|
expect(Namespaces::RootStatisticsWorker)
|
||
|
.to receive(:perform_in).exactly(5).times
|
||
|
|
||
|
worker.perform
|
||
|
end
|
||
|
end
|