3ef4f74b1a
This adds counters for build artifacts and LFS objects, and moves the preexisting repository_size and commit_count from the projects table into a new project_statistics table. The counters are displayed in the administration area for projects and groups, and also available through the API for admins (on */all) and normal users (on */owned) The statistics are updated through ProjectCacheWorker, which can now do more granular updates with the new :statistics argument.
78 lines
2.1 KiB
Ruby
78 lines
2.1 KiB
Ruby
require 'spec_helper'
|
|
|
|
describe ProjectCacheWorker do
|
|
let(:worker) { described_class.new }
|
|
let(:project) { create(:project) }
|
|
let(:statistics) { project.statistics }
|
|
|
|
describe '#perform' do
|
|
before do
|
|
allow_any_instance_of(Gitlab::ExclusiveLease).to receive(:try_obtain).
|
|
and_return(true)
|
|
end
|
|
|
|
context 'with a non-existing project' do
|
|
it 'does nothing' do
|
|
expect(worker).not_to receive(:update_statistics)
|
|
|
|
worker.perform(-1)
|
|
end
|
|
end
|
|
|
|
context 'with an existing project without a repository' do
|
|
it 'does nothing' do
|
|
allow_any_instance_of(Repository).to receive(:exists?).and_return(false)
|
|
|
|
expect(worker).not_to receive(:update_statistics)
|
|
|
|
worker.perform(project.id)
|
|
end
|
|
end
|
|
|
|
context 'with an existing project' do
|
|
it 'updates the project statistics' do
|
|
expect(worker).to receive(:update_statistics)
|
|
.with(kind_of(Project), %i(repository_size))
|
|
.and_call_original
|
|
|
|
worker.perform(project.id, [], %w(repository_size))
|
|
end
|
|
|
|
it 'refreshes the method caches' do
|
|
expect_any_instance_of(Repository).to receive(:refresh_method_caches).
|
|
with(%i(readme)).
|
|
and_call_original
|
|
|
|
worker.perform(project.id, %w(readme))
|
|
end
|
|
end
|
|
end
|
|
|
|
describe '#update_statistics' do
|
|
context 'when a lease could not be obtained' do
|
|
it 'does not update the repository size' do
|
|
allow(worker).to receive(:try_obtain_lease_for).
|
|
with(project.id, :update_statistics).
|
|
and_return(false)
|
|
|
|
expect(statistics).not_to receive(:refresh!)
|
|
|
|
worker.update_statistics(project)
|
|
end
|
|
end
|
|
|
|
context 'when a lease could be obtained' do
|
|
it 'updates the project statistics' do
|
|
allow(worker).to receive(:try_obtain_lease_for).
|
|
with(project.id, :update_statistics).
|
|
and_return(true)
|
|
|
|
expect(statistics).to receive(:refresh!)
|
|
.with(only: %i(repository_size))
|
|
.and_call_original
|
|
|
|
worker.update_statistics(project, %i(repository_size))
|
|
end
|
|
end
|
|
end
|
|
end
|