19 lines
667 B
Python
19 lines
667 B
Python
from unittest.mock import AsyncMock, MagicMock
|
|
|
|
import pytest
|
|
|
|
from ingestor.connection.security import SESSION_TIMEOUT_MS, apply_security
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_apply_security_sets_sha256_and_channel_lifetime():
|
|
client = MagicMock()
|
|
client.set_security = AsyncMock()
|
|
await apply_security(client, 'urn:x', 'c.pem', 'k.pem', None)
|
|
assert client.session_timeout == SESSION_TIMEOUT_MS
|
|
assert client.secure_channel_timeout == int(SESSION_TIMEOUT_MS * 1.25)
|
|
args, kwargs = client.set_security.call_args
|
|
from asyncua.crypto.security_policies import SecurityPolicyBasic256Sha256
|
|
|
|
assert args[0] is SecurityPolicyBasic256Sha256
|