1 files changed, 62 insertions, 0 deletions
diff --git a/test/test-exception-py2.py b/test/test-exception-py2.py
new file mode 100644
index 0000000..02cf93d
--- /dev/null
+++ b/test/test-exception-py2.py
@@ -0,0 +1,62 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+
+import sys
+import unittest
+
+import dbus
+
+# from test-service.py
+class ServiceError(dbus.DBusException):
+    """Exception representing a normal "environmental" error"""
+    include_traceback = False
+    _dbus_error_name = 'com.example.Networking.ServerError'
+
+
+class DBusExceptionTestCase(unittest.TestCase):
+    """Test the DBusException str/unicode behavior with py2"""
+
+    def test_dbus_exception_normal(self):
+        """Test the normal Exception behavior"""
+        e = dbus.exceptions.DBusException("baaa")
+        msg = e.get_dbus_message()
+        self.assertEqual(msg, u"baaa")
+
+    def test_dbus_exception_unicode(self):
+        """Test that DBusExceptions that take a py2 unicode work"""
+        e = dbus.exceptions.DBusException(u"bäää")
+        msg = e.get_dbus_message()
+        self.assertEqual(msg, u"bäää")
+
+    def test_dbus_exception_convert_str(self):
+        """Test that converting a DBusException to str() works as expected"""
+        e = dbus.exceptions.DBusException(u"bxxx")
+        self.assertEqual(str(e), "bxxx")
+
+    def test_dbus_exception_convert_str_fail(self):
+        """Test that a non-ascii Exception fails to convert to str"""
+        if sys.getdefaultencoding() == 'ascii':
+            self.assertRaises(UnicodeEncodeError,
+                    lambda: str(dbus.exceptions.DBusException(u"bä")))
+        else:
+            self.skipTest("you're using a weird non-ascii "
+                    "sys.getdefaultencoding()")
+
+    def test_dbus_exception_convert_unicode(self):
+        """Test that converting a DBusEception to unicode works"""
+        e = dbus.exceptions.DBusException(u"bäää")
+        self.assertEqual(e.get_dbus_message(), u"bäää")
+        self.assertEqual(e.__unicode__(), u"bäää")
+        self.assertEqual(unicode(e), u"bäää")
+
+    def test_subclass_exception_unicode(self):
+        """Test that DBusExceptions that take a py2 unicode work"""
+        e = ServiceError(u"bäää")
+        msg = e.get_dbus_message()
+        self.assertEqual(msg, u"bäää")
+        self.assertEqual(
+            unicode(e), u"com.example.Networking.ServerError: bäää")
+
+
+if __name__ == "__main__":
+    unittest.main()