Fix flaky UPNP test (#58493)
parent
7024a5d7d9
commit
953ad71d22
|
@ -52,6 +52,7 @@ class MockDevice:
|
|||
self._udn = udn
|
||||
self.traffic_times_polled = 0
|
||||
self.status_times_polled = 0
|
||||
self._timestamp = dt.utcnow()
|
||||
|
||||
@classmethod
|
||||
async def async_create_device(cls, hass, ssdp_location) -> "MockDevice":
|
||||
|
@ -108,7 +109,7 @@ class MockDevice:
|
|||
"""Get traffic data."""
|
||||
self.traffic_times_polled += 1
|
||||
return {
|
||||
TIMESTAMP: dt.utcnow(),
|
||||
TIMESTAMP: self._timestamp,
|
||||
BYTES_RECEIVED: 0,
|
||||
BYTES_SENT: 0,
|
||||
PACKETS_RECEIVED: 0,
|
||||
|
|
|
@ -44,7 +44,7 @@ async def test_upnp_sensors(hass: HomeAssistant, setup_integration: MockConfigEn
|
|||
# Second poll.
|
||||
mock_device.async_get_traffic_data = AsyncMock(
|
||||
return_value={
|
||||
TIMESTAMP: dt_util.utcnow() + UPDATE_INTERVAL,
|
||||
TIMESTAMP: mock_device._timestamp + UPDATE_INTERVAL,
|
||||
BYTES_RECEIVED: 10240,
|
||||
BYTES_SENT: 20480,
|
||||
PACKETS_RECEIVED: 30,
|
||||
|
@ -94,7 +94,7 @@ async def test_derived_upnp_sensors(
|
|||
# Second poll.
|
||||
mock_device.async_get_traffic_data = AsyncMock(
|
||||
return_value={
|
||||
TIMESTAMP: dt_util.utcnow() + UPDATE_INTERVAL,
|
||||
TIMESTAMP: mock_device._timestamp + UPDATE_INTERVAL,
|
||||
BYTES_RECEIVED: int(10240 * UPDATE_INTERVAL.total_seconds()),
|
||||
BYTES_SENT: int(20480 * UPDATE_INTERVAL.total_seconds()),
|
||||
PACKETS_RECEIVED: int(30 * UPDATE_INTERVAL.total_seconds()),
|
||||
|
|
Loading…
Reference in New Issue