Fix flaky UPNP test (#58493)

pull/58500/head
Erik Montnemery 2021-10-26 22:35:09 +02:00 committed by GitHub
parent 7024a5d7d9
commit 953ad71d22
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 3 deletions

View File

@ -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,

View File

@ -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()),