require 'rexml/document'

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
