2 files changed, 70 insertions, 0 deletions
diff --git a/spec/ruby/library/rexml/shared/each_element.rb b/spec/ruby/library/rexml/shared/each_element.rb
new file mode 100644
index 0000000000..1cb79c8d3a
--- /dev/null
+++ b/spec/ruby/library/rexml/shared/each_element.rb
@@ -0,0 +1,36 @@
+require 'rexml/document'
+require File.expand_path('../../../../spec_helper', __FILE__)
+
+describe :rexml_each_element, shared: true do
+  before :each do
+    @e = REXML::Element.new "root"
+    s1 = REXML::Element.new "node1"
+    s2 = REXML::Element.new "node2"
+    s3 = REXML::Element.new "node3"
+    s4 = REXML::Element.new "sub_node"
+    @e << s1
+    @e << s2
+    @e << s3
+    @e << s4
+  end
+
+  it "iterates through element" do
+    str = ""
+      @e.send(@method) { |elem| str << elem.name << " " }
+    str.should == "node1 node2 node3 sub_node "
+  end
+
+  it "iterates through element filtering with XPath" do
+    str = ""
+     @e.send(@method, "/*"){ |e| str << e.name << " "}
+     str.should == "node1 node2 node3 sub_node "
+  end
+end
+
+describe "REXML::Element#each_element" do
+ it_behaves_like :rexml_each_element, :each_element
+end
+
+describe "REXML::Elements#each" do
+  it_behaves_like :rexml_each_element, :each
+end
diff --git a/spec/ruby/library/rexml/shared/elements_to_a.rb b/spec/ruby/library/rexml/shared/elements_to_a.rb
new file mode 100644
index 0000000000..6299c628c3
--- /dev/null
+++ b/spec/ruby/library/rexml/shared/elements_to_a.rb
@@ -0,0 +1,34 @@
+require 'rexml/document'
+require File.expand_path('../../../../spec_helper', __FILE__)
+
+describe :rexml_elements_to_a, shared: true do
+  before :each do
+    @e = REXML::Element.new "root"
+    @first = REXML::Element.new("FirstChild")
+    @second = REXML::Element.new("SecondChild")
+    @e << @first
+    @e << @second
+  end
+
+  it "returns elements that match xpath" do
+    @e.elements.send(@method, "FirstChild").first.should == @first
+  end
+
+  # According to the docs REXML::Element#get_elements is an alias for
+  # REXML::Elements.to_a. Implementation wise there's a difference, get_elements
+  # always needs the first param (even if it's nil).
+  # A patch was submitted:
+  # http://rubyforge.org/tracker/index.php?func=detail&aid=19354&group_id=426&atid=1698
+  it "returns all childs if xpath is nil" do
+    @e.elements.send(@method).should == [@first, @second]
+  end
+
+end
+
+describe "REXML::REXML::Elements#to_a" do
+  it_behaves_like :rexml_elements_to_a, :to_a
+end
+
+describe "REXML::REXML::Element#get_elements" do
+  it_behaves_like :rexml_elements_to_a, :get_elements
+end