sidekiq 3.4.1 → 4.0.2

data/test/test_actors.rb

@@ -0,0 +1,137 @@
+require_relative 'helper'
+require 'sidekiq/cli'
+require 'sidekiq/fetch'
+require 'sidekiq/scheduled'
+require 'sidekiq/processor'
+
+class TestActors < Sidekiq::Test
+  class JoeWorker
+    include Sidekiq::Worker
+    def perform(slp)
+      raise "boom" if slp == "boom"
+      sleep(slp) if slp > 0
+      $count += 1
+    end
+  end
+
+  describe 'threads' do
+    before do
+      Sidekiq.redis {|c| c.flushdb}
+    end
+
+    describe 'scheduler' do
+      it 'can start and stop' do
+        f = Sidekiq::Scheduled::Poller.new
+        f.start
+        f.terminate
+      end
+
+      it 'can schedule' do
+        ss = Sidekiq::ScheduledSet.new
+        q = Sidekiq::Queue.new
+
+        JoeWorker.perform_in(0.01, 0)
+
+        assert_equal 0, q.size
+        assert_equal 1, ss.size
+
+        sleep 0.015
+        s = Sidekiq::Scheduled::Poller.new
+        s.enqueue
+        assert_equal 1, q.size
+        assert_equal 0, ss.size
+        s.terminate
+      end
+    end
+
+    describe 'processor' do
+      before do
+        $count = 0
+      end
+
+      it 'can start and stop' do
+        f = Sidekiq::Processor.new(Mgr.new)
+        f.terminate
+      end
+
+      class Mgr
+        attr_reader :latest_error
+        attr_reader :mutex
+        attr_reader :cond
+        def initialize
+          @mutex = ::Mutex.new
+          @cond = ::ConditionVariable.new
+        end
+        def processor_died(inst, err)
+          @latest_error = err
+          @mutex.synchronize do
+            @cond.signal
+          end
+        end
+        def processor_stopped(inst)
+          @mutex.synchronize do
+            @cond.signal
+          end
+        end
+        def options
+          { :concurrency => 3, :queues => ['default'] }
+        end
+      end
+
+      it 'can process' do
+        mgr = Mgr.new
+
+        p = Sidekiq::Processor.new(mgr)
+        JoeWorker.perform_async(0)
+
+        a = $count
+        p.process_one
+        b = $count
+        assert_equal a + 1, b
+      end
+
+      it 'deals with errors' do
+        mgr = Mgr.new
+
+        p = Sidekiq::Processor.new(mgr)
+        JoeWorker.perform_async("boom")
+        q = Sidekiq::Queue.new
+        assert_equal 1, q.size
+
+        a = $count
+        mgr.mutex.synchronize do
+          p.start
+          mgr.cond.wait(mgr.mutex)
+        end
+        b = $count
+        assert_equal a, b
+
+        sleep 0.001
+        assert_equal false, p.thread.status
+        p.terminate(true)
+        refute_nil mgr.latest_error
+        assert_equal RuntimeError, mgr.latest_error.class
+      end
+
+      it 'gracefully kills' do
+        mgr = Mgr.new
+
+        p = Sidekiq::Processor.new(mgr)
+        JoeWorker.perform_async(1)
+        q = Sidekiq::Queue.new
+        assert_equal 1, q.size
+
+        a = $count
+        p.start
+        sleep(0.02)
+        p.terminate
+        p.kill(true)
+
+        b = $count
+        assert_equal a, b
+        assert_equal false, p.thread.status
+        refute mgr.latest_error, mgr.latest_error.to_s
+      end
+    end
+  end
+end