mirror of
https://github.com/esphome/aioesphomeapi.git
synced 2024-11-05 09:29:42 +01:00
Add additional coverage for starting noise connections (#653)
This commit is contained in:
parent
0afa8c6832
commit
2501ea4275
@ -20,7 +20,8 @@ from google.protobuf import message
|
||||
|
||||
import aioesphomeapi.host_resolver as hr
|
||||
|
||||
from ._frame_helper import APINoiseFrameHelper, APIPlaintextFrameHelper
|
||||
from ._frame_helper.noise import APINoiseFrameHelper
|
||||
from ._frame_helper.plain_text import APIPlaintextFrameHelper
|
||||
from .api_pb2 import ( # type: ignore
|
||||
ConnectRequest,
|
||||
ConnectResponse,
|
||||
|
@ -57,13 +57,33 @@ def connection_params() -> ConnectionParams:
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def conn(connection_params) -> APIConnection:
|
||||
async def on_stop(expected_disconnect: bool) -> None:
|
||||
pass
|
||||
def noise_connection_params() -> ConnectionParams:
|
||||
return ConnectionParams(
|
||||
address="fake.address",
|
||||
port=6052,
|
||||
password=None,
|
||||
client_info="Tests client",
|
||||
keepalive=KEEP_ALIVE_INTERVAL,
|
||||
zeroconf_manager=ZeroconfManager(),
|
||||
noise_psk="QRTIErOb/fcE9Ukd/5qA3RGYMn0Y+p06U58SCtOXvPc=",
|
||||
expected_name="test",
|
||||
)
|
||||
|
||||
|
||||
async def on_stop(expected_disconnect: bool) -> None:
|
||||
pass
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def conn(connection_params: ConnectionParams) -> APIConnection:
|
||||
return APIConnection(connection_params, on_stop)
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def noise_conn(noise_connection_params: ConnectionParams) -> APIConnection:
|
||||
return APIConnection(noise_connection_params, on_stop)
|
||||
|
||||
|
||||
@pytest_asyncio.fixture(name="plaintext_connect_task_no_login")
|
||||
async def plaintext_connect_task_no_login(
|
||||
conn: APIConnection, resolve_host, socket_socket, event_loop
|
||||
|
@ -596,6 +596,31 @@ async def test_init_plaintext_with_wrong_preamble(conn: APIConnection):
|
||||
await task
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_init_noise_with_wrong_byte_marker(noise_conn: APIConnection) -> None:
|
||||
loop = asyncio.get_event_loop()
|
||||
transport = MagicMock()
|
||||
protocol: APINoiseFrameHelper | None = None
|
||||
|
||||
async def _create_connection(create, sock, *args, **kwargs):
|
||||
nonlocal protocol
|
||||
protocol = create()
|
||||
protocol.connection_made(transport)
|
||||
return transport, protocol
|
||||
|
||||
with patch.object(loop, "create_connection", side_effect=_create_connection):
|
||||
task = asyncio.create_task(noise_conn._connect_init_frame_helper())
|
||||
await asyncio.sleep(0)
|
||||
|
||||
assert protocol is not None
|
||||
assert isinstance(noise_conn._frame_helper, APINoiseFrameHelper)
|
||||
|
||||
protocol.data_received(b"\x00\x00\x00")
|
||||
|
||||
with pytest.raises(ProtocolAPIError, match="Marker byte invalid"):
|
||||
await task
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_eof_received_closes_connection(
|
||||
plaintext_connect_task_with_login: tuple[
|
||||
|
Loading…
Reference in New Issue
Block a user