fix: redis plugin testing (#2938)
This commit is contained in:
@@ -283,6 +283,12 @@ class NoopAsyncRedisClient(AsyncRedisClient):
|
||||
async def scard(self, key: str) -> int:
|
||||
return 0
|
||||
|
||||
async def smembers(self, key: str) -> Set[str]:
|
||||
return set()
|
||||
|
||||
async def srem(self, key: str, *members: Union[str, int, float]) -> int:
|
||||
return 0
|
||||
|
||||
|
||||
async def get_redis_client() -> AsyncRedisClient:
|
||||
global _client_instance
|
||||
|
||||
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user