From c36073b40e2caef337511fa3a4e386dce5449bd2 Mon Sep 17 00:00:00 2001
From: Lance Stout <lancestout@gmail.com>
Date: Sun, 19 Feb 2012 20:27:19 -0800
Subject: xml.etree.ElementTree raises ExpatError instead of SyntaxError or
 ParseError.

---
 sleekxmpp/test/sleektest.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

(limited to 'sleekxmpp/test')

diff --git a/sleekxmpp/test/sleektest.py b/sleekxmpp/test/sleektest.py
index dd3df29a..364e5939 100644
--- a/sleekxmpp/test/sleektest.py
+++ b/sleekxmpp/test/sleektest.py
@@ -7,6 +7,7 @@
 """
 
 import unittest
+from xml.parsers.expat import ExpatError
 try:
     import Queue as queue
 except:
@@ -62,8 +63,9 @@ class SleekTest(unittest.TestCase):
         try:
             xml = ET.fromstring(xml_string)
             return xml
-        except SyntaxError as e:
-            if 'unbound' in e.msg:
+        except (SyntaxError, ExpatError) as e:
+            msg = e.msg if hasattr(e, 'msg') else e.message
+            if 'unbound' in msg:
                 known_prefixes = {
                         'stream': 'http://etherx.jabber.org/streams'}
 
-- 
cgit v1.2.3