mirror of
https://github.com/home-assistant/core.git
synced 2025-07-26 22:57:17 +00:00
Fix flaky coverage in UK transport test (#32053)
This commit is contained in:
parent
7c2765fbff
commit
5776b9f17d
@ -2,6 +2,7 @@
|
|||||||
import re
|
import re
|
||||||
import unittest
|
import unittest
|
||||||
|
|
||||||
|
from asynctest import patch
|
||||||
import requests_mock
|
import requests_mock
|
||||||
|
|
||||||
from homeassistant.components.uk_transport.sensor import (
|
from homeassistant.components.uk_transport.sensor import (
|
||||||
@ -17,6 +18,7 @@ from homeassistant.components.uk_transport.sensor import (
|
|||||||
UkTransportSensor,
|
UkTransportSensor,
|
||||||
)
|
)
|
||||||
from homeassistant.setup import setup_component
|
from homeassistant.setup import setup_component
|
||||||
|
from homeassistant.util.dt import now
|
||||||
|
|
||||||
from tests.common import get_test_home_assistant, load_fixture
|
from tests.common import get_test_home_assistant, load_fixture
|
||||||
|
|
||||||
@ -77,7 +79,9 @@ class TestUkTransportSensor(unittest.TestCase):
|
|||||||
@requests_mock.Mocker()
|
@requests_mock.Mocker()
|
||||||
def test_train(self, mock_req):
|
def test_train(self, mock_req):
|
||||||
"""Test for operational uk_transport sensor with proper attributes."""
|
"""Test for operational uk_transport sensor with proper attributes."""
|
||||||
with requests_mock.Mocker() as mock_req:
|
with requests_mock.Mocker() as mock_req, patch(
|
||||||
|
"homeassistant.util.dt.now", return_value=now().replace(hour=13)
|
||||||
|
):
|
||||||
uri = re.compile(UkTransportSensor.TRANSPORT_API_URL_BASE + "*")
|
uri = re.compile(UkTransportSensor.TRANSPORT_API_URL_BASE + "*")
|
||||||
mock_req.get(uri, text=load_fixture("uk_transport_train.json"))
|
mock_req.get(uri, text=load_fixture("uk_transport_train.json"))
|
||||||
assert setup_component(self.hass, "sensor", {"sensor": self.config})
|
assert setup_component(self.hass, "sensor", {"sensor": self.config})
|
||||||
|
Loading…
x
Reference in New Issue
Block a user