async def test_delete_error_not_found(self, client):
# patch what is necesary for returnning an error string
connection = CoroutineMock()
connection.delete_command = CoroutineMock(return_value=NOT_FOUND)
connection_context = AsyncMagicMock()
connection_context.__aenter__.return_value = connection
node = Mock()
node.connection.return_value = connection_context
client._cluster.pick_node.return_value = node
with pytest.raises(NotFoundCommandError):
await client.delete(b"foo")
async def test_delete_use_timeout(self, client, mocker):
optimeout_class = mocker.patch("emcache.client.OpTimeout",
AsyncMagicMock())
connection = CoroutineMock()
connection.delete_command = CoroutineMock(return_value=DELETED)
connection_context = AsyncMagicMock()
connection_context.__aenter__.return_value = connection
node = Mock()
node.connection.return_value = connection_context
client._cluster.pick_node.return_value = node
await client.delete(b"foo")
optimeout_class.assert_called()
