evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add outage map alerts to PECO (#69825)

Browse source
This commit is contained in:
IceBotYT 2022-04-18 17:22:14 -04:00 • committed by GitHub
parent c87992715f
commit 6c75eaa1bc
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
8 changed files with 86 additions and 33 deletions
Download patch file Download diff file Expand all files Collapse all files

22
tests/components/peco/test_sensor.py
View file

@ -1,7 +1,7 @@
"""Test the PECO Outage Counter sensors."""
from unittest.mock import patch
from peco import OutageResults
from peco import AlertResults, OutageResults
import pytest
from homeassistant.components.peco.const import DOMAIN
@ -42,8 +42,14 @@ async def test_sensor_available(
customers_served=789,
),
):
assert await hass.config_entries.async_setup(config_entry.entry_id)
await hass.async_block_till_done()
with patch(
"peco.PecoOutageApi.get_map_alerts",
return_value=AlertResults(
alert_content="Testing 1234", alert_title="Testing 4321"
),
):
assert await hass.config_entries.async_setup(config_entry.entry_id)
await hass.async_block_till_done()
assert hass.data[DOMAIN]
entries = hass.config_entries.async_entries(DOMAIN)
@ -69,8 +75,14 @@ async def test_sensor_available(
customers_served=789,
),
):
assert await hass.config_entries.async_setup(config_entry.entry_id)
await hass.async_block_till_done()
with patch(
"peco.PecoOutageApi.get_map_alerts",
return_value=AlertResults(
alert_content="Testing 1234", alert_title="Testing 4321"
),
):
assert await hass.config_entries.async_setup(config_entry.entry_id)
await hass.async_block_till_done()
entries = hass.config_entries.async_entries(DOMAIN)
assert len(entries) == 2