diff options
Diffstat (limited to 'pymemcache/test/conftest.py')
-rw-r--r-- | pymemcache/test/conftest.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/pymemcache/test/conftest.py b/pymemcache/test/conftest.py index 8792cc3..5d1f9ce 100644 --- a/pymemcache/test/conftest.py +++ b/pymemcache/test/conftest.py @@ -33,3 +33,14 @@ def pytest_generate_tests(metafunc): socket_modules.append(gevent_socket) metafunc.parametrize("socket_module", socket_modules) + + if 'client_class' in metafunc.fixturenames: + from pymemcache.client.base import PooledClient, Client + from pymemcache.client.hash import HashClient + class HashClientSingle(HashClient): + def __init__(self, server, *args, **kwargs): + super(HashClientSingle, self).__init__([server], *args, **kwargs) + + metafunc.parametrize( + "client_class", [Client, PooledClient, HashClientSingle] + ) |