Source Code for Module lxml.tests.test_errors

 1  # -*- coding: utf-8 -*- 
 2  import unittest, doctest 
 3   
 4  # These tests check that error handling in the Pyrex code is 
 5  # complete. 
 6  # It is likely that if there are errors, instead of failing the code 
 7  # will simply crash. 
 8   
 9  import sys, gc, os.path 
10  from lxml import etree 
11   
12  this_dir = os.path.dirname(__file__) 
13  if this_dir not in sys.path: 
14      sys.path.insert(0, this_dir) # needed for Py3 
15   
16  from common_imports import HelperTestCase 
17   
18   
19 -class ErrorTestCase(HelperTestCase): 
20      etree = etree 
21   
22 -    def test_bad_element(self): 
23          # attrib argument of Element() should be a dictionary, so if 
24          # we pass a string we should get an error. 
25          self.assertRaises(TypeError, self.etree.Element, 'a', 'b') 
26   
27 -    def test_empty_parse(self): 
28          self.assertRaises(etree.XMLSyntaxError, etree.fromstring, '') 
29   
30 -    def test_element_cyclic_gc_none(self): 
31          # test if cyclic reference can crash etree 
32          Element = self.etree.Element 
33          getrefcount = sys.getrefcount 
34   
35          # must disable tracing as it could change the refcounts 
36          trace_func = sys.gettrace() 
37          try: 
38              sys.settrace(None) 
39              gc.collect() 
40   
41              count = getrefcount(None) 
42   
43              l = [Element('name'), Element('name')] 
44              l.append(l) 
45   
46              del l 
47              gc.collect() 
48              count = getrefcount(None) - count 
49   
50              self.assertEqual(count, 0) 
51          finally: 
52              sys.settrace(trace_func) 
53   
54 -    def test_xmlsyntaxerror_has_info(self): 
55          broken_xml_name = 'test_broken.xml' 
56          broken_xml_path = os.path.join(this_dir, broken_xml_name) 
57          fail_msg = 'test_broken.xml should raise an etree.XMLSyntaxError' 
58          try: 
59              etree.parse(broken_xml_path) 
60          except etree.XMLSyntaxError as e: 
61              # invariant 
62              self.assertEqual(e.position, (e.lineno, e.offset + 1), 'position and lineno/offset out of sync') 
63              # SyntaxError info derived from file & contents 
64              self.assertTrue(e.filename.endswith(broken_xml_name), 'filename must be preserved') 
65              self.assertEqual(e.lineno, 1) 
66              self.assertEqual(e.offset, 10) 
67          except Exception as e: 
68              self.fail('{0}, not {1}'.format(fail_msg, type(e))) 
69          else: 
70              self.fail('test_broken.xml should raise an etree.XMLSyntaxError') 
71   
72   
73 -def test_suite(): 
74      suite = unittest.TestSuite() 
75      suite.addTests([unittest.makeSuite(ErrorTestCase)]) 
76      return suite 
77   
78  if __name__ == '__main__': 
79      print('to test use test.py %s' % __file__) 
80