95 lines
2.7 KiB
Ruby
95 lines
2.7 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
require 'spec_helper'
|
||
|
|
|
||
|
|
describe Environments::AutoStopService, :clean_gitlab_redis_shared_state do
|
||
|
|
include CreateEnvironmentsHelpers
|
||
|
|
include ExclusiveLeaseHelpers
|
||
|
|
|
||
|
|
let_it_be(:project) { create(:project, :repository) }
|
||
|
|
let_it_be(:user) { create(:user) }
|
||
|
|
let(:service) { described_class.new }
|
||
|
|
|
||
|
|
before_all do
|
||
|
|
project.add_developer(user)
|
||
|
|
end
|
||
|
|
|
||
|
|
describe '#execute' do
|
||
|
|
subject { service.execute }
|
||
|
|
|
||
|
|
let_it_be(:project) { create(:project, :repository) }
|
||
|
|
let_it_be(:user) { create(:user) }
|
||
|
|
let(:environments) { Environment.all }
|
||
|
|
|
||
|
|
before_all do
|
||
|
|
project.add_developer(user)
|
||
|
|
project.repository.add_branch(user, 'review/feature-1', 'master')
|
||
|
|
project.repository.add_branch(user, 'review/feature-2', 'master')
|
||
|
|
end
|
||
|
|
|
||
|
|
before do
|
||
|
|
create_review_app(user, project, 'review/feature-1')
|
||
|
|
create_review_app(user, project, 'review/feature-2')
|
||
|
|
end
|
||
|
|
|
||
|
|
it 'stops environments and play stop jobs' do
|
||
|
|
expect { subject }
|
||
|
|
.to change { Environment.all.map(&:state).uniq }
|
||
|
|
.from(['available']).to(['stopped'])
|
||
|
|
|
||
|
|
expect(Ci::Build.where(name: 'stop_review_app').map(&:status).uniq).to eq(['pending'])
|
||
|
|
end
|
||
|
|
|
||
|
|
context 'when auto_stop_environments feature flag is disabled' do
|
||
|
|
before do
|
||
|
|
stub_feature_flags(auto_stop_environments: false)
|
||
|
|
end
|
||
|
|
|
||
|
|
it 'does not execute Ci::StopEnvironmentsService' do
|
||
|
|
expect(Ci::StopEnvironmentsService).not_to receive(:execute_in_batch)
|
||
|
|
|
||
|
|
subject
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
context 'when the other sidekiq worker has already been running' do
|
||
|
|
before do
|
||
|
|
stub_exclusive_lease_taken(described_class::EXCLUSIVE_LOCK_KEY)
|
||
|
|
end
|
||
|
|
|
||
|
|
it 'does not execute stop_in_batch' do
|
||
|
|
expect_next_instance_of(described_class) do |service|
|
||
|
|
expect(service).not_to receive(:stop_in_batch)
|
||
|
|
end
|
||
|
|
|
||
|
|
expect { subject }.to raise_error(Gitlab::ExclusiveLeaseHelpers::FailedToObtainLockError)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
context 'when loop reached timeout' do
|
||
|
|
before do
|
||
|
|
stub_const("#{described_class}::LOOP_TIMEOUT", 0.seconds)
|
||
|
|
stub_const("#{described_class}::LOOP_LIMIT", 100_000)
|
||
|
|
allow_next_instance_of(described_class) do |service|
|
||
|
|
allow(service).to receive(:stop_in_batch) { true }
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
it 'returns false and does not continue the process' do
|
||
|
|
is_expected.to eq(false)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
context 'when loop reached loop limit' do
|
||
|
|
before do
|
||
|
|
stub_const("#{described_class}::LOOP_LIMIT", 1)
|
||
|
|
stub_const("#{described_class}::BATCH_SIZE", 1)
|
||
|
|
end
|
||
|
|
|
||
|
|
it 'stops only one available environment' do
|
||
|
|
expect { subject }.to change { Environment.available.count }.by(-1)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|