diff options
author | Adam Chainz <adam@adamj.eu> | 2016-08-16 18:37:46 +0100 |
---|---|---|
committer | Adam Chainz <adam@adamj.eu> | 2016-08-17 12:20:45 +0100 |
commit | b75316f17617032c8dfbde2ad7b0d26ba048046a (patch) | |
tree | c08ffa852832c74b9fc0d5dcf735643f807bd019 | |
parent | e7489e533a2f173ae9d79251c2ed96ddf8fcb519 (diff) | |
download | pymemcache-b75316f17617032c8dfbde2ad7b0d26ba048046a.tar.gz |
Add tests for serde module
-rw-r--r-- | pymemcache/serde.py | 6 | ||||
-rw-r--r-- | pymemcache/test/test_serde.py | 24 |
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'}) |