cmlenz@1: # -*- coding: utf-8 -*- cmlenz@1: # cmlenz@66: # Copyright (C) 2006 Edgewall Software cmlenz@1: # All rights reserved. cmlenz@1: # cmlenz@1: # This software is licensed as described in the file COPYING, which cmlenz@1: # you should have received as part of this distribution. The terms cmlenz@66: # are also available at http://markup.edgewall.org/wiki/License. cmlenz@1: # cmlenz@1: # This software consists of voluntary contributions made by many cmlenz@1: # individuals. For the exact contribution history, see the revision cmlenz@66: # history and logs, available at http://markup.edgewall.org/log/. cmlenz@1: cmlenz@1: import doctest cmlenz@1: import unittest cmlenz@1: import sys cmlenz@1: cmlenz@85: from markup.core import Stream cmlenz@85: from markup.output import DocType, XMLSerializer cmlenz@85: cmlenz@85: cmlenz@85: class XMLSerializerTestCase(unittest.TestCase): cmlenz@85: cmlenz@85: def test_doctype_in_stream(self): cmlenz@85: stream = Stream([(Stream.DOCTYPE, DocType.HTML_STRICT, ('?', -1, -1))]) cmlenz@85: output = stream.render(XMLSerializer) cmlenz@85: self.assertEqual('\n', cmlenz@85: output) cmlenz@85: cmlenz@85: def test_doctype_in_stream_no_sysid(self): cmlenz@85: stream = Stream([(Stream.DOCTYPE, cmlenz@85: ('html', '-//W3C//DTD HTML 4.01//EN', None), cmlenz@85: ('?', -1, -1))]) cmlenz@85: output = stream.render(XMLSerializer) cmlenz@85: self.assertEqual('\n', cmlenz@85: output) cmlenz@85: cmlenz@85: def test_doctype_in_stream_no_pubid(self): cmlenz@85: stream = Stream([(Stream.DOCTYPE, cmlenz@85: ('html', None, 'http://www.w3.org/TR/html4/strict.dtd'), cmlenz@85: ('?', -1, -1))]) cmlenz@85: output = stream.render(XMLSerializer) cmlenz@85: self.assertEqual('\n', cmlenz@85: output) cmlenz@85: cmlenz@85: def test_doctype_in_stream_no_pubid_or_sysid(self): cmlenz@85: stream = Stream([(Stream.DOCTYPE, ('html', None, None), cmlenz@85: ('?', -1, -1))]) cmlenz@85: output = stream.render(XMLSerializer) cmlenz@85: self.assertEqual('\n', output) cmlenz@85: cmlenz@85: def test_serializer_doctype(self): cmlenz@85: stream = Stream([]) cmlenz@85: output = stream.render(XMLSerializer, doctype=DocType.HTML_STRICT) cmlenz@85: self.assertEqual('\n', cmlenz@85: output) cmlenz@85: cmlenz@85: def test_doctype_one_and_only(self): cmlenz@85: stream = Stream([(Stream.DOCTYPE, ('html', None, None), ('?', -1, -1))]) cmlenz@85: output = stream.render(XMLSerializer, doctype=DocType.HTML_STRICT) cmlenz@85: self.assertEqual('\n', cmlenz@85: output) cmlenz@85: cmlenz@1: cmlenz@1: def suite(): cmlenz@1: suite = unittest.TestSuite() cmlenz@85: suite.addTest(unittest.makeSuite(XMLSerializerTestCase, 'test')) cmlenz@85: suite.addTest(doctest.DocTestSuite(XMLSerializer.__module__)) cmlenz@1: return suite cmlenz@1: cmlenz@1: if __name__ == '__main__': cmlenz@1: unittest.main(defaultTest='suite')