refactor(tests): replace mock_logger with caplog in service tests (#37468) (#37617)

This commit is contained in:
Evan 2026-06-18 15:55:48 +08:00 committed by GitHub
parent 694bf3754c
commit 2604c33e54
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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(