8555e17eb9
* Add hourly forecast * fix tests to assert for 2 entities created * fix test to assert for 4 calls * correct test tracking home number of calls * fox tests * fix test * Apply suggestions from code review * black Co-authored-by: Chris Talkington <chris@talkingtontech.com>
41 lines
1.4 KiB
Python
41 lines
1.4 KiB
Python
"""Test Met weather entity."""
|
|
|
|
|
|
async def test_tracking_home(hass, mock_weather):
|
|
"""Test we track home."""
|
|
await hass.config_entries.flow.async_init("met", context={"source": "onboarding"})
|
|
await hass.async_block_till_done()
|
|
assert len(hass.states.async_entity_ids("weather")) == 2
|
|
assert len(mock_weather.mock_calls) == 4
|
|
|
|
# Test we track config
|
|
await hass.config.async_update(latitude=10, longitude=20)
|
|
await hass.async_block_till_done()
|
|
|
|
assert len(mock_weather.mock_calls) == 8
|
|
|
|
entry = hass.config_entries.async_entries()[0]
|
|
await hass.config_entries.async_remove(entry.entry_id)
|
|
assert len(hass.states.async_entity_ids("weather")) == 0
|
|
|
|
|
|
async def test_not_tracking_home(hass, mock_weather):
|
|
"""Test when we not track home."""
|
|
await hass.config_entries.flow.async_init(
|
|
"met",
|
|
context={"source": "user"},
|
|
data={"name": "Somewhere", "latitude": 10, "longitude": 20, "elevation": 0},
|
|
)
|
|
await hass.async_block_till_done()
|
|
assert len(hass.states.async_entity_ids("weather")) == 2
|
|
assert len(mock_weather.mock_calls) == 4
|
|
|
|
# Test we do not track config
|
|
await hass.config.async_update(latitude=10, longitude=20)
|
|
await hass.async_block_till_done()
|
|
|
|
assert len(mock_weather.mock_calls) == 4
|
|
|
|
entry = hass.config_entries.async_entries()[0]
|
|
await hass.config_entries.async_remove(entry.entry_id)
|
|
assert len(hass.states.async_entity_ids("weather")) == 0
|