def test_set_calls_pool_request_with_correct_content_type(self):
pool = mock.Mock(spec=riakcached.pools.Pool)
pool.request.return_value = None, None, None
pool.url = "http://127.0.0.1:8098"
client = RiakClient("test_bucket", pool=pool)
client.set("test", "value", content_type="application/test")
self.assertTrue(pool.request.called)
pool.request.assert_called_once_with(
method="POST",
url="http://127.0.0.1:8098/buckets/test_bucket/keys/test",
body="value",
headers={
"Content-Type": "application/test",
},
)
def test_set_calls_pool_request_with_correct_content_type(self):
pool = mock.Mock(spec=riakcached.pools.Pool)
pool.request.return_value = None, None, None
pool.url = "http://127.0.0.1:8098"
client = RiakClient("test_bucket", pool=pool)
client.set("test", "value", content_type="application/test")
self.assertTrue(pool.request.called)
pool.request.assert_called_once_with(
method="POST",
url="http://127.0.0.1:8098/buckets/test_bucket/keys/test",
body="value",
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",
},
)
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",
},
)