require File.join(File.dirname(__FILE__), "test_helper")

class TestLibXMLTestHelper < Test::Unit::TestCase
  include LibXMLTestHelper
  #testing the convenience methods!
  def test_remove_nodes
    input_xml=<<-END
      <list>
        <food type='dairy' name='milk'>
          <cost>32</cost>
        </food>
        <thing2></thing2>
        <food type='other' name='food2'>
        </food>
      </list>
    END

    #test remove only first node
    expected_xml=<<-END
      <list>
        <thing2></thing2>
        <food type='other' name='food2'>
        </food>
      </list>
    END
   assert_equal expected_xml.gsub(/\n\W*/, ''), remove_first_node_called('food', input_xml).gsub(/\n\W*/, '')

    #test remove all nodes
    expected_xml=<<-END
      <list>
        <thing2></thing2>
      </list>
    END
   assert_equal expected_xml.gsub(/\n\W*/, ''), remove_nodes_called('food', input_xml).gsub(/\n\W*/, '')

  end




  #testing the convenience methods!
  def test_replace_nodes
    input_xml=<<-END
      <list>
        <food type='dairy' name='milk'>
          <cost>32</cost>
        </food>
        <thing2></thing2>
        <food type='other' name='food2'>
        </food>
      </list>
    END


    #test replace only first node
    expected_xml=<<-END
      <list>
        <snack name='lions'></snack><snack name='tigers'></snack>
        <thing2></thing2>
        <food type='other' name='food2'>
        </food>
      </list>
    END
    replacement_string = "<snack name='lions'></snack><snack name='tigers'></snack>"

    output = replace_nodes_called('food', input_xml, replacement_string, true)
    assert_equal expected_xml.gsub(/\n\W*/, ''), output.gsub(/\n\W*/, '')

    output = replace_first_node_called('food', input_xml, replacement_string)
    assert_equal expected_xml.gsub(/\n\W*/, ''), output.gsub(/\n\W*/, '')

    #test replace all matching nodes
    expected_xml=<<-END
      <list>
        <snack name='lions'></snack><snack name='tigers'></snack>
        <thing2></thing2>
        <snack name='lions'></snack><snack name='tigers'></snack>
      </list>
    END
    output = replace_nodes_called('food', input_xml, replacement_string, false)
    assert_equal expected_xml.gsub(/\n\W*/, ''), output.gsub(/\n\W*/, '')

  end
end