def test_client_adds_serializer(self): serializer = mock.Mock() client = RiakClient("test_bucket", pool=mock.Mock(spec=riakcached.pools.Pool)) client.add_serializer("application/test", serializer) self.assertEqual(client._serializers["application/test"], serializer) client.serialize("test", "application/test") serializer.assert_called()
def test_client_adds_serializer(self): serializer = mock.Mock() client = RiakClient("test_bucket", pool=mock.Mock(spec=riakcached.pools.Pool)) client.add_serializer("application/test", serializer) self.assertEqual(client._serializers["application/test"], serializer) client.serialize("test", "application/test") serializer.assert_called()
def test_set_uses_serializer(self): pool = mock.Mock(spec=riakcached.pools.Pool) pool.request.return_value = 204, "", {} pool.url = "http://127.0.0.1:8098" def serializer(data): return "serialized" client = RiakClient("test_bucket", pool=pool) client.add_serializer("application/test", serializer) client.set("test", "value", content_type="application/test") pool.request.assert_called_once_with( method="POST", url="http://127.0.0.1:8098/buckets/test_bucket/keys/test", body="serialized", headers={ "Content-Type": "application/test", }, )
def test_set_uses_serializer(self): pool = mock.Mock(spec=riakcached.pools.Pool) pool.request.return_value = 204, "", {} pool.url = "http://127.0.0.1:8098" def serializer(data): return "serialized" client = RiakClient("test_bucket", pool=pool) client.add_serializer("application/test", serializer) client.set("test", "value", content_type="application/test") pool.request.assert_called_once_with( method="POST", url="http://127.0.0.1:8098/buckets/test_bucket/keys/test", body="serialized", headers={ "Content-Type": "application/test", }, )