Fix dangerous-default-value warnings in plex tests (#119603)

pull/119621/head
epenet 2024-06-13 17:01:52 +02:00 committed by GitHub
parent 97e19cb61c
commit 1440ad26c8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 2 deletions

View File

@ -1,9 +1,11 @@
"""Helper methods for Plex tests."""
from datetime import timedelta
from typing import Any
from plexwebsocket import SIGNAL_CONNECTION_STATE, STATE_CONNECTED
from homeassistant.helpers.typing import UNDEFINED, UndefinedType
import homeassistant.util.dt as dt_util
from tests.common import async_fire_time_changed
@ -27,10 +29,14 @@ def websocket_connected(mock_websocket):
callback(SIGNAL_CONNECTION_STATE, STATE_CONNECTED, None)
def trigger_plex_update(mock_websocket, msgtype="playing", payload=UPDATE_PAYLOAD):
def trigger_plex_update(
mock_websocket,
msgtype="playing",
payload: dict[str, Any] | UndefinedType = UNDEFINED,
):
"""Call the websocket callback method with a Plex update."""
callback = mock_websocket.call_args[0][1]
callback(msgtype, payload, None)
callback(msgtype, UPDATE_PAYLOAD if payload is UNDEFINED else payload, None)
async def wait_for_debouncer(hass):