From b75316f17617032c8dfbde2ad7b0d26ba048046a Mon Sep 17 00:00:00 2001 From: Adam Chainz Date: Tue, 16 Aug 2016 18:37:46 +0100 Subject: Add tests for serde module --- pymemcache/serde.py | 6 +----- pymemcache/test/test_serde.py | 24 ++++++++++++++++++++++++ 2 files changed, 25 insertions(+), 5 deletions(-) create mode 100644 pymemcache/test/test_serde.py 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'}) -- cgit v1.2.1