mirror of
https://github.com/langgenius/dify.git
synced 2026-06-23 04:11:09 +08:00
This commit is contained in:
parent
694bf3754c
commit
2604c33e54
@ -2,6 +2,7 @@ from __future__ import annotations
|
||||
|
||||
import contextlib
|
||||
import json
|
||||
import logging
|
||||
from types import SimpleNamespace
|
||||
from typing import Any
|
||||
from unittest.mock import MagicMock
|
||||
@ -252,27 +253,28 @@ def test_add_trigger_subscription_should_raise_error_when_provider_limit_reached
|
||||
mock_session: MagicMock,
|
||||
provider_id: TriggerProviderID,
|
||||
provider_controller: MagicMock,
|
||||
caplog,
|
||||
) -> None:
|
||||
# Arrange
|
||||
_patch_redis_lock(mocker)
|
||||
mock_session.scalar.return_value = TriggerProviderService.__MAX_TRIGGER_PROVIDER_COUNT__
|
||||
_mock_get_trigger_provider(mocker, provider_controller)
|
||||
mock_logger = mocker.patch("services.trigger.trigger_provider_service.logger")
|
||||
|
||||
# Act + Assert
|
||||
with pytest.raises(ValueError, match="Maximum number of providers"):
|
||||
TriggerProviderService.add_trigger_subscription(
|
||||
tenant_id="tenant-1",
|
||||
user_id="user-1",
|
||||
name="main",
|
||||
provider_id=provider_id,
|
||||
endpoint_id="endpoint-1",
|
||||
credential_type=CredentialType.API_KEY,
|
||||
parameters={},
|
||||
properties={},
|
||||
credentials={},
|
||||
)
|
||||
mock_logger.exception.assert_called_once()
|
||||
with caplog.at_level(logging.ERROR, logger="services.trigger.trigger_provider_service"):
|
||||
with pytest.raises(ValueError, match="Maximum number of providers"):
|
||||
TriggerProviderService.add_trigger_subscription(
|
||||
tenant_id="tenant-1",
|
||||
user_id="user-1",
|
||||
name="main",
|
||||
provider_id=provider_id,
|
||||
endpoint_id="endpoint-1",
|
||||
credential_type=CredentialType.API_KEY,
|
||||
parameters={},
|
||||
properties={},
|
||||
credentials={},
|
||||
)
|
||||
assert sum(1 for r in caplog.records if r.levelno >= logging.ERROR) == 1
|
||||
|
||||
|
||||
def test_add_trigger_subscription_should_raise_error_when_name_exists(
|
||||
|
||||
Loading…
Reference in New Issue
Block a user