require_relative 'test_helper'

describe Adrian::Dispatcher do
  before do
    $done_items = []
    @q = Adrian::ArrayQueue.new
    @dispatcher = Adrian::Dispatcher.new(:stop_when_done => true)
  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

        def perform
          $done_items << [@boss, @item.value]
        end

        def report_to(boss)
          @boss = boss
        end
      end

      @q.push(1)
      @q.push(2)
      @q.push(3)

      @dispatcher.start(@q, worker)

      $done_items.must_equal([[@dispatcher, 1], [@dispatcher, 2], [@dispatcher, 3]])
    end
  end

  describe "work evaluation" do
    it "should use the failure handler to handle the result" do
      @dispatcher.on_failure(RuntimeError) do |item, worker, exception|
        @q.push(item)
      end

      @dispatcher.work_done(1, nil)
      @q.pop.must_be_nil

      @dispatcher.work_done(1, nil, nil)
      @q.pop.must_be_nil

      @dispatcher.work_done(1, nil, RuntimeError.new)
      @q.pop.value.must_equal 1
    end
  end
end