Use byte strings after serializing with serde

The pymemcache client will return a byte string, so we'll do the same to test that the deserializer works as expected.

This currently fails with Python 3.

1 files changed, 14 insertions, 2 deletions

diff --git a/pymemcache/test/test_serde.py b/pymemcache/test/test_serde.py
index 9849cdc..4792010 100644
--- a/pymemcache/test/test_serde.py
+++ b/pymemcache/test/test_serde.py
@@ -2,17 +2,29 @@ from unittest import TestCase
 
 from pymemcache.serde import (python_memcache_serializer,
                               python_memcache_deserializer)
+import pytest
+import six
 
 
+@pytest.mark.unit()
 class TestSerde(TestCase):
 
     def check(self, value):
         serialized, flags = python_memcache_serializer(b'key', value)
+
+        # pymemcache stores values as byte strings, so we immediately the value
+        # if needed so deserialized works as it would with a real server
+        if not isinstance(serialized, six.binary_type):
+            serialized = six.text_type(serialized).encode('ascii')
+
         deserialized = python_memcache_deserializer(b'key', serialized, flags)
         assert deserialized == value
 
-    def test_str(self):
-        self.check('value')
+    def test_bytes(self):
+        self.check(b'value')
+
+    def test_unicode(self):
+        self.check(u'value')
 
     def test_int(self):
         self.check(1)