138 lines
3.5 KiB
Ruby
138 lines
3.5 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'rake_helper'
|
|
|
|
RSpec.describe 'gitlab:db:lock_writes', :reestablished_active_record_base, feature_category: :cell do
|
|
before :all do
|
|
Rake.application.rake_require 'active_record/railties/databases'
|
|
Rake.application.rake_require 'tasks/seed_fu'
|
|
Rake.application.rake_require 'tasks/gitlab/db/validate_config'
|
|
Rake.application.rake_require 'tasks/gitlab/db/lock_writes'
|
|
|
|
# empty task as env is already loaded
|
|
Rake::Task.define_task :environment
|
|
end
|
|
|
|
let(:table_locker) { instance_double(Gitlab::Database::TablesLocker) }
|
|
let(:logger) { instance_double(Logger, level: nil) }
|
|
let(:dry_run) { false }
|
|
let(:verbose) { false }
|
|
|
|
before do
|
|
allow(Logger).to receive(:new).with($stdout).and_return(logger)
|
|
allow(Gitlab::Database::TablesLocker).to receive(:new).with(
|
|
logger: logger, dry_run: dry_run
|
|
).and_return(table_locker)
|
|
end
|
|
|
|
shared_examples "call table locker" do |method|
|
|
let(:log_level) { verbose ? Logger::INFO : Logger::WARN }
|
|
|
|
it "creates TablesLocker with dry run set and calls #{method}" do
|
|
expect(logger).to receive(:level=).with(log_level)
|
|
expect(table_locker).to receive(method)
|
|
|
|
run_rake_task("gitlab:db:#{method}")
|
|
end
|
|
end
|
|
|
|
describe 'lock_writes' do
|
|
context 'when environment sets DRY_RUN to true' do
|
|
let(:dry_run) { true }
|
|
|
|
before do
|
|
stub_env('DRY_RUN', 'true')
|
|
end
|
|
|
|
include_examples "call table locker", :lock_writes
|
|
end
|
|
|
|
context 'when environment sets DRY_RUN to false' do
|
|
let(:dry_run) { false }
|
|
|
|
before do
|
|
stub_env('DRY_RUN', 'false')
|
|
end
|
|
|
|
include_examples "call table locker", :lock_writes
|
|
end
|
|
|
|
context 'when environment does not define DRY_RUN' do
|
|
let(:dry_run) { false }
|
|
|
|
include_examples "call table locker", :lock_writes
|
|
end
|
|
|
|
context 'when environment sets VERBOSE to true' do
|
|
let(:verbose) { true }
|
|
|
|
before do
|
|
stub_env('VERBOSE', 'true')
|
|
end
|
|
|
|
include_examples "call table locker", :lock_writes
|
|
end
|
|
|
|
context 'when environment sets VERBOSE to false' do
|
|
let(:verbose) { false }
|
|
|
|
before do
|
|
stub_env('VERBOSE', 'false')
|
|
end
|
|
|
|
include_examples "call table locker", :lock_writes
|
|
end
|
|
|
|
context 'when environment does not define VERBOSE' do
|
|
include_examples "call table locker", :lock_writes
|
|
end
|
|
end
|
|
|
|
describe 'unlock_writes' do
|
|
context 'when environment sets DRY_RUN to true' do
|
|
let(:dry_run) { true }
|
|
|
|
before do
|
|
stub_env('DRY_RUN', 'true')
|
|
end
|
|
|
|
include_examples "call table locker", :unlock_writes
|
|
end
|
|
|
|
context 'when environment sets DRY_RUN to false' do
|
|
before do
|
|
stub_env('DRY_RUN', 'false')
|
|
end
|
|
|
|
include_examples "call table locker", :unlock_writes
|
|
end
|
|
|
|
context 'when environment does not define DRY_RUN' do
|
|
include_examples "call table locker", :unlock_writes
|
|
end
|
|
|
|
context 'when environment sets VERBOSE to true' do
|
|
let(:verbose) { true }
|
|
|
|
before do
|
|
stub_env('VERBOSE', 'true')
|
|
end
|
|
|
|
include_examples "call table locker", :lock_writes
|
|
end
|
|
|
|
context 'when environment sets VERBOSE to false' do
|
|
let(:verbose) { false }
|
|
|
|
before do
|
|
stub_env('VERBOSE', 'false')
|
|
end
|
|
|
|
include_examples "call table locker", :lock_writes
|
|
end
|
|
|
|
context 'when environment does not define VERBOSE' do
|
|
include_examples "call table locker", :lock_writes
|
|
end
|
|
end
|
|
end
|