import sys
import unittest


class TestSAXImporting(unittest.TestCase):

    def setUp(self):
        self._modules = sys.modules.copy()

    def tearDown(self):
        sys.modules = self._modules

    def test_xml_exceptions(self):
        from xml.sax import SAXException, expatreader, xmlreader

        self.assert_(issubclass(expatreader.SAXParseException, SAXException))
        self.assert_(issubclass(xmlreader.SAXNotSupportedException, SAXException))

    def test_xml_expatreader(self):
        import xml.sax.expatreader
        self.assert_(hasattr(xml.sax, "expatreader"))

    def test_xml_subclasses(self):
        from xml.sax import ContentHandler
        from xml.sax.expatreader import ExpatLocator, ExpatParser
        from xml.sax.saxutils import XMLGenerator, XMLFilterBase
        from xml.sax.xmlreader import XMLReader, Locator

        self.assert_(issubclass(ExpatLocator, Locator))
        self.assert_(issubclass(ExpatParser, Locator))
        self.assert_(issubclass(ExpatParser, XMLReader))
        self.assert_(issubclass(XMLGenerator, ContentHandler))
        self.assert_(issubclass(XMLFilterBase, XMLReader))

    def test_xmlcore_exceptions(self):
        from xmlcore.sax import SAXException, expatreader, xmlreader

        self.assert_(issubclass(expatreader.SAXParseException, SAXException))
        self.assert_(issubclass(xmlreader.SAXNotSupportedException, SAXException))

    def test_xmlcore_subclasses(self):
        from xmlcore.sax import ContentHandler
        from xmlcore.sax.expatreader import ExpatLocator, ExpatParser
        from xmlcore.sax.saxutils import XMLGenerator, XMLFilterBase
        from xmlcore.sax.xmlreader import XMLReader, Locator

        self.assert_(issubclass(ExpatLocator, Locator))
        self.assert_(issubclass(ExpatParser, Locator))
        self.assert_(issubclass(ExpatParser, XMLReader))
        self.assert_(issubclass(XMLGenerator, ContentHandler))
        self.assert_(issubclass(XMLFilterBase, XMLReader))


if __name__ == "__main__":
    unittest.main()