def test_serialize_no_compress(self): zlib_patch = mock.patch.object(memcache_lib, "zlib") zlib = zlib_patch.start() self.addCleanup(zlib_patch.stop) dump_no_compress = memcache_lib.make_dump_and_compress_fn( min_compress_length=0 # disable compression ) value, flags = dump_no_compress(key="key", value="simple string") self.assertEqual(value, "simple string") self.assertEqual(flags, 0) zlib.compress.assert_not_called()
def test_serialize_no_compress(self): zlib_patch = mock.patch.object(memcache_lib, "zlib") zlib = zlib_patch.start() self.addCleanup(zlib_patch.stop) dump_no_compress = memcache_lib.make_dump_and_compress_fn( min_compress_length=0, # disable compression ) value, flags = dump_no_compress(key="key", value="simple string") self.assertEqual(value, "simple string") self.assertEqual(flags, 0) zlib.compress.assert_not_called()
def test_serialize_other(self): json_patch = mock.patch.object(memcache_lib, "json") json = json_patch.start() self.addCleanup(json_patch.stop) expected_value = object() json.dumps.return_value = expected_value dump_no_compress = memcache_lib.make_dump_and_compress_fn() value, flags = dump_no_compress(key="key", value=("stuff", 1, False)) json.dumps.assert_called_with(("stuff", 1, False)) self.assertEqual(value, expected_value) self.assertEqual(flags, memcache_lib.Flags.JSON)
def test_serialize_compress(self): zlib_patch = mock.patch.object(memcache_lib, "zlib") zlib = zlib_patch.start() self.addCleanup(zlib_patch.stop) expected_value = object() zlib.compress.return_value = expected_value json_and_compress = memcache_lib.make_dump_and_compress_fn( min_compress_length=1, compress_level=1) value, flags = json_and_compress(key="key", value="simple string") self.assertEqual(value, expected_value) self.assertEqual(flags, memcache_lib.Flags.ZLIB) zlib.compress.assert_called_with("simple string", 1)
def test_serialize_long(self): try: long except NameError: # python3 value = 100 expected_flags = memcache_lib.Flags.INTEGER else: value = long(100) expected_flags = memcache_lib.Flags.LONG dump_no_compress = memcache_lib.make_dump_and_compress_fn() value, flags = dump_no_compress(key="key", value=value) self.assertEqual(value, "100") self.assertEqual(flags, expected_flags)
def test_serialize_compress(self): zlib_patch = mock.patch.object(memcache_lib, "zlib") zlib = zlib_patch.start() self.addCleanup(zlib_patch.stop) expected_value = object() zlib.compress.return_value = expected_value json_and_compress = memcache_lib.make_dump_and_compress_fn( min_compress_length=1, compress_level=1, ) value, flags = json_and_compress(key="key", value="simple string") self.assertEqual(value, expected_value) self.assertEqual(flags, memcache_lib.Flags.ZLIB) zlib.compress.assert_called_with("simple string", 1)
def test_serialize_int(self): dump_no_compress = memcache_lib.make_dump_and_compress_fn() value, flags = dump_no_compress(key="key", value=100) self.assertEqual(value, "100") self.assertEqual(flags, memcache_lib.Flags.INTEGER)
def test_serialize_str(self): dump_no_compress = memcache_lib.make_dump_and_compress_fn() value, flags = dump_no_compress(key="key", value="val") self.assertEqual(value, "val") self.assertEqual(flags, 0)