mirror of
https://github.com/langgenius/dify.git
synced 2026-09-01 05:33:25 +08:00
55 lines
1.9 KiB
Python
55 lines
1.9 KiB
Python
import ssl
|
|
from collections.abc import Callable
|
|
|
|
import socketio
|
|
|
|
from configs import dify_config
|
|
from extensions import ext_socketio
|
|
|
|
|
|
def test_socketio_server_uses_redis_manager() -> None:
|
|
assert isinstance(ext_socketio.sio.manager, socketio.RedisManager)
|
|
|
|
|
|
def test_create_socketio_client_manager_uses_pubsub_url_and_prefixed_channel(
|
|
config_overrides: Callable[..., None],
|
|
) -> None:
|
|
config_overrides(PUBSUB_REDIS_URL="redis://redis.example.com:6380/3", REDIS_KEY_PREFIX="tenant-a")
|
|
|
|
manager = ext_socketio.create_socketio_client_manager()
|
|
|
|
assert manager.redis_url == "redis://redis.example.com:6380/3"
|
|
assert manager.channel == "tenant-a:socketio"
|
|
|
|
|
|
def test_build_redis_options_includes_tls_options_for_rediss(config_overrides: Callable[..., None]) -> None:
|
|
config_overrides(
|
|
REDIS_SSL_CERT_REQS="CERT_REQUIRED",
|
|
REDIS_SSL_CA_CERTS="/ca.pem",
|
|
REDIS_SSL_CERTFILE="/cert.pem",
|
|
REDIS_SSL_KEYFILE="/key.pem",
|
|
)
|
|
|
|
options = ext_socketio._build_redis_options("rediss://redis.example.com:6380/3")
|
|
|
|
assert options["ssl_cert_reqs"] == ssl.CERT_REQUIRED
|
|
assert options["ssl_ca_certs"] == "/ca.pem"
|
|
assert options["ssl_certfile"] == "/cert.pem"
|
|
assert options["ssl_keyfile"] == "/key.pem"
|
|
|
|
|
|
def test_build_redis_options_omits_socket_timeout(config_overrides: Callable[..., None]) -> None:
|
|
# socket_timeout must not be passed to RedisManager because the pub/sub
|
|
# listen loop blocks indefinitely between messages; a read timeout there
|
|
# triggers an infinite reconnect storm (issue #39423).
|
|
config_overrides(REDIS_SOCKET_TIMEOUT=5.0)
|
|
|
|
options = ext_socketio._build_redis_options("redis://redis.example.com:6380/3")
|
|
|
|
assert "socket_timeout" not in options
|
|
assert "socket_connect_timeout" in options
|
|
|
|
|
|
def test_socketio_server_uses_configured_max_http_buffer_size() -> None:
|
|
assert ext_socketio.sio.eio.max_http_buffer_size == dify_config.WEBSOCKET_MAX_HTTP_BUFFER_SIZE
|