rails--rails/activesupport/test/xml_mini/common.rb

module CommonXMLMiniAdapterTest
  def setup
    @default_backend = ActiveSupport::XmlMini.backend
    ActiveSupport::XmlMini.backend = adapter_name
  end

  def teardown
    ActiveSupport::XmlMini.backend = @default_backend
  end

  def test_file_from_xml
    hash = Hash.from_xml(<<-eoxml)
    <blog>
      <logo type="file" name="logo.png" content_type="image/png">
      </logo>
    </blog>
  eoxml
    assert hash.has_key?("blog")
    assert hash["blog"].has_key?("logo")

    file = hash["blog"]["logo"]
    assert_equal "logo.png", file.original_filename
    assert_equal "image/png", file.content_type
  end

  def test_exception_thrown_on_expansion_attack
    assert_raise expansion_attack_error do
      attack_xml = <<-EOT
    <?xml version="1.0" encoding="UTF-8"?>
    <!DOCTYPE member [
      <!ENTITY a "&b;&b;&b;&b;&b;&b;&b;&b;&b;&b;">
      <!ENTITY b "&c;&c;&c;&c;&c;&c;&c;&c;&c;&c;">
      <!ENTITY c "&d;&d;&d;&d;&d;&d;&d;&d;&d;&d;">
      <!ENTITY d "&e;&e;&e;&e;&e;&e;&e;&e;&e;&e;">
      <!ENTITY e "&f;&f;&f;&f;&f;&f;&f;&f;&f;&f;">
      <!ENTITY f "&g;&g;&g;&g;&g;&g;&g;&g;&g;&g;">
      <!ENTITY g "xxxxxxxxxxxxxxxxxxxxxxxxxxxxxx">
    ]>
    <member>
    &a;
    </member>
    EOT
      Hash.from_xml(attack_xml)
    end
  end

  def test_setting_backend
    ActiveSupport::XmlMini.backend = adapter_name
    assert_equal ActiveSupport.const_get("XmlMini_#{adapter_name}"), ActiveSupport::XmlMini.backend
    #assert_equal ActiveSupport::XmlMini_Nokogiri
  end

  def test_blank_returns_empty_hash
    assert_equal({}, ActiveSupport::XmlMini.parse(nil))
    assert_equal({}, ActiveSupport::XmlMini.parse(""))
  end

  def test_array_type_makes_an_array
    assert_equal_rexml(<<-eoxml)
    <blog>
      <posts type="array">
        <post>a post</post>
        <post>another post</post>
      </posts>
    </blog>
  eoxml
  end

  def test_one_node_document_as_hash
    assert_equal_rexml(<<-eoxml)
    <products/>
    eoxml
  end

  def test_one_node_with_attributes_document_as_hash
    assert_equal_rexml(<<-eoxml)
    <products foo="bar"/>
    eoxml
  end

  def test_products_node_with_book_node_as_hash
    assert_equal_rexml(<<-eoxml)
  <products>
    <book name="awesome" id="12345" />
  </products>
  eoxml
  end

  def test_products_node_with_two_book_nodes_as_hash
    assert_equal_rexml(<<-eoxml)
  <products>
    <book name="awesome" id="12345" />
    <book name="america" id="67890" />
  </products>
  eoxml
  end

  def test_single_node_with_content_as_hash
    assert_equal_rexml(<<-eoxml)
    <products>
      hello world
    </products>
  eoxml
  end

  def test_children_with_children
    assert_equal_rexml(<<-eoxml)
  <root>
    <products>
      <book name="america" id="67890" />
    </products>
  </root>
  eoxml
  end

  def test_children_with_text
    assert_equal_rexml(<<-eoxml)
  <root>
    <products>
      hello everyone
    </products>
  </root>
  eoxml
  end

  def test_children_with_non_adjacent_text
    assert_equal_rexml(<<-eoxml)
  <root>
    good
    <products>
      hello everyone
    </products>
    morning
  </root>
  eoxml
  end

  def test_parse_from_io
    jdom_skip

    io = StringIO.new(<<-eoxml)
  <root>
    good
    <products>
      hello everyone
    </products>
    morning
  </root>
  eoxml
    assert_equal_rexml(io)
  end

  def test_children_with_simple_cdata
    jdom_skip

    assert_equal_rexml(<<-eoxml)
  <root>
    <products>
       <![CDATA[cdatablock]]>
    </products>
  </root>
  eoxml
  end

  def test_children_with_multiple_cdata
    jdom_skip

    assert_equal_rexml(<<-eoxml)
  <root>
    <products>
       <![CDATA[cdatablock1]]><![CDATA[cdatablock2]]>
    </products>
  </root>
  eoxml
  end

  def test_children_with_text_and_cdata
    jdom_skip

    assert_equal_rexml(<<-eoxml)
  <root>
    <products>
      hello <![CDATA[cdatablock]]>
      morning
    </products>
  </root>
  eoxml
  end

  def test_children_with_blank_text
    jdom_skip

    assert_equal_rexml(<<-eoxml)
  <root>
    <products>   </products>
  </root>
  eoxml
  end

  def test_children_with_blank_text_and_attribute
    jdom_skip

    assert_equal_rexml(<<-eoxml)
  <root>
    <products type="file">   </products>
  </root>
  eoxml
  end

  private
    def adapter_name
      raise NotImplementedError
    end

    def assert_equal_rexml(xml)
      parsed_xml = ActiveSupport::XmlMini.parse(xml)
      xml.rewind if xml.respond_to?(:rewind)
      hash = ActiveSupport::XmlMini.with_backend("REXML") { ActiveSupport::XmlMini.parse(xml) }
      assert_equal(hash, parsed_xml)
    end

    def expansion_attack_error
      raise NotImplementedError
    end

    def jdom_skip
      skip "JDOM does not support this test" if adapter_name == "JDOM"
    end
end