Land rapid7#6267, fix Rex::Parser::Ini#each_group

Author: Brent Cook

lib/rex/parser/ini.rb

@@ -49,14 +49,7 @@ def initialize(path = nil)
     end
   end
 
-  #
-  # Enumerates the groups hash keys.
-  #
-  def each_group(&block)
-    self.keys.each { |k|
-      yield
-    }
-  end
+  alias each_group each_key
 
   #
   # Adds a group of the supplied name if it doesn't already exist.

spec/lib/rex/parser/ini_spec.rb

@@ -0,0 +1,34 @@
+require 'rex/parser/ini'
+
+describe Rex::Parser::Ini do
+  let(:ini_contents) { <<EOF
+# global comment
+global settting = blah
+[foo]
+a = b
+[bar]
+b = c
+
+[baf]
+c = d
+EOF
+  }
+
+  let(:ini) { described_class.from_s(ini_contents) }
+
+  context "#each_group" do
+    it "enumerates the groups" do
+      groups = []
+      ini.each_group { |group| groups << group }
+      groups.should eq(%w(foo bar baf))
+    end
+  end
+
+  context "#each_key" do
+    it "enumerates the groups" do
+      groups = []
+      ini.each_key.map { |group| groups << group }
+      groups.should eq(%w(foo bar baf))
+    end
+  end
+end