mirror of
https://github.com/home-assistant/core.git
synced 2025-07-26 22:57:17 +00:00
Fix flaky UPNP test (#58493)
This commit is contained in:
parent
7024a5d7d9
commit
953ad71d22
@ -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,
|
||||||
|
@ -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()),
|
||||||
|
Loading…
x
Reference in New Issue
Block a user