/usr/lib/python3/dist-packages/pylibmc/test.py is in python3-pylibmc 1.5.2-1build1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 | """Test utils."""
import os
import socket
import _pylibmc, pylibmc
from textwrap import dedent
class NotAliveError(Exception):
template = dedent("""
test memcached %s:%s not alive
consider setting MEMCACHED_HOST and/or MEMCACHED_PORT
to something more appropriate.""").lstrip()
def __init__(self, svr):
(self.server_type, self.address, self.port) = svr
def __str__(self):
return self.template % (self.address, self.port)
def get_version(addr):
(type_, host, port) = addr
if type_ != _pylibmc.server_type_tcp:
raise NotImplementedError("test server can only be on tcp for now")
else:
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
try:
s.connect((host, port))
s.send(b"version\r\n")
version = s.recv(4096)
finally:
s.close()
vstr = b"VERSION "
rnstr = b"\r\n"
if not version.startswith(vstr) or not version.endswith(rnstr):
raise ValueError("unexpected version return: %r" % (version,))
else:
version = version[8:-2]
return version
def is_alive(addr):
try:
version = get_version(addr)
except (ValueError, socket.error):
version = None
return bool(version), version
def make_test_client(cls=pylibmc.Client, env=None, server_type="tcp",
host="127.0.0.1", port=11211, **kwds):
"""Make a test client. `env` overrides keyword arguments.
>>> make_test_client() is not None
True
"""
server_types = {"tcp": _pylibmc.server_type_tcp,
"udp": _pylibmc.server_type_udp,
"unix": _pylibmc.server_type_unix}
if env is None:
env = os.environ
styp = env.get("MEMCACHED_TYPE", server_type)
host = env.get("MEMCACHED_HOST", host)
port = env.get("MEMCACHED_PORT", port)
test_server = (server_types[styp], str(host), int(port))
alive, version = is_alive(test_server)
if alive:
return cls([test_server], **kwds)
else:
raise NotAliveError(test_server)
|