pytest config file

1 files changed, 26 insertions, 0 deletions

diff --git a/tests/conftest.py b/tests/conftest.py
new file mode 100644
index 0000000..b3a2e5b
--- /dev/null
+++ b/tests/conftest.py
@@ -0,0 +1,26 @@
+import pytest
+import redis
+
+
+def _get_client(cls, request=None):
+    client = cls(host='localhost', port=6379, db=9)
+    client.flushdb()
+    if request:
+        request.addfinalizer(client.flushdb)
+    return client
+
+
+def skip_if_server_version_lt(min_version):
+    version = _get_client(redis.Redis).info()['redis_version']
+    c = "StrictVersion('%s') < StrictVersion('%s')" % (version, min_version)
+    return pytest.mark.skipif(c)
+
+
+@pytest.fixture()
+def r(request):
+    return _get_client(redis.Redis, request)
+
+
+@pytest.fixture()
+def sr(request):
+    return _get_client(redis.StrictRedis, request)