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

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
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