# encoding: UTF-8

require_relative './test_helper'

class TestNode < Minitest::Test
  def setup
    @file_name = "model/bands.utf-8.xml"

    # Strip spaces to make testing easier
    file = File.join(File.dirname(__FILE__), @file_name)
    @doc = LibXML::XML::Document.file(file, options: LibXML::XML::Parser::Options::NOBLANKS)
  end
  
  def teardown
    @doc = nil
  end

  def nodes
    # Find all nodes with a country attributes
    @doc.find('*[@country]')
  end

  def test_doc_class
    assert_instance_of(LibXML::XML::Document, @doc)
  end

  def test_doc_node_type
    assert_equal LibXML::XML::Node::DOCUMENT_NODE, @doc.node_type
  end

  def test_root_class
    assert_instance_of(LibXML::XML::Node, @doc.root)
  end

  def test_root_node_type
    assert_equal LibXML::XML::Node::ELEMENT_NODE, @doc.root.node_type
  end

  def test_node_class
    for n in nodes
      assert_instance_of(LibXML::XML::Node, n)
    end
  end

  def test_context
    node = @doc.root
    context = node.context
    assert_instance_of(LibXML::XML::XPath::Context, context)
  end

  def test_find
    assert_instance_of(LibXML::XML::XPath::Object, self.nodes)
  end

  def test_node_child_get
    assert_instance_of(TrueClass, @doc.root.child?)
    assert_instance_of(LibXML::XML::Node, @doc.root.child)

    assert_equal(Encoding::UTF_8, @doc.root.child.name.encoding)
    assert_equal("m\u00F6tley_cr\u00FCe", @doc.root.child.name)
  end

  def test_node_doc
    for n in nodes
      assert_instance_of(LibXML::XML::Document, n.doc) if n.document?
    end
  end

  def test_name
    node = @doc.root.children.last
    assert_equal("iron_maiden", node.name)
  end

  def test_node_find
    nodes = @doc.root.find('./fixnum')
    for node in nodes
      assert_instance_of(LibXML::XML::Node, node)
    end
  end

  def test_equality
    node_a = @doc.find_first('*[@country]')
    node_b = @doc.root.child

    # On the ruby side these are different objects
    refute(node_a.equal?(node_b))

    # But they are the same underlying libxml node so specify they are equal
    assert(node_a == node_b)
    assert(node_a.eql?(node_b))

    file = File.join(File.dirname(__FILE__), @file_name)
    doc2 = LibXML::XML::Document.file(file)

    node_a2 = doc2.find_first('*[@country]')

    refute(node_a == node_a2)
    refute(node_a.eql?(node_a2))
    assert_equal(node_a.to_s, node_a2.to_s)
    refute(node_a.equal?(node_a2))
  end

  def test_equality_2
    parent = LibXML::XML::Node.new('parent')
    child = LibXML::XML::Node.new('child')
    parent << child

    node_a = child.parent
    node_b = child.parent
    # In this case the nodes are equal - the parent being the root
    assert(node_a.equal?(node_b))
    assert(node_a == node_b)
    assert(node_a.eql?(node_b))
  end

  def test_equality_nil
    node = @doc.root
    assert(node != nil)
  end

  def test_equality_wrong_type
    node = @doc.root

    assert_raises(TypeError) do
      assert(node != 'abc')
    end
  end

  def test_content
    node = @doc.root.last
    assert_equal("Iron Maiden is a British heavy metal band formed in 1975.",
                 node.content)
  end

  def test_base
    doc = LibXML::XML::Parser.string('<person />').parse
    assert_nil(doc.root.base_uri)
  end

  # We use the same facility that libXSLT does here to disable output escaping.
  # This lets you specify that the node's content should be rendered unaltered
  # whenever it is being output.  This is useful for things like <script> and
  # <style> nodes in HTML documents if you don't want to be forced to wrap them
  # in CDATA nodes.  Or if you are sanitizing existing HTML documents and want
  # to preserve the content of any of the text nodes.
  #
  def test_output_escaping
		text = '<bad-script>if (a &lt; b || b &gt; c) { return "text"; }<stop/>return "&gt;&gt;&gt;snip&lt;&lt;&lt;";</bad-script>'
    node = LibXML::XML::Parser.string(text).parse.root
		assert_equal text, node.to_s

		text_noenc = '<bad-script>if (a < b || b > c) { return "text"; }<stop/>return ">>>snip<<<";</bad-script>'
		node.output_escaping = false
		assert_equal text_noenc, node.to_s

		node.output_escaping = true
		assert_equal text, node.to_s

		node.output_escaping = nil
		assert_equal text_noenc, node.to_s

		node.output_escaping = true
		assert_equal text, node.to_s
  end

  # Just a sanity check for output escaping.
  def test_output_escaping_sanity
		text = '<bad-script>if (a &lt; b || b &gt; c) { return "text"; }<stop/>return "&gt;&gt;&gt;snip&lt;&lt;&lt;";</bad-script>'
    node = LibXML::XML::Parser.string(text).parse.root
		affected = node.find('//text()')

		check_escaping = lambda do |flag|
			assert_equal('bad-script', node.name)
			assert_equal(flag, node.output_escaping?)
			affected.each do |x|
				assert_equal(flag ? 'text' : 'textnoenc', x.name)
				assert_equal(flag, x.output_escaping?)
			end
		end

		node.output_escaping = false
		check_escaping[false]

		node.output_escaping = true
		check_escaping[true]

		node.output_escaping = nil
		check_escaping[false]

		node.output_escaping = true
		check_escaping[true]

		affected.first.output_escaping = true
		affected.last.output_escaping = false
		assert node.output_escaping?.nil?
  end

  def test_space_preserve
    node = @doc.root

    node.space_preserve = false
    assert_equal LibXML::XML::Node::SPACE_DEFAULT, node.space_preserve

    node.space_preserve = true
    assert_equal LibXML::XML::Node::SPACE_PRESERVE, node.space_preserve
  end

  def test_empty
    text = '<name> </name>'
    doc = LibXML::XML::Parser.string(text).parse

    node = doc.root
    assert(!node.empty?)

    text_node = node.first
    assert(text_node.empty?)
  end

  def test_save_no_empty_tags
    node = LibXML::XML::Node.new('test')
    assert_equal '<test/>', node.to_s

    LibXML::XML.default_save_no_empty_tags = true
    assert_equal '<test></test>', node.to_s

    LibXML::XML.default_save_no_empty_tags = false
  end

  def test_set_content
    node = LibXML::XML::Node.new('test')
    node.content = "unescaped & string"
    assert_equal("unescaped & string", node.content)
    assert_equal("<test>unescaped &amp; string</test>", node.to_s)
  end

  def test_document_node_marks_document
    GC.stress = true
    node = create_document_child
    assert_equal("child", node.name)
    refute_nil(node.doc)
  ensure
    GC.stress = false
  end

  private

  def create_document_child
    doc = LibXML::XML::Document.string("<root><child/></root>")
    doc.root.children.first
  end
end
