require File.join(File.dirname(__FILE__), "test_helper")

require "peach"

require 'thread'

class PeachTest < Test::Unit::TestCase
  [:peach, :pmap, :pselect].each do |f|
    context "Parallel function #{f}" do
      normal_f = f.to_s[1..-1].to_sym

      setup do
        @data = [1, 2, 3, 5, 8]*1001
        @block = lambda{|i| i**2}
      end
      should "return the same result as #{normal_f}" do
        assert_equal @data.send(normal_f, &@block),
          @data.send(f, 100, &@block)
      end

      should "return the same result as #{normal_f} for empty list" do
        assert_equal [].send(normal_f, &@block),
          [].send(f, nil, &@block)
      end
    end
  end

  [:peach, :pmap, :pselect].each do |f|
    context "#{f}" do
      [nil, 101, 99, 51, 49, 20, 5, 1].each do |pool|
        should "invoke the block exactly once for each array item," +
          " with thread pool size of #{pool.inspect}" do
          source_array = (0...100).to_a
          q = Queue.new()
          source_array.send(f, pool) {|i| q.push(i)}
          result_array = []
          until q.empty?
            result_array << q.pop
          end
          assert_equal source_array, result_array.sort
        end
      end
    end
  end

end