RSpec.shared_examples "a Celluloid Group" do let!(:queue) { Queue.new } let!(:busy_queue) { Queue.new } let(:logger) { Specs::FakeLogger.current } def wait_until_busy(busy_queue = nil) return busy_queue.pop if busy_queue Specs.sleep_and_wait_until { subject.busy? } end def wait_until_idle Specs.sleep_and_wait_until { subject.idle? } end before { subject } after do subject.shutdown end it "gets threads" do expect(subject.get { queue.pop }).to be_a Thread queue << nil wait_until_idle end [::StandardError, ::Exception].each do |exception_class| context "with an #{exception_class} in the thread" do before do @wait_queue = Queue.new # doesn't work if in a let() allow(logger).to receive(:crash) subject.get do busy_queue << nil @wait_queue.pop raise exception_class, "Error" end wait_until_busy(busy_queue) end it "logs the crash" do expect(logger).to receive(:crash).with("thread crashed", exception_class) @wait_queue << nil # let the thread fail wait_until_idle end it "puts error'd threads back" do @wait_queue << nil # let the thread fail wait_until_idle expect(subject.idle?).to be_truthy end end end context "when a thread has local variables" do before do @thread = subject.get do Thread.current[:foo] = :bar queue.pop end wait_until_busy queue << nil # let the thread finish if Celluloid.group_class == Celluloid::Group::Pool # Wait until we get the same thread for a different proc Specs.sleep_and_wait_until { subject.get { sleep 0.1 } == @thread } else wait_until_idle end end # Cleaning not necessary for Spawner unless Celluloid.group_class == Celluloid::Group::Spawner it "cleans thread locals from old threads" do expect(@thread[:foo]).to be_nil end end end it "shuts down" do subject thread = Queue.new expect( subject.get do thread << Thread.current sleep end ).to be_a(Celluloid::Thread) thread.pop # wait for 3rd-party thread to get strated expect(subject.active?).to eq true subject.shutdown expect(subject.active?).to eq false expect(subject.group.length).to eq 0 end context "with a dead thread" do before do if Celluloid.group_class == Celluloid::Group::Pool subject.max_idle = 0 # Instruct the pool to immediately shut down the thread. end subject.get { queue.pop } wait_until_busy queue << nil wait_until_idle subject.shutdown if Celluloid.group_class == Celluloid::Group::Spawner end it "doesn't leak dead threads" do expect(subject.to_a.size).to eq(0) end end end