mirror of
https://github.com/langgenius/dify.git
synced 2026-07-15 01:17:04 +08:00
e13271ba29
Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> Co-authored-by: Blackoutta <hyytez@gmail.com> Co-authored-by: Blackoutta <37723456+Blackoutta@users.noreply.github.com>
34 lines
1.3 KiB
Python
34 lines
1.3 KiB
Python
import ssl
|
|
|
|
import socketio
|
|
|
|
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(monkeypatch) -> None:
|
|
monkeypatch.setattr(ext_socketio.dify_config, "PUBSUB_REDIS_URL", "redis://redis.example.com:6380/3")
|
|
monkeypatch.setattr(ext_socketio.dify_config, "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(monkeypatch) -> None:
|
|
monkeypatch.setattr(ext_socketio.dify_config, "REDIS_SSL_CERT_REQS", "CERT_REQUIRED")
|
|
monkeypatch.setattr(ext_socketio.dify_config, "REDIS_SSL_CA_CERTS", "/ca.pem")
|
|
monkeypatch.setattr(ext_socketio.dify_config, "REDIS_SSL_CERTFILE", "/cert.pem")
|
|
monkeypatch.setattr(ext_socketio.dify_config, "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"
|