Add Enumerable#group_by and Array#in_groups_of

git-svn-id: http://svn-commit.rubyonrails.org/rails/trunk@3726 5ecf4fe2-1ee6-0310-87b1-e25e094e27de

2 files changed, 47 insertions, 1 deletions

diff --git a/activesupport/test/core_ext/array_ext_test.rb b/activesupport/test/core_ext/array_ext_test.rb
index 159dc2a478..9e23eed450 100644
--- a/activesupport/test/core_ext/array_ext_test.rb
+++ b/activesupport/test/core_ext/array_ext_test.rb
@@ -34,3 +34,34 @@ class ArrayExtConversionTests < Test::Unit::TestCase
     assert_equal "one", ['one'].to_sentence
   end
 end
+
+class ArrayExtGroupingTests < Test::Unit::TestCase
+  def test_group_by_with_perfect_fit
+    groups = []
+    ('a'..'i').to_a.in_groups_of(3) do |group|
+      groups << group
+    end
+
+    assert_equal [%w(a b c), %w(d e f), %w(g h i)], groups
+  end
+
+  def test_group_by_with_padding
+    groups = []
+    ('a'..'g').to_a.in_groups_of(3) do |group|
+      groups << group
+    end
+
+    assert_equal [%w(a b c), %w(d e f), ['g', nil, nil]], groups
+  end
+
+  def test_group_by_pads_with_specified_values
+    groups = []
+
+    ('a'..'g').to_a.in_groups_of(3, false) do |group|
+      groups << group
+    end
+
+    assert_equal [%w(a b c), %w(d e f), ['g', false, false]], groups
+  end
+
+end
diff --git a/activesupport/test/core_ext/enumerable_test.rb b/activesupport/test/core_ext/enumerable_test.rb
index 1fcaa9163e..6ca41f9116 100644
--- a/activesupport/test/core_ext/enumerable_test.rb
+++ b/activesupport/test/core_ext/enumerable_test.rb
@@ -12,4 +12,19 @@ class EnumerableTests < Test::Unit::TestCase
     assert_equal true, (1..10).first_match {|x| x > 9}
     assert_equal :aba, {:a => 10, :aba => 50, :bac => 40}.first_match {|k, v| k if v > 45}
   end
-end
\ No newline at end of file
+
+  def test_group_by
+    names = %w(marcel sam david jeremy)
+    klass = Class.new
+    klass.send(:attr_accessor, :name)
+    objects = (1..50).inject([]) do |people,| 
+      p = klass.new
+      p.name = names.sort_by { rand }.first
+      people << p
+    end
+
+    objects.group_by {|object| object.name}.each do |name, group|
+      assert group.all? {|person| person.name == name}
+    end
+  end
+end