Package lxml :: Package tests :: Module test_incremental_xmlfile |
[hide private]
[frames] | no frames]

# Source Code for Module lxml.tests.test_incremental_xmlfile

      1  # -*- coding: utf-8 -*-
      2
      3  """
      4  Tests for the incremental XML serialisation API.
      5
      6  Tests require Python 2.5 or later.
      7  """
      8
      9  from __future__ import with_statement
     10
     11  import unittest
     12  import tempfile, os, sys
     13
     14  this_dir = os.path.dirname(__file__)
     15  if this_dir not in sys.path:
     16      sys.path.insert(0, this_dir) # needed for Py3
     17
     18  from common_imports import etree, BytesIO
     19  from common_imports import HelperTestCase
     20
     21 -class _XmlFileTestCaseBase(HelperTestCase):
     22      _file = None  # to be set by specific subtypes below
     23
     24 -    def test_element(self):
     25          with etree.xmlfile(self._file) as xf:
     26              with xf.element('test'):
     27                  pass
     28          self.assertXml('<test></test>')
     29
     30 -    def test_element_write_text(self):
     31          with etree.xmlfile(self._file) as xf:
     32              with xf.element('test'):
     33                  xf.write('toast')
     34          self.assertXml('<test>toast</test>')
     35
     36 -    def test_element_nested(self):
     37          with etree.xmlfile(self._file) as xf:
     38              with xf.element('test'):
     39                  with xf.element('toast'):
     40                      with xf.element('taste'):
     41                          xf.write('conTent')
     42          self.assertXml('<test><toast><taste>conTent</taste></toast></test>')
     43
     44 -    def test_element_nested_with_text(self):
     45          with etree.xmlfile(self._file) as xf:
     46              with xf.element('test'):
     47                  xf.write('con')
     48                  with xf.element('toast'):
     49                      xf.write('tent')
     50                      with xf.element('taste'):
     51                          xf.write('inside')
     52                      xf.write('tnet')
     53                  xf.write('noc')
     54          self.assertXml('<test>con<toast>tent<taste>inside</taste>'
     55                         'tnet</toast>noc</test>')
     56
     57 -    def test_write_Element(self):
     58          with etree.xmlfile(self._file) as xf:
     59              xf.write(etree.Element('test'))
     60          self.assertXml('<test/>')
     61
     62 -    def test_write_Element_repeatedly(self):
     63          element = etree.Element('test')
     64          with etree.xmlfile(self._file) as xf:
     65              with xf.element('test'):
     66                  for i in range(100):
     67                      xf.write(element)
     68
     69          tree = self._parse_file()
     70          self.assertTrue(tree is not None)
     71          self.assertEqual(100, len(tree.getroot()))
     72          self.assertEqual(set(['test']), set(el.tag for el in tree.getroot()))
     73
     74 -    def test_namespace_nsmap(self):
     75          with etree.xmlfile(self._file) as xf:
     76              with xf.element('{nsURI}test', nsmap={'x': 'nsURI'}):
     77                  pass
     78          self.assertXml('<x:test xmlns:x="nsURI"></x:test>')
     79
     80 -    def test_namespace_nested_nsmap(self):
     81          with etree.xmlfile(self._file) as xf:
     82              with xf.element('test', nsmap={'x': 'nsURI'}):
     83                  with xf.element('{nsURI}toast'):
     84                      pass
     85          self.assertXml('<test xmlns:x="nsURI"><x:toast></x:toast></test>')
     86
     87 -    def test_anonymous_namespace(self):
     88          with etree.xmlfile(self._file) as xf:
     89              with xf.element('{nsURI}test'):
     90                  pass
     91          self.assertXml('<ns0:test xmlns:ns0="nsURI"></ns0:test>')
     92
     93 -    def test_namespace_nested_anonymous(self):
     94          with etree.xmlfile(self._file) as xf:
     95              with xf.element('test'):
     96                  with xf.element('{nsURI}toast'):
     97                      pass
     98          self.assertXml('<test><ns0:toast xmlns:ns0="nsURI"></ns0:toast></test>')
     99
    100 -    def test_default_namespace(self):
    101          with etree.xmlfile(self._file) as xf:
    102              with xf.element('{nsURI}test', nsmap={None: 'nsURI'}):
    103                  pass
    104          self.assertXml('<test xmlns="nsURI"></test>')
    105
    106 -    def test_nested_default_namespace(self):
    107          with etree.xmlfile(self._file) as xf:
    108              with xf.element('{nsURI}test', nsmap={None: 'nsURI'}):
    109                  with xf.element('{nsURI}toast'):
    110                      pass
    111          self.assertXml('<test xmlns="nsURI"><toast></toast></test>')
    112
    113 -    def test_pi(self):
    114          with etree.xmlfile(self._file) as xf:
    115              xf.write(etree.ProcessingInstruction('pypi'))
    116              with xf.element('test'):
    117                  pass
    118          self.assertXml('<?pypi ?><test></test>')
    119
    120 -    def test_comment(self):
    121          with etree.xmlfile(self._file) as xf:
    122              xf.write(etree.Comment('a comment'))
    123              with xf.element('test'):
    124                  pass
    125          self.assertXml('<!--a comment--><test></test>')
    126
    127 -    def test_attribute(self):
    128          with etree.xmlfile(self._file) as xf:
    129              with xf.element('test', attrib={'k': 'v'}):
    130                  pass
    131          self.assertXml('<test k="v"></test>')
    132
    133 -    def test_escaping(self):
    134          with etree.xmlfile(self._file) as xf:
    135              with xf.element('test'):
    136                  xf.write('Comments: <!-- text -->\n')
    137                  xf.write('Entities: &amp;')
    138          self.assertXml(
    139              '<test>Comments: &lt;!-- text --&gt;\nEntities: &amp;amp;</test>')
    140
    141 -    def test_encoding(self):
    142          with etree.xmlfile(self._file, encoding='utf16') as xf:
    143              with xf.element('test'):
    144                  xf.write('toast')
    145          self.assertXml('<test>toast</test>', encoding='utf16')
    146
    147 -    def test_failure_preceding_text(self):
    148          try:
    149              with etree.xmlfile(self._file) as xf:
    150                  xf.write('toast')
    151          except etree.LxmlSyntaxError:
    152              self.assertTrue(True)
    153          else:
    154              self.assertTrue(False)
    155
    156 -    def test_failure_trailing_text(self):
    157          with etree.xmlfile(self._file) as xf:
    158              with xf.element('test'):
    159                  pass
    160              try:
    161                  xf.write('toast')
    162              except etree.LxmlSyntaxError:
    163                  self.assertTrue(True)
    164              else:
    165                  self.assertTrue(False)
    166
    167 -    def test_failure_trailing_Element(self):
    168          with etree.xmlfile(self._file) as xf:
    169              with xf.element('test'):
    170                  pass
    171              try:
    172                  xf.write(etree.Element('test'))
    173              except etree.LxmlSyntaxError:
    174                  self.assertTrue(True)
    175              else:
    176                  self.assertTrue(False)
    177
    178 -    def _read_file(self):
    179          self._file.seek(0)
    180          return self._file.read()
    181
    182 -    def _parse_file(self):
    183          self._file.seek(0)
    184          return etree.parse(self._file)
    185
    186 -    def tearDown(self):
    187          self._file.close()
    188
    189 -    def assertXml(self, expected, encoding='utf8'):
    190          self.assertEqual(self._read_file().decode(encoding), expected)
    191
    192
    193 -class BytesIOXmlFileTestCase(_XmlFileTestCaseBase):
    194 -    def setUp(self):
    195          self._file = BytesIO()
    196
    197 -class TempXmlFileTestCase(_XmlFileTestCaseBase):
    198 -    def setUp(self):
    199          self._file = tempfile.NamedTemporaryFile()
    200
    201 -class SimpleFileLikeXmlFileTestCase(_XmlFileTestCaseBase):
    202 -    class SimpleFileLike(object):
    203 -        def __init__(self, target):
    204              self._target = target
    205              self.write = target.write
    206              self.close = target.close
    207
    208 -    def setUp(self):
    209          self._target = BytesIO()
    210          self._file = self.SimpleFileLike(self._target)
    211
    212 -    def _read_file(self):
    213          return self._target.getvalue()
    214
    215 -    def _parse_file(self):
    216          self._target.seek(0)
    217          return etree.parse(self._target)
    218
    219
    220 -def test_suite():
    221      suite = unittest.TestSuite()
    222      suite.addTests([unittest.makeSuite(BytesIOXmlFileTestCase),
    223                      unittest.makeSuite(TempXmlFileTestCase),
    224                      unittest.makeSuite(SimpleFileLikeXmlFileTestCase),
    225                      ])
    226      return suite
    227
    228  if __name__ == '__main__':
    229      print('to test use test.py %s' % __file__)
    230