Add mock for listen in Teslemetry tests (#148853)
parent
549069e22c
commit
ffc2b0a8cf
|
@ -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:
|
||||
|
|
Loading…
Reference in New Issue