Add tests for serde module

2 files changed, 25 insertions, 5 deletions

diff --git a/pymemcache/serde.py b/pymemcache/serde.py
index 1a5a31a..3140046 100644
--- a/pymemcache/serde.py
+++ b/pymemcache/serde.py
@@ -14,11 +14,7 @@
 
 import logging
 import pickle
-
-try:
-    from cStringIO import StringIO
-except ImportError:
-    from StringIO import StringIO
+from six.moves import cStringIO as StringIO
 
 try:
     long_type = long  # noqa
diff --git a/pymemcache/test/test_serde.py b/pymemcache/test/test_serde.py
new file mode 100644
index 0000000..9849cdc
--- /dev/null
+++ b/pymemcache/test/test_serde.py
@@ -0,0 +1,24 @@
+from unittest import TestCase
+
+from pymemcache.serde import (python_memcache_serializer,
+                              python_memcache_deserializer)
+
+
+class TestSerde(TestCase):
+
+    def check(self, value):
+        serialized, flags = python_memcache_serializer(b'key', value)
+        deserialized = python_memcache_deserializer(b'key', serialized, flags)
+        assert deserialized == value
+
+    def test_str(self):
+        self.check('value')
+
+    def test_int(self):
+        self.check(1)
+
+    def test_long(self):
+        self.check(123123123123123123123)
+
+    def test_pickleable(self):
+        self.check({'a': 'dict'})