cb193cd6b5
- Retry connection when it fails - Properly shutdown daemon - Stop monitor if the Exception is raised - Properly guard exception handling
123 lines
3.1 KiB
Ruby
123 lines
3.1 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'spec_helper'
|
|
|
|
describe Gitlab::Daemon do
|
|
subject { described_class.new }
|
|
|
|
before do
|
|
allow(subject).to receive(:start_working)
|
|
allow(subject).to receive(:stop_working)
|
|
end
|
|
|
|
describe '.instance' do
|
|
before do
|
|
allow(Kernel).to receive(:at_exit)
|
|
end
|
|
|
|
after do
|
|
described_class.instance_variable_set(:@instance, nil)
|
|
end
|
|
|
|
it 'provides instance of Daemon' do
|
|
expect(described_class.instance).to be_instance_of(described_class)
|
|
end
|
|
|
|
it 'subsequent invocations provide the same instance' do
|
|
expect(described_class.instance).to eq(described_class.instance)
|
|
end
|
|
|
|
it 'creates at_exit hook when instance is created' do
|
|
expect(described_class.instance).not_to be_nil
|
|
|
|
expect(Kernel).to have_received(:at_exit)
|
|
end
|
|
end
|
|
|
|
context 'when Daemon is enabled' do
|
|
before do
|
|
allow(subject).to receive(:enabled?).and_return(true)
|
|
end
|
|
|
|
context 'when Daemon is stopped' do
|
|
describe '#start' do
|
|
it 'starts the Daemon' do
|
|
expect { subject.start.join }.to change { subject.thread? }.from(false).to(true)
|
|
|
|
expect(subject).to have_received(:start_working)
|
|
end
|
|
end
|
|
|
|
describe '#stop' do
|
|
it "doesn't shutdown stopped Daemon" do
|
|
expect { subject.stop }.not_to change { subject.thread? }
|
|
|
|
expect(subject).not_to have_received(:start_working)
|
|
end
|
|
end
|
|
end
|
|
|
|
context 'when Daemon is running' do
|
|
before do
|
|
subject.start
|
|
end
|
|
|
|
describe '#start' do
|
|
it "doesn't start running Daemon" do
|
|
expect { subject.start.join }.not_to change { subject.thread }
|
|
|
|
expect(subject).to have_received(:start_working).once
|
|
end
|
|
end
|
|
|
|
describe '#stop' do
|
|
it 'shutdowns Daemon' do
|
|
expect { subject.stop }.to change { subject.thread? }.from(true).to(false)
|
|
|
|
expect(subject).to have_received(:stop_working)
|
|
end
|
|
|
|
context 'when stop_working raises exception' do
|
|
before do
|
|
allow(subject).to receive(:start_working) do
|
|
sleep(1000)
|
|
end
|
|
end
|
|
|
|
it 'shutdowns Daemon' do
|
|
expect(subject).to receive(:stop_working) do
|
|
subject.thread.raise(Interrupt)
|
|
end
|
|
|
|
expect(subject.thread).to be_alive
|
|
expect { subject.stop }.not_to raise_error
|
|
expect(subject.thread).to be_nil
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
context 'when Daemon is disabled' do
|
|
before do
|
|
allow(subject).to receive(:enabled?).and_return(false)
|
|
end
|
|
|
|
describe '#start' do
|
|
it "doesn't start working" do
|
|
expect(subject.start).to be_nil
|
|
expect { subject.start }.not_to change { subject.thread? }
|
|
|
|
expect(subject).not_to have_received(:start_working)
|
|
end
|
|
end
|
|
|
|
describe '#stop' do
|
|
it "doesn't stop working" do
|
|
expect { subject.stop }.not_to change { subject.thread? }
|
|
|
|
expect(subject).not_to have_received(:stop_working)
|
|
end
|
|
end
|
|
end
|
|
end
|