mirror of
https://github.com/Skyvern-AI/skyvern.git
synced 2026-04-30 12:40:14 +00:00
22 lines
689 B
Python
22 lines
689 B
Python
"""Tests for RedisClientFactory."""
|
|
|
|
from unittest.mock import MagicMock
|
|
|
|
from skyvern.forge.sdk.redis.factory import RedisClientFactory
|
|
|
|
|
|
def test_default_is_none():
|
|
"""Factory returns None when no client has been set."""
|
|
# Reset to default state
|
|
RedisClientFactory.set_client(None) # type: ignore[arg-type]
|
|
assert RedisClientFactory.get_client() is None
|
|
|
|
|
|
def test_set_and_get():
|
|
"""Round-trip: set_client then get_client returns the same object."""
|
|
mock_client = MagicMock()
|
|
RedisClientFactory.set_client(mock_client)
|
|
assert RedisClientFactory.get_client() is mock_client
|
|
|
|
# Cleanup
|
|
RedisClientFactory.set_client(None) # type: ignore[arg-type]
|