better unit testing

hivesolutions · Apr 28, 2015 · 3b8d962 · 3b8d962
1 parent fba1b5d
commit 3b8d962
Showing 1 changed file with 15 additions and 10 deletions.
diff --git a/src/appier/test/util.py b/src/appier/test/util.py
@@ -50,14 +50,19 @@ def test_email_parts(self):
         self.assertEqual(name, "João Magalhães")
         self.assertEqual(email, "joamag@hive.pt")
 
-        name, email = appier.email_parts(u"João Magalhães <joamag@hive.pt>")
-        self.assertEqual(type(name), str)
-        self.assertEqual(type(email), str)
-        self.assertEqual(name, "João Magalhães")
-        self.assertEqual(email, "joamag@hive.pt")
+        name, email = appier.email_parts(appier.legacy.u("João Magalhães <joamag@hive.pt>"))
+        self.assertEqual(type(name), appier.legacy.UNICODE)
+        self.assertEqual(type(email), appier.legacy.UNICODE)
+        self.assertEqual(name, appier.legacy.u("João Magalhães"))
+        self.assertEqual(email, appier.legacy.u("joamag@hive.pt"))
 
-        name, email = appier.email_parts(u"你好世界 <hello@hive.pt>")
-        self.assertEqual(type(name), str)
-        self.assertEqual(type(email), str)
-        self.assertEqual(name, "你好世界")
-        self.assertEqual(email, "hello@hive.pt")
+        name, email = appier.email_parts(appier.legacy.u("你好世界 <hello@hive.pt>"))
+        self.assertEqual(type(name), appier.legacy.UNICODE)
+        self.assertEqual(type(email), appier.legacy.UNICODE)
+        self.assertEqual(name, appier.legacy.u("你好世界"))
+        self.assertEqual(email, appier.legacy.u("hello@hive.pt"))
+
+    def test_email_mime(self):
+        result = appier.email_mime("João Magalhães <joamag@hive.pt>")
+        self.assertEqual(type(result), str)
+        self.assertEqual(result, "João Magalhães <joamag@hive.pt>")