138 lines
		
	
	
		
			3.4 KiB
		
	
	
	
		
			Ruby
		
	
	
	
			
		
		
	
	
			138 lines
		
	
	
		
			3.4 KiB
		
	
	
	
		
			Ruby
		
	
	
	
| # frozen_string_literal: true
 | |
| 
 | |
| require 'spec_helper'
 | |
| 
 | |
| RSpec.describe Gitlab::Daemon do
 | |
|   subject { described_class.new }
 | |
| 
 | |
|   before do
 | |
|     allow(subject).to receive(:run_thread)
 | |
|     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(:run_thread)
 | |
|         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(:run_thread)
 | |
|         end
 | |
|       end
 | |
|     end
 | |
| 
 | |
|     describe '#start_working' do
 | |
|       context 'when start_working fails' do
 | |
|         before do
 | |
|           expect(subject).to receive(:start_working) { false }
 | |
|         end
 | |
| 
 | |
|         it 'does not start thread' do
 | |
|           expect(subject).not_to receive(:run_thread)
 | |
| 
 | |
|           expect(subject.start).to eq(nil)
 | |
|         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(:run_thread).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(:run_thread) 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(:run_thread)
 | |
|       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
 |