From 953ad71d2201dcbefc3f628e2e89bd51b083aeb9 Mon Sep 17 00:00:00 2001 From: Erik Montnemery Date: Tue, 26 Oct 2021 22:35:09 +0200 Subject: [PATCH] Fix flaky UPNP test (#58493) --- tests/components/upnp/conftest.py | 3 ++- tests/components/upnp/test_sensor.py | 4 ++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/tests/components/upnp/conftest.py b/tests/components/upnp/conftest.py index 54a7fce44fb..7c68bfb1db1 100644 --- a/tests/components/upnp/conftest.py +++ b/tests/components/upnp/conftest.py @@ -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, diff --git a/tests/components/upnp/test_sensor.py b/tests/components/upnp/test_sensor.py index 068b5260d45..7d6b498ab24 100644 --- a/tests/components/upnp/test_sensor.py +++ b/tests/components/upnp/test_sensor.py @@ -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()),