mirror of
https://github.com/esphome/esphome.git
synced 2025-11-01 15:41:52 +00:00
Merge branch 'addressable_light_tests' into light-addr
This commit is contained in:
@@ -49,3 +49,42 @@ async def test_noise_encryption_key_protection(
|
|||||||
with pytest.raises(InvalidEncryptionKeyAPIError):
|
with pytest.raises(InvalidEncryptionKeyAPIError):
|
||||||
async with api_client_connected(noise_psk=wrong_key) as client:
|
async with api_client_connected(noise_psk=wrong_key) as client:
|
||||||
await client.device_info()
|
await client.device_info()
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_noise_encryption_key_clear_protection(
|
||||||
|
yaml_config: str,
|
||||||
|
run_compiled: RunCompiledFunction,
|
||||||
|
api_client_connected: APIClientConnectedFactory,
|
||||||
|
) -> None:
|
||||||
|
"""Test that noise encryption key set in YAML cannot be changed via API."""
|
||||||
|
# The key that's set in the YAML fixture
|
||||||
|
noise_psk = "zX9/JHxMKwpP0jUGsF0iESCm1wRvNgR6NkKVOhn7kSs="
|
||||||
|
|
||||||
|
# Keep ESPHome process running throughout all tests
|
||||||
|
async with run_compiled(yaml_config):
|
||||||
|
# First connection - test key change attempt
|
||||||
|
async with api_client_connected(noise_psk=noise_psk) as client:
|
||||||
|
# Verify connection is established
|
||||||
|
device_info = await client.device_info()
|
||||||
|
assert device_info is not None
|
||||||
|
|
||||||
|
# Try to set a new encryption key via API
|
||||||
|
new_key = b"" # Empty key to attempt to clear
|
||||||
|
|
||||||
|
# This should fail since key was set in YAML
|
||||||
|
success = await client.noise_encryption_set_key(new_key)
|
||||||
|
assert success is False
|
||||||
|
|
||||||
|
# Reconnect with the original key to verify it still works
|
||||||
|
async with api_client_connected(noise_psk=noise_psk) as client:
|
||||||
|
# Verify connection is still successful with original key
|
||||||
|
device_info = await client.device_info()
|
||||||
|
assert device_info is not None
|
||||||
|
assert device_info.name == "noise-key-test"
|
||||||
|
|
||||||
|
# Verify that connecting with a wrong key fails
|
||||||
|
wrong_key = base64.b64encode(b"y" * 32).decode() # Different key
|
||||||
|
with pytest.raises(InvalidEncryptionKeyAPIError):
|
||||||
|
async with api_client_connected(noise_psk=wrong_key) as client:
|
||||||
|
await client.device_info()
|
||||||
|
|||||||
Reference in New Issue
Block a user