1 files changed, 36 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