Add mock for listen in Teslemetry tests (#148853)

This commit is contained in:
Brett Adams 2025-07-16 16:09:54 +10:00 committed by GitHub
parent 549069e22c
commit ffc2b0a8cf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -119,8 +119,17 @@ def mock_energy_history():
@pytest.fixture(autouse=True)
def mock_add_listener():
def mock_stream_listen():
"""Mock Teslemetry Stream listen method."""
with patch(
"teslemetry_stream.TeslemetryStream.listen",
) as mock_stream_listen:
yield mock_stream_listen
@pytest.fixture(autouse=True)
def mock_add_listener():
"""Mock Teslemetry Stream add listener method."""
with patch(
"teslemetry_stream.TeslemetryStream.async_add_listener",
) as mock_add_listener: