import support
import StringIO
import xml.sax
import new
import unittest

PREFERRED_XML_PARSERS = ["drv_libxml2"]

from html5lib.treebuilders import dom
from html5lib.liberalxmlparser import *

class SAXLogger: 
  def __init__(self):
    self.log = []
  def setDocumentLocator(self, locator):
    pass
  def startElement(self, name, attrs):
    self.log.append(['startElement', name, dict(attrs.items())])
  def startElementNS(self, name, qname, attrs):
    if qname is None:
      qname = name[1]
    self.log.append(['startElementNS', name, qname, dict(attrs.items())])
  def endElementNS(self, name, qname):
    if qname is None:
      qname = name[1]
    self.log.append(['endElementNS', name, qname])
  def __getattr__(self, name):
    def function(self, *args): self.log.append([name]+list(args))
    return new.instancemethod(function, self, SAXLogger)

class SAXTest(unittest.TestCase):
  def DOMParse(self, input):
    return XMLParser(tree=dom.TreeBuilder).parse(input)

  def setNS(self, saxparser):
    import xml.dom
    saxparser.setFeature(xml.sax.handler.feature_namespaces, 1)
    return {'xml':xml.dom.XML_NAMESPACE}

  def saxdiff(self, input):
    domhandler = SAXLogger()

    saxhandler = SAXLogger()
    saxparser = xml.sax.make_parser(PREFERRED_XML_PARSERS)

    dom.dom2sax(self.DOMParse(input), domhandler, self.setNS(saxparser))

    saxparser.setContentHandler(saxhandler)
    source = xml.sax.xmlreader.InputSource()
    source.setByteStream(StringIO.StringIO(input))
    saxparser.parse(source)

    for i in range(0,len(saxhandler.log)):
      if i > len(domhandler.log):
        self.assertEqual(saxhandler.log[i:], domhandler.log[i:])
      elif saxhandler.log[i] != domhandler.log[i]:
        self.assertEqual(saxhandler.log[i], domhandler.log[i])
    else:
      self.assertEquals(saxhandler.log, domhandler.log)

  def test_nodes(self):
    self.saxdiff(''x' +
      '')

  def test_xmllang(self):
    self.saxdiff(''
      "foo")

  def test_ns(self):
    self.saxdiff(
'' +
'XLINK' +
'' +
  '' +
    '' +
      '' +
        '' +
        '' +
      '' +
      '' +
      '' +
      '' +
    '' +
    '' +
      '' +
      '' +
      '' +

      '' +
      '' +
      '' +
    '' +
  '' +
'')

# Repeat tests without namespace support
class nonamespaceTest(SAXTest):
  def setNS(self, saxparser):
    return None

# Redundantly rerun all tests using the "real" minidom parser, just to be
# sure that the output is consistent
class minidomTest(SAXTest):
  def DOMParse(self, input):
    return xml.dom.minidom.parseString(input)

def buildTestSuite():
  return unittest.defaultTestLoader.loadTestsFromName(__name__)

def main():
    buildTestSuite()
    unittest.main()

if __name__ == '__main__':
    main()