|
3 | 3 | import pytest |
4 | 4 | import random |
5 | 5 | import redis |
6 | | -from packaging.version import Version |
| 6 | +from distutils.version import LooseVersion |
7 | 7 | from redis.connection import parse_url |
8 | 8 | from unittest.mock import Mock |
9 | 9 | from urllib.parse import urlparse |
@@ -38,15 +38,15 @@ def pytest_sessionstart(session): |
38 | 38 |
|
39 | 39 | def skip_if_server_version_lt(min_version): |
40 | 40 | redis_version = REDIS_INFO["version"] |
41 | | - check = Version(redis_version) < Version(min_version) |
| 41 | + check = LooseVersion(redis_version) < LooseVersion(min_version) |
42 | 42 | return pytest.mark.skipif( |
43 | 43 | check, |
44 | 44 | reason="Redis version required >= {}".format(min_version)) |
45 | 45 |
|
46 | 46 |
|
47 | 47 | def skip_if_server_version_gte(min_version): |
48 | 48 | redis_version = REDIS_INFO["version"] |
49 | | - check = Version(redis_version) >= Version(min_version) |
| 49 | + check = LooseVersion(redis_version) >= LooseVersion(min_version) |
50 | 50 | return pytest.mark.skipif( |
51 | 51 | check, |
52 | 52 | reason="Redis version required < {}".format(min_version)) |
@@ -183,7 +183,7 @@ def wait_for_command(client, monitor, command): |
183 | 183 | # if we find a command with our key before the command we're waiting |
184 | 184 | # for, something went wrong |
185 | 185 | redis_version = REDIS_INFO["version"] |
186 | | - if Version(redis_version) >= Version('5.0.0'): |
| 186 | + if LooseVersion(redis_version) >= LooseVersion('5.0.0'): |
187 | 187 | id_str = str(client.client_id()) |
188 | 188 | else: |
189 | 189 | id_str = '%08x' % random.randrange(2**32) |
|
0 commit comments