Source Code for Module lxml.tests.test_errors

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