153 lines
4.2 KiB
Ruby
153 lines
4.2 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'rails_helper'
|
|
require_relative 'shared_context_for_backup_restore'
|
|
|
|
describe BackupRestore::SystemInterface do
|
|
include_context "shared stuff"
|
|
|
|
subject { BackupRestore::SystemInterface.new(logger) }
|
|
|
|
context "readonly mode" do
|
|
after do
|
|
Discourse::READONLY_KEYS.each { |key| $redis.del(key) }
|
|
end
|
|
|
|
describe "#enable_readonly_mode" do
|
|
it "enables readonly mode" do
|
|
Discourse.expects(:enable_readonly_mode).once
|
|
subject.enable_readonly_mode
|
|
end
|
|
|
|
it "does not enable readonly mode when it is already in readonly mode" do
|
|
Discourse.enable_readonly_mode
|
|
Discourse.expects(:enable_readonly_mode).never
|
|
subject.enable_readonly_mode
|
|
end
|
|
end
|
|
|
|
describe "#disable_readonly_mode" do
|
|
it "disables readonly mode" do
|
|
Discourse.expects(:disable_readonly_mode).once
|
|
subject.disable_readonly_mode
|
|
end
|
|
|
|
it "does not disable readonly mode when readonly mode was explicitly enabled" do
|
|
Discourse.enable_readonly_mode
|
|
Discourse.expects(:disable_readonly_mode).never
|
|
subject.disable_readonly_mode
|
|
end
|
|
end
|
|
end
|
|
|
|
describe "#mark_restore_as_running" do
|
|
it "calls mark_restore_as_running" do
|
|
BackupRestore.expects(:mark_as_running!).once
|
|
subject.mark_restore_as_running
|
|
end
|
|
end
|
|
|
|
describe "#mark_restore_as_not_running" do
|
|
it "calls mark_restore_as_not_running" do
|
|
BackupRestore.expects(:mark_as_not_running!).once
|
|
subject.mark_restore_as_not_running
|
|
end
|
|
end
|
|
|
|
describe "#listen_for_shutdown_signal" do
|
|
before { BackupRestore.mark_as_running! }
|
|
|
|
after do
|
|
BackupRestore.clear_shutdown_signal!
|
|
BackupRestore.mark_as_not_running!
|
|
end
|
|
|
|
it "exits the process when shutdown signal is set" do
|
|
expect do
|
|
thread = subject.listen_for_shutdown_signal
|
|
BackupRestore.set_shutdown_signal!
|
|
thread.join
|
|
end.to raise_error(SystemExit)
|
|
end
|
|
end
|
|
|
|
describe "#pause_sidekiq" do
|
|
it "calls pause!" do
|
|
Sidekiq.expects(:pause!).once
|
|
subject.pause_sidekiq
|
|
end
|
|
end
|
|
|
|
describe "#unpause_sidekiq" do
|
|
it "calls unpause!" do
|
|
Sidekiq.expects(:unpause!).once
|
|
subject.unpause_sidekiq
|
|
end
|
|
end
|
|
|
|
describe "#wait_for_sidekiq" do
|
|
it "waits 6 seconds even when there are no running Sidekiq jobs" do
|
|
subject.expects(:sleep).with(6).once
|
|
subject.wait_for_sidekiq
|
|
end
|
|
|
|
context "with Sidekiq workers" do
|
|
before { $redis.flushall }
|
|
after { $redis.flushall }
|
|
|
|
def create_workers(site_id: nil, all_sites: false)
|
|
$redis.flushall
|
|
|
|
payload = Sidekiq::Testing.fake! do
|
|
data = { post_id: 1 }
|
|
|
|
if all_sites
|
|
data[:all_sites] = true
|
|
else
|
|
data[:current_site_id] = site_id || RailsMultisite::ConnectionManagement.current_db
|
|
end
|
|
|
|
Jobs.enqueue(:process_post, data)
|
|
Jobs::ProcessPost.jobs.last
|
|
end
|
|
|
|
Sidekiq.redis do |conn|
|
|
hostname = "localhost"
|
|
pid = 7890
|
|
key = "#{hostname}:#{pid}"
|
|
process = { pid: pid, hostname: hostname }
|
|
|
|
conn.sadd('processes', key)
|
|
conn.hmset(key, 'info', Sidekiq.dump_json(process))
|
|
|
|
data = Sidekiq.dump_json(
|
|
queue: 'default',
|
|
run_at: Time.now.to_i,
|
|
payload: Sidekiq.dump_json(payload)
|
|
)
|
|
conn.hmset("#{key}:workers", '444', data)
|
|
end
|
|
end
|
|
|
|
it "waits up to 60 seconds for jobs running for the current site to finish" do
|
|
subject.expects(:sleep).with(6).times(10)
|
|
create_workers
|
|
expect { subject.wait_for_sidekiq }.to raise_error(BackupRestore::RunningSidekiqJobsError)
|
|
end
|
|
|
|
it "waits up to 60 seconds for jobs running on all sites to finish" do
|
|
subject.expects(:sleep).with(6).times(10)
|
|
create_workers(all_sites: true)
|
|
expect { subject.wait_for_sidekiq }.to raise_error(BackupRestore::RunningSidekiqJobsError)
|
|
end
|
|
|
|
it "ignores jobs of other sites" do
|
|
subject.expects(:sleep).with(6).once
|
|
create_workers(site_id: "another_site")
|
|
|
|
subject.wait_for_sidekiq
|
|
end
|
|
end
|
|
end
|
|
end
|