Kris
1022121bee
添加了channels限界上下文的大量单元测试文件，包括： 1. 各层级通用与专用的conftest夹具 2. 核心领域模型、事件、服务测试 3. 应用层流水线、扩展处理器测试 4. 适配器与插件层测试 5. 修复并补充了pool manager测试用例
134 lines
3.6 KiB
Python
134 lines
3.6 KiB
Python
"""yuxi.channels.adapters.masking_adapter 单元测试。
|
||
|
||
覆盖 ``MaskingAdapter`` 的 ``mask_full`` / ``mask_partial`` /
|
||
``is_sensitive_key`` 方法，使用 ``patch`` 替换 ``CryptoHelper``，
|
||
不依赖真实加解密配置。
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
from unittest.mock import MagicMock, patch
|
||
|
||
import pytest
|
||
|
||
from yuxi.channels.adapters.masking_adapter import MaskingAdapter
|
||
|
||
|
||
pytestmark = pytest.mark.unit
|
||
|
||
|
||
def _make_helper() -> MagicMock:
|
||
"""构造 CryptoHelper 桩。"""
|
||
helper = MagicMock()
|
||
helper.mask_full = MagicMock(return_value="***")
|
||
helper.mask_partial = MagicMock(return_value="abcd***")
|
||
helper._is_sensitive_key = MagicMock(return_value=True)
|
||
return helper
|
||
|
||
|
||
@pytest.mark.unit
|
||
class TestMaskingAdapterConstruction:
|
||
def test_init_loads_crypto_helper(self):
|
||
# Arrange
|
||
helper = _make_helper()
|
||
|
||
# Act
|
||
with patch(
|
||
"yuxi.channels.adapters.masking_adapter.get_crypto_helper",
|
||
return_value=helper,
|
||
):
|
||
adapter = MaskingAdapter()
|
||
|
||
# Assert
|
||
assert adapter._helper is helper
|
||
|
||
|
||
@pytest.mark.unit
|
||
class TestMaskingAdapterMaskFull:
|
||
def test_mask_full_delegates_to_helper(self):
|
||
# Arrange
|
||
helper = _make_helper()
|
||
with patch(
|
||
"yuxi.channels.adapters.masking_adapter.get_crypto_helper",
|
||
return_value=helper,
|
||
):
|
||
adapter = MaskingAdapter()
|
||
|
||
# Act
|
||
result = adapter.mask_full("secret-value")
|
||
|
||
# Assert
|
||
helper.mask_full.assert_called_once_with("secret-value")
|
||
assert result == "***"
|
||
|
||
def test_mask_full_passes_any_value_type(self):
|
||
# Arrange
|
||
helper = _make_helper()
|
||
helper.mask_full.return_value = {"masked": True}
|
||
with patch(
|
||
"yuxi.channels.adapters.masking_adapter.get_crypto_helper",
|
||
return_value=helper,
|
||
):
|
||
adapter = MaskingAdapter()
|
||
|
||
# Act
|
||
result = adapter.mask_full({"key": "value"})
|
||
|
||
# Assert
|
||
helper.mask_full.assert_called_once_with({"key": "value"})
|
||
assert result == {"masked": True}
|
||
|
||
|
||
@pytest.mark.unit
|
||
class TestMaskingAdapterMaskPartial:
|
||
def test_mask_partial_delegates_to_helper(self):
|
||
# Arrange
|
||
helper = _make_helper()
|
||
with patch(
|
||
"yuxi.channels.adapters.masking_adapter.get_crypto_helper",
|
||
return_value=helper,
|
||
):
|
||
adapter = MaskingAdapter()
|
||
|
||
# Act
|
||
result = adapter.mask_partial("token-value")
|
||
|
||
# Assert
|
||
helper.mask_partial.assert_called_once_with("token-value")
|
||
assert result == "abcd***"
|
||
|
||
|
||
@pytest.mark.unit
|
||
class TestMaskingAdapterIsSensitiveKey:
|
||
def test_is_sensitive_key_delegates_to_helper(self):
|
||
# Arrange
|
||
helper = _make_helper()
|
||
with patch(
|
||
"yuxi.channels.adapters.masking_adapter.get_crypto_helper",
|
||
return_value=helper,
|
||
):
|
||
adapter = MaskingAdapter()
|
||
|
||
# Act
|
||
result = adapter.is_sensitive_key("api_key")
|
||
|
||
# Assert
|
||
helper._is_sensitive_key.assert_called_once_with("api_key")
|
||
assert result is True
|
||
|
||
def test_is_sensitive_key_returns_false_for_non_sensitive(self):
|
||
# Arrange
|
||
helper = _make_helper()
|
||
helper._is_sensitive_key.return_value = False
|
||
with patch(
|
||
"yuxi.channels.adapters.masking_adapter.get_crypto_helper",
|
||
return_value=helper,
|
||
):
|
||
adapter = MaskingAdapter()
|
||
|
||
# Act
|
||
result = adapter.is_sensitive_key("display_name")
|
||
|
||
# Assert
|
||
assert result is False
|