Cleanup unnecessary definition in lamarzocco tests (#121655)

This commit is contained in:
epenet 2024-07-10 09:08:39 +02:00 committed by GitHub
parent fb758bd8b6
commit 7a337ecd1d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1,6 +1,6 @@
"""Lamarzocco session fixtures."""
from collections.abc import Callable, Generator
from collections.abc import Generator
import json
from unittest.mock import MagicMock, patch
@ -128,14 +128,6 @@ def mock_lamarzocco(device_fixture: MachineModel) -> Generator[MagicMock]:
lamarzocco.firmware[FirmwareType.GATEWAY].latest_version = "v3.5-rc3"
lamarzocco.firmware[FirmwareType.MACHINE].latest_version = "1.55"
async def websocket_connect_mock(
notify_callback: Callable | None,
) -> None:
"""Mock the websocket connect method."""
return None
lamarzocco.websocket_connect = websocket_connect_mock
yield lamarzocco