test/dispatcher_test.rb in adrian-1.0.1 vs test/dispatcher_test.rb in adrian-1.1.0

@@ -1,40 +1,56 @@
 require_relative 'test_helper'
 
 describe Adrian::Dispatcher do
   before do
     $done_items = []
-    @q = Adrian::ArrayQueue.new
+    @q          = Adrian::ArrayQueue.new
     @dispatcher = Adrian::Dispatcher.new(:stop_when_done => true)
+    @worker     = TestWorker
   end
 
-  describe "work delegation" do
-    it "should instantiate an instance of the worker for each item and ask it to perform" do
-      worker = Class.new do
-        def initialize(item)
-          @item = item
-        end
+  class TestWorker
 
-        def perform
-          $done_items << [@boss, @item.value]
-        end
+    def initialize(item)
+      @item       = item
+      @done_items = []
+    end
 
-        def report_to(boss)
-          @boss = boss
-        end
-      end
+    def perform
+      $done_items << [@boss, @item.value]
+    end
 
+    def report_to(boss)
+      @boss = boss
+    end
+
+  end
+
+  describe "work delegation" do
+    it "should instantiate an instance of the worker for each item and ask it to perform" do
       @q.push(1)
       @q.push(2)
       @q.push(3)
 
-      @dispatcher.start(@q, worker)
+      @dispatcher.start(@q, @worker)
 
       $done_items.must_equal([[@dispatcher, 1], [@dispatcher, 2], [@dispatcher, 3]])
     end
   end
 
   describe "work evaluation" do
+
+    it "stops when receiving a termination signal" do
+      @dispatcher = Adrian::Dispatcher.new(:stop_when_done => false)
+      dispatch_thread = Thread.new { @dispatcher.start(@q, @worker) }
+      sleep(0.1)
+      assert_equal true, @dispatcher.running
+
+      Process.kill('TERM', Process.pid)
+      assert_equal false, @dispatcher.running
+      dispatch_thread.exit
+    end
+
     it "should use the failure handler to handle the result" do
       @dispatcher.on_failure(RuntimeError) do |item, worker, exception|
         @q.push(item)
       end