Skip to content

Commit 114fc0c

Browse files
basic test suite
1 parent c4e3017 commit 114fc0c

File tree

1 file changed

+27
-6
lines changed

1 file changed

+27
-6
lines changed

tests/integration/test_connection.py

Lines changed: 27 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -6,13 +6,16 @@
66
from redis.exceptions import ConnectionError
77

88
from redisvl.redis.connection import RedisConnectionFactory, get_address_from_env
9+
from redisvl.version import __version__
10+
11+
EXPECTED_LIB_NAME = f"redis-py(redisvl_v{__version__})"
912

1013

1114
def test_get_address_from_env(redis_url):
1215
assert get_address_from_env() == redis_url
1316

1417

15-
def test_sync_redis_connection(redis_url):
18+
def test_sync_redis_connect(redis_url):
1619
client = RedisConnectionFactory.connect(redis_url)
1720
assert client is not None
1821
assert isinstance(client, Redis)
@@ -21,7 +24,7 @@ def test_sync_redis_connection(redis_url):
2124

2225

2326
@pytest.mark.asyncio
24-
async def test_async_redis_connection(redis_url):
27+
async def test_async_redis_connect(redis_url):
2528
client = RedisConnectionFactory.connect(redis_url, use_async=True)
2629
assert client is not None
2730
assert isinstance(client, AsyncRedis)
@@ -49,11 +52,29 @@ def test_unknown_redis():
4952
bad_client.ping()
5053

5154

52-
def test_required_modules(client):
53-
RedisConnectionFactory.validate_redis_modules(client)
55+
def test_validate_redis(client):
56+
RedisConnectionFactory.validate_redis(client)
57+
lib_name = client.client_info()
58+
assert lib_name["lib-name"] == EXPECTED_LIB_NAME
59+
60+
61+
@pytest.mark.asyncio
62+
async def test_validate_async_redis(async_client):
63+
client = await async_client
64+
RedisConnectionFactory.validate_async_redis(client)
65+
lib_name = await client.client_info()
66+
assert lib_name["lib-name"] == EXPECTED_LIB_NAME
67+
68+
69+
def test_custom_lib_name(client):
70+
RedisConnectionFactory.validate_redis(client, "langchain_v0.1.0")
71+
lib_name = client.client_info()
72+
assert lib_name["lib-name"] == f"redis-py(redisvl_v{__version__};langchain_v0.1.0)"
5473

5574

5675
@pytest.mark.asyncio
57-
async def test_async_required_modules(async_client):
76+
async def test_async_custom_lib_name(async_client):
5877
client = await async_client
59-
RedisConnectionFactory.validate_async_redis_modules(client)
78+
RedisConnectionFactory.validate_async_redis(client, "langchain_v0.1.0")
79+
lib_name = await client.client_info()
80+
assert lib_name["lib-name"] == f"redis-py(redisvl_v{__version__};langchain_v0.1.0)"

0 commit comments

Comments
 (0)