xxx

Author: albu-diku

tests/test_mig_shared_base.py

@@ -33,7 +33,9 @@
 
 from tests.support import PY2, MigTestCase, testmain
 
-from mig.shared.base import force_default_fs_coding_rec, force_default_str_coding_rec, force_utf8
+from mig.shared.base import force_default_fs_coding_rec, \
+    force_default_fs_coding, force_default_str_coding_rec, \
+    force_default_str_coding, force_utf8, force_unicode
 
 DUMMY_BYTECHARS = b'DEADBEEF'
 DUMMY_BYTESRAW = binascii.unhexlify('DEADBEEF') # 4 bytes
@@ -56,46 +58,66 @@ def test_encode_within_a_dict(self):
     def test_encode_within_a_list(self):
         output = force_default_fs_coding_rec(['foo', 'bar', 'baz'])
 
-        self.assertEqual(output, [b'foo', b'bar', b'baz'])
+        truth = [force_default_str_coding('foo'),
+                 force_default_str_coding('bar'),
+                 force_default_str_coding('baz')]
+
+        self.assertEqual(output, truth)
 
     def test_encode_within_a_tuple(self):
         output = force_default_fs_coding_rec(('foo', 'bar', 'baz'))
+        truth = (force_default_str_coding('foo'),
+                 force_default_str_coding('bar'),
+                 force_default_str_coding('baz'))
 
-        self.assertEqual(output, (b'foo', b'bar', b'baz'))
+        self.assertEqual(output, truth)
 
+    def test_encode_within_a_tuple_bytes(self):
+        output = force_default_fs_coding_rec(('foo', 'bar', 'baz'))
+        truth = (force_default_str_coding('foo'),
+                 force_default_str_coding('bar'),
+                 force_default_str_coding('baz'))
 
-class MigSharedBase__force_default_str_coding_rec(MigTestCase):
-    """Unit tests of mig.shared.base force_default_str_coding_rec()"""
+        self.assertEqual(output, truth)
 
-    def test_encode_a_string(self):
-        output = force_default_str_coding_rec('foobar')
+    def test_encode_within_a_tuple_unicode(self):
+        output = force_default_fs_coding_rec(('foo', 'bar', 'baz'))
+        truth = (force_default_str_coding('foo'),
+                 force_default_str_coding('bar'),
+                 force_default_str_coding('baz'))
 
-        self.assertEqual(output, 'foobar')
+        self.assertEqual(output, truth)
 
-    def test_encode_within_a_dict(self):
-        output = force_default_str_coding_rec({ 'key': 'value' })
 
-        self.assertEqual(output, { 'key': 'value' })
+class MigSharedBase__force_utf8(MigTestCase):
+    """Unit tests of mig.shared.base force_utf8()"""
 
-    def test_encode_within_a_list(self):
-        output = force_default_str_coding_rec(['foo', 'bar', 'baz'])
+    def test_encode_string(self):
+        output = force_utf8('foobar')
 
-        self.assertEqual(output, ['foo', 'bar', 'baz'])
+        self.assertEqual(output, truth)
 
-    def test_encode_within_a_tuple(self):
-        output = force_default_str_coding_rec(('foo', 'bar', 'baz'))
+    def test_encode_bytes(self):
+        output = force_utf8(b'foobar')
 
-        self.assertEqual(output, ('foo', 'bar', 'baz'))
+        self.assertEqual(output, truth)
 
+    def test_encode_unicode(self):
+        output = force_utf8(u'foobar')
 
-class MigSharedBase__force_utf8(MigTestCase):
-    """Unit tests of mig.shared.base force_utf8()"""
+        self.assertEqual(output, truth)
 
-    def test_encode_a_string(self):
-        output = force_utf8('foobar')
 
-        self.assertEqual(output, b'foobar')
+class MigSharedBase__force_unicode(MigTestCase):
+    """Unit tests of mig.shared.base force_unicode()"""
 
+    def test_encode_a_string(self):
+        # input_vals = (b'foobar', u'foobar')
+        # truth = u'foobar'
+        # for val in input_vals:
+        #     output = force_unicode(val)
+        #     self.assertEqual(output, truth)
+        pass
 
 if __name__ == '__main__':
-    testmain()
+    testmain(failfast=True)