import six

from lxml import etree
from six import binary_type, string_types, text_type


def load_xml(xml):
    parser = etree.XMLParser(remove_blank_text=True, remove_comments=True)
    return etree.fromstring(xml.strip(), parser=parser)


def assert_nodes_equal(node_1, node_2):
    def _convert_node(node):
        if isinstance(node, (string_types, binary_type)):
            return load_xml(node)
        return node

    # assert node_1 == node_2
    text_1 = etree.tostring(_convert_node(node_1), pretty_print=True)
    text_2 = etree.tostring(_convert_node(node_2), pretty_print=True)

    if six.PY3:
        text_1 = text_1.decode('utf-8')
        text_2 = text_2.decode('utf-8')
    assert text_1 == text_2


def render_node(element, value):
    node = etree.Element('document')
    element.render(node, value)
    return node


class DummyTransport(object):
    def __init__(self):
        self._items = {}

    def bind(self, url, node):
        self._items[url] = node

    def load(self, url):
        data = self._items[url]
        if isinstance(data, string_types):
            return data
        return etree.tostring(data)
