jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-26 22:57:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix flaky UPNP test (#58493)

Browse Source
This commit is contained in:
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/components/upnp/conftest.py
View File

@ -52,6 +52,7 @@ class MockDevice:
self._udn = udn self._udn = udn
self.traffic_times_polled = 0 self.traffic_times_polled = 0
self.status_times_polled = 0 self.status_times_polled = 0
self._timestamp = dt.utcnow()
@classmethod @classmethod
async def async_create_device(cls, hass, ssdp_location) -> "MockDevice": async def async_create_device(cls, hass, ssdp_location) -> "MockDevice":
@ -108,7 +109,7 @@ class MockDevice:
"""Get traffic data.""" """Get traffic data."""
self.traffic_times_polled += 1 self.traffic_times_polled += 1
return { return {
TIMESTAMP: dt.utcnow(), TIMESTAMP: self._timestamp,
BYTES_RECEIVED: 0, BYTES_RECEIVED: 0,
BYTES_SENT: 0, BYTES_SENT: 0,
PACKETS_RECEIVED: 0, PACKETS_RECEIVED: 0,

4
tests/components/upnp/test_sensor.py
View File

@ -44,7 +44,7 @@ async def test_upnp_sensors(hass: HomeAssistant, setup_integration: MockConfigEn
# Second poll. # Second poll.
mock_device.async_get_traffic_data = AsyncMock( mock_device.async_get_traffic_data = AsyncMock(
return_value={ return_value={
TIMESTAMP: dt_util.utcnow() + UPDATE_INTERVAL, TIMESTAMP: mock_device._timestamp + UPDATE_INTERVAL,
BYTES_RECEIVED: 10240, BYTES_RECEIVED: 10240,
BYTES_SENT: 20480, BYTES_SENT: 20480,
PACKETS_RECEIVED: 30, PACKETS_RECEIVED: 30,
@ -94,7 +94,7 @@ async def test_derived_upnp_sensors(
# Second poll. # Second poll.
mock_device.async_get_traffic_data = AsyncMock( mock_device.async_get_traffic_data = AsyncMock(
return_value={ return_value={
TIMESTAMP: dt_util.utcnow() + UPDATE_INTERVAL, TIMESTAMP: mock_device._timestamp + UPDATE_INTERVAL,
BYTES_RECEIVED: int(10240 * UPDATE_INTERVAL.total_seconds()), BYTES_RECEIVED: int(10240 * UPDATE_INTERVAL.total_seconds()),
BYTES_SENT: int(20480 * UPDATE_INTERVAL.total_seconds()), BYTES_SENT: int(20480 * UPDATE_INTERVAL.total_seconds()),
PACKETS_RECEIVED: int(30 * UPDATE_INTERVAL.total_seconds()), PACKETS_RECEIVED: int(30 * UPDATE_INTERVAL.total_seconds()),