mirror of
https://github.com/langgenius/dify.git
synced 2026-07-31 01:09:32 +08:00
35 lines
1.3 KiB
Python
35 lines
1.3 KiB
Python
from __future__ import annotations
|
|
|
|
import pytest
|
|
from pydantic import ValidationError
|
|
|
|
from dify_agent.runtime_backend.e2b import E2B_MAX_ACTIVE_TIMEOUT_SECONDS
|
|
from dify_agent.runtime_backend.profile import DEFAULT_E2B_TEMPLATE, RuntimeBackendSettings
|
|
|
|
|
|
def test_e2b_backend_uses_prepared_dify_template_by_default() -> None:
|
|
settings = RuntimeBackendSettings(runtime_backend="e2b", e2b_api_key="secret")
|
|
|
|
assert settings.e2b_template == "difys-default-team/dify-agent-local-sandbox"
|
|
assert settings.e2b_template == DEFAULT_E2B_TEMPLATE
|
|
assert settings.e2b_active_timeout_seconds == E2B_MAX_ACTIVE_TIMEOUT_SECONDS
|
|
|
|
|
|
def test_e2b_backend_requires_api_key() -> None:
|
|
with pytest.raises(ValidationError, match="e2b_api_key"):
|
|
_ = RuntimeBackendSettings(runtime_backend="e2b")
|
|
|
|
|
|
def test_e2b_backend_rejects_active_timeout_above_platform_limit() -> None:
|
|
with pytest.raises(ValidationError, match="less than or equal"):
|
|
_ = RuntimeBackendSettings(
|
|
runtime_backend="e2b",
|
|
e2b_api_key="secret",
|
|
e2b_active_timeout_seconds=E2B_MAX_ACTIVE_TIMEOUT_SECONDS + 1,
|
|
)
|
|
|
|
|
|
def test_local_backend_requires_shellctl_endpoint() -> None:
|
|
with pytest.raises(ValidationError, match="local_sandbox_endpoint"):
|
|
_ = RuntimeBackendSettings(runtime_backend="local")
|