########################################################################
# tc_enumerable_extra.rb
#
# Test case for the enumerable-extra library. You should run this
# test via the 'rake test' task.
########################################################################
require 'test/unit'
require 'enumerable/extra'

class TC_Enumerable_Extra < Test::Unit::TestCase
   def setup
      @words   = %w/foo bar baz/
      @numbers = [1,2,3]
      @hash    = {'foo' => 1, 'bar' => 2}
      @array   = []
   end

   def test_version
      assert_equal('0.1.1', Enumerable::EXTRA_VERSION)
   end

   def test_sum
      assert_respond_to(@numbers, :sum)
      assert_equal(6, @numbers.sum)
      assert_equal(20, @numbers.sum(14))
   end

   def test_sum_expected_errors
      assert_raises(TypeError){ @words.sum }
   end

   def test_map_array_no_block
      assert_nothing_raised{ @words.map }
      assert_equal(%w/foo bar baz/, @words.map)
      assert_equal(%w/FOO BAR BAZ/, @words.map(:upcase))
      assert_equal(%w/fooA barA bazA/, @words.map(:+, 'A'))
      assert_equal(%w/foo bar baz/, @words) # Verify receiver unmodified
   end

   # Test the alias explicitly
   def test_collect_array_no_block
      assert_nothing_raised{ @words.collect }
      assert_equal(%w/foo bar baz/, @words.collect)
      assert_equal(%w/FOO BAR BAZ/, @words.collect(:upcase))
      assert_equal(%w/fooA barA bazA/, @words.collect(:+, 'A'))
      assert_equal(%w/foo bar baz/, @words) # Verify receiver unmodified
   end

   def test_map_bang_array_no_block
      assert_nothing_raised{ @words.map! }
      assert_equal(%w/foo bar baz/, @words.map!)
      assert_equal(%w/FOO BAR BAZ/, @words.map!(:upcase))
      assert_equal(%w/FOO BAR BAZ/, @words) # Verify receiver modified
   end

   def test_map_with_block
      assert_nothing_raised{ @words.map{} }
      assert_nothing_raised{ @words.map{ |e| @array << e } }
      assert_equal(%w/foo bar baz/, @array)
      
      @array = []
      assert_nothing_raised{ @words.map(:upcase){ |e| @array << e } }
      assert_equal(%w/FOO BAR BAZ/, @array)

      @array = []
      assert_nothing_raised{ @words.map(:+, 'A'){ |e| @array << e } }
      assert_equal(%w/fooA barA bazA/, @array)
      assert_equal(%w/foo bar baz/, @words) # Verify receiver unmodified
   end

   def teardown
      @words   = nil
      @numbers = nil
      @hash    = nil
      @array   = nil
   end
end