Source Code for Module lxml.tests.test_dtd

  1  # -*- coding: utf-8 -*- 
  2   
  3  """ 
  4  Test cases related to DTD parsing and validation 
  5  """ 
  6   
  7  import unittest 
  8   
  9  from common_imports import etree, StringIO, doctest 
 10  from common_imports import HelperTestCase, fileInTestDir 
 11   
 12 -class ETreeDtdTestCase(HelperTestCase): 
 13 -    def test_dtd(self): 
 14          pass 
 15   
 16 -    def test_dtd_file(self): 
 17          parse = etree.parse 
 18          tree = parse(fileInTestDir("test.xml")) 
 19          root = tree.getroot() 
 20   
 21          dtd = etree.DTD(fileInTestDir("test.dtd")) 
 22          self.assert_(dtd.validate(root)) 
 23   
 24 -    def test_dtd_stringio(self): 
 25          root = etree.XML("<b/>") 
 26          dtd = etree.DTD(StringIO("<!ELEMENT b EMPTY>")) 
 27          self.assert_(dtd.validate(root)) 
 28   
 29 -    def test_dtd_parse_invalid(self): 
 30          fromstring = etree.fromstring 
 31          parser = etree.XMLParser(dtd_validation=True) 
 32          xml = '<!DOCTYPE b SYSTEM "%s"><b><a/></b>' % fileInTestDir("test.dtd") 
 33          self.assertRaises(etree.XMLSyntaxError, 
 34                            fromstring, xml, parser=parser) 
 35   
 36 -    def test_dtd_parse_file_not_found(self): 
 37          fromstring = etree.fromstring 
 38          dtd_filename = fileInTestDir("__nosuch.dtd") 
 39          parser = etree.XMLParser(dtd_validation=True) 
 40          xml = '<!DOCTYPE b SYSTEM "%s"><b><a/></b>' % dtd_filename 
 41          self.assertRaises(etree.XMLSyntaxError, 
 42                            fromstring, xml, parser=parser) 
 43          errors = None 
 44          try: 
 45              fromstring(xml, parser=parser) 
 46          except etree.XMLSyntaxError, e: 
 47              errors = [ entry.message for entry in e.error_log 
 48                         if dtd_filename in entry.message ] 
 49          self.assert_(errors) 
 50   
 51 -    def test_dtd_parse_valid(self): 
 52          parser = etree.XMLParser(dtd_validation=True) 
 53          xml = '<!DOCTYPE a SYSTEM "%s"><a><b/></a>' % fileInTestDir("test.dtd") 
 54          root = etree.fromstring(xml, parser=parser) 
 55   
 56 -    def test_dtd_parse_valid_relative(self): 
 57          parser = etree.XMLParser(dtd_validation=True) 
 58          xml = '<!DOCTYPE a SYSTEM "test.dtd"><a><b/></a>' 
 59          root = etree.fromstring(xml, parser=parser, 
 60                                  base_url=fileInTestDir("test.xml")) 
 61   
 62 -    def test_dtd_invalid(self): 
 63          root = etree.XML("<b><a/></b>") 
 64          dtd = etree.DTD(StringIO("<!ELEMENT b EMPTY>")) 
 65          self.assertRaises(etree.DocumentInvalid, dtd.assertValid, root) 
 66   
 67 -    def test_dtd_assertValid(self): 
 68          root = etree.XML("<b><a/></b>") 
 69          dtd = etree.DTD(StringIO("<!ELEMENT b (a)><!ELEMENT a EMPTY>")) 
 70          dtd.assertValid(root) 
 71   
 72 -    def test_dtd_internal(self): 
 73          root = etree.XML(''' 
 74          <!DOCTYPE b SYSTEM "none" [ 
 75          <!ELEMENT b (a)> 
 76          <!ELEMENT a EMPTY> 
 77          ]> 
 78          <b><a/></b> 
 79          ''') 
 80          dtd = etree.ElementTree(root).docinfo.internalDTD 
 81          self.assert_(dtd) 
 82          dtd.assertValid(root) 
 83   
 84 -    def test_dtd_internal_invalid(self): 
 85          root = etree.XML(''' 
 86          <!DOCTYPE b SYSTEM "none" [ 
 87          <!ELEMENT b (a)> 
 88          <!ELEMENT a (c)> 
 89          <!ELEMENT c EMPTY> 
 90          ]> 
 91          <b><a/></b> 
 92          ''') 
 93          dtd = etree.ElementTree(root).docinfo.internalDTD 
 94          self.assert_(dtd) 
 95          self.assertFalse(dtd.validate(root)) 
 96   
 97 -    def test_dtd_broken(self): 
 98          self.assertRaises(etree.DTDParseError, etree.DTD, 
 99                            StringIO("<!ELEMENT b HONKEY>")) 
100   
101 -    def test_parse_file_dtd(self): 
102          parser = etree.XMLParser(attribute_defaults=True) 
103   
104          tree = etree.parse(fileInTestDir('test.xml'), parser) 
105          root = tree.getroot() 
106   
107          self.assertEquals( 
108              "valueA", 
109              root.get("default")) 
110          self.assertEquals( 
111              "valueB", 
112              root[0].get("default")) 
113   
114 -    def test_iterparse_file_dtd(self): 
115          iterparse = etree.iterparse 
116          iterator = iterparse(fileInTestDir("test.xml"), events=("start",), 
117                               attribute_defaults=True) 
118          attributes = [ element.get("default") 
119                         for event, element in iterator ] 
120          self.assertEquals( 
121              ["valueA", "valueB"], 
122              attributes) 
123   
124   
125 -def test_suite(): 
126      suite = unittest.TestSuite() 
127      suite.addTests([unittest.makeSuite(ETreeDtdTestCase)]) 
128      suite.addTests( 
129          [doctest.DocFileSuite('../../../doc/validation.txt')]) 
130      return suite 
131   
132  if __name__ == '__main__': 
133      print 'to test use test.py %s' % __file__ 
134