Fimeg/letta-server
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix: redis plugin testing (#2938)

Browse Source
This commit is contained in:
Andy Li
2025-06-20 13:35:35 -07:00
committed by GitHub
parent 4deaea4d95
commit 971fc37cdb
2 changed files with 15 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tests/test_plugins.py
View File

@@ -1,6 +1,6 @@
import pytest
from letta.data_sources.redis_client import get_redis_client
from letta.data_sources.redis_client import NoopAsyncRedisClient, get_redis_client
from letta.helpers.decorators import experimental
from letta.settings import settings
@@ -86,7 +86,11 @@ async def test_redis_flag(event_loop):
await redis_client.create_inclusion_exclusion_keys(group=group_name)
await redis_client.sadd(include_key, test_user)
assert await _new_feature(user_id=test_user) == "new_feature"
with pytest.raises(Exception):
assert await _new_feature(user_id=test_user + "1")
print("members: ", await redis_client.smembers(include_key))
if not isinstance(redis_client, NoopAsyncRedisClient):
assert await _new_feature(user_id=test_user) == "new_feature"
with pytest.raises(Exception):
await _new_feature(user_id=test_user + "1")
print("members: ", await redis_client.smembers(include_key))
else:
with pytest.raises(Exception):
await _new_feature(user_id=test_user)