Workday create repair if named holiday missing (#101201)

This commit is contained in:
G Johansson 2023-12-19 12:33:05 +01:00 committed by GitHub
parent c87f2027d4
commit c64c1c8f08
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 196 additions and 3 deletions

View File

@ -21,7 +21,8 @@ from homeassistant.helpers.entity_platform import (
AddEntitiesCallback,
async_get_current_platform,
)
from homeassistant.util import dt as dt_util
from homeassistant.helpers.issue_registry import IssueSeverity, async_create_issue
from homeassistant.util import dt as dt_util, slugify
from .const import (
ALLOWED_DAYS,
@ -122,6 +123,25 @@ async def async_setup_entry(
LOGGER.debug("Removed %s by name '%s'", holiday, remove_holiday)
except KeyError as unmatched:
LOGGER.warning("No holiday found matching %s", unmatched)
async_create_issue(
hass,
DOMAIN,
f"bad_named_holiday-{entry.entry_id}-{slugify(remove_holiday)}",
is_fixable=True,
is_persistent=False,
severity=IssueSeverity.WARNING,
translation_key="bad_named_holiday",
translation_placeholders={
CONF_COUNTRY: country if country else "-",
"title": entry.title,
CONF_REMOVE_HOLIDAYS: remove_holiday,
},
data={
"entry_id": entry.entry_id,
"country": country,
"named_holiday": remove_holiday,
},
)
LOGGER.debug("Found the following holidays for your configuration:")
for holiday_date, name in sorted(obj_holidays.items()):

View File

@ -18,7 +18,8 @@ from homeassistant.helpers.selector import (
SelectSelectorMode,
)
from .const import CONF_PROVINCE
from .config_flow import validate_custom_dates
from .const import CONF_PROVINCE, CONF_REMOVE_HOLIDAYS
class CountryFixFlow(RepairsFlow):
@ -108,6 +109,76 @@ class CountryFixFlow(RepairsFlow):
)
class HolidayFixFlow(RepairsFlow):
"""Handler for an issue fixing flow."""
def __init__(
self, entry: ConfigEntry, country: str | None, named_holiday: str
) -> None:
"""Create flow."""
self.entry = entry
self.country: str | None = country
self.named_holiday: str = named_holiday
super().__init__()
async def async_step_init(
self, user_input: dict[str, str] | None = None
) -> data_entry_flow.FlowResult:
"""Handle the first step of a fix flow."""
return await self.async_step_named_holiday()
async def async_step_named_holiday(
self, user_input: dict[str, Any] | None = None
) -> data_entry_flow.FlowResult:
"""Handle the options step of a fix flow."""
errors: dict[str, str] = {}
if user_input:
options = dict(self.entry.options)
new_options = {**options, **user_input}
try:
await self.hass.async_add_executor_job(
validate_custom_dates, new_options
)
except Exception: # pylint: disable=broad-except
errors["remove_holidays"] = "remove_holiday_error"
else:
self.hass.config_entries.async_update_entry(
self.entry, options=new_options
)
await self.hass.config_entries.async_reload(self.entry.entry_id)
return self.async_create_entry(data={})
remove_holidays = self.entry.options[CONF_REMOVE_HOLIDAYS]
removed_named_holiday = [
value for value in remove_holidays if value != self.named_holiday
]
new_schema = self.add_suggested_values_to_schema(
vol.Schema(
{
vol.Optional(CONF_REMOVE_HOLIDAYS, default=[]): SelectSelector(
SelectSelectorConfig(
options=[],
multiple=True,
custom_value=True,
mode=SelectSelectorMode.DROPDOWN,
)
),
}
),
{CONF_REMOVE_HOLIDAYS: removed_named_holiday},
)
return self.async_show_form(
step_id="named_holiday",
data_schema=new_schema,
description_placeholders={
CONF_COUNTRY: self.country if self.country else "-",
CONF_REMOVE_HOLIDAYS: self.named_holiday,
"title": self.entry.title,
},
errors=errors,
)
async def async_create_fix_flow(
hass: HomeAssistant,
issue_id: str,
@ -119,6 +190,10 @@ async def async_create_fix_flow(
entry_id = cast(str, entry_id)
entry = hass.config_entries.async_get_entry(entry_id)
if data and (holiday := data.get("named_holiday")) and entry:
# Bad named holiday in configuration
return HolidayFixFlow(entry, data.get("country"), holiday)
if data and entry:
# Country or province does not exist
return CountryFixFlow(entry, data.get("country"))

View File

@ -132,6 +132,26 @@
}
}
}
},
"bad_named_holiday": {
"title": "Configured named holiday {remove_holidays} for {title} does not exist",
"fix_flow": {
"step": {
"named_holiday": {
"title": "[%key:component::workday::issues::bad_named_holiday::title%]",
"description": "Remove named holiday `{remove_holidays}` as it can't be found in country {country}.",
"data": {
"remove_holidays": "[%key:component::workday::config::step::options::data::remove_holidays%]"
},
"data_description": {
"remove_holidays": "[%key:component::workday::config::step::options::data_description::remove_holidays%]"
}
}
},
"error": {
"remove_holiday_error": "[%key:component::workday::config::error::remove_holiday_error%]"
}
}
}
},
"entity": {

View File

@ -7,7 +7,7 @@ from homeassistant.components.repairs.websocket_api import (
RepairsFlowIndexView,
RepairsFlowResourceView,
)
from homeassistant.components.workday.const import DOMAIN
from homeassistant.components.workday.const import CONF_REMOVE_HOLIDAYS, DOMAIN
from homeassistant.const import CONF_COUNTRY
from homeassistant.core import HomeAssistant
from homeassistant.helpers.issue_registry import async_create_issue
@ -16,6 +16,7 @@ from homeassistant.setup import async_setup_component
from . import (
TEST_CONFIG_INCORRECT_COUNTRY,
TEST_CONFIG_INCORRECT_PROVINCE,
TEST_CONFIG_REMOVE_NAMED,
init_integration,
)
@ -324,6 +325,83 @@ async def test_bad_province_none(
assert not issue
async def test_bad_named_holiday(
hass: HomeAssistant,
hass_client: ClientSessionGenerator,
hass_ws_client: WebSocketGenerator,
) -> None:
"""Test fixing bad province selecting none."""
assert await async_setup_component(hass, "repairs", {})
entry = await init_integration(hass, TEST_CONFIG_REMOVE_NAMED)
state = hass.states.get("binary_sensor.workday_sensor")
assert state
ws_client = await hass_ws_client(hass)
client = await hass_client()
await ws_client.send_json({"id": 1, "type": "repairs/list_issues"})
msg = await ws_client.receive_json()
assert msg["success"]
assert len(msg["result"]["issues"]) > 0
issue = None
for i in msg["result"]["issues"]:
if i["issue_id"] == "bad_named_holiday-1-not_a_holiday":
issue = i
assert issue is not None
url = RepairsFlowIndexView.url
resp = await client.post(
url,
json={"handler": DOMAIN, "issue_id": "bad_named_holiday-1-not_a_holiday"},
)
assert resp.status == HTTPStatus.OK
data = await resp.json()
flow_id = data["flow_id"]
assert data["description_placeholders"] == {
CONF_COUNTRY: "US",
CONF_REMOVE_HOLIDAYS: "Not a Holiday",
"title": entry.title,
}
assert data["step_id"] == "named_holiday"
url = RepairsFlowResourceView.url.format(flow_id=flow_id)
resp = await client.post(
url, json={"remove_holidays": ["Christmas", "Not exist 2"]}
)
assert resp.status == HTTPStatus.OK
data = await resp.json()
assert data["errors"] == {
CONF_REMOVE_HOLIDAYS: "remove_holiday_error",
}
url = RepairsFlowResourceView.url.format(flow_id=flow_id)
resp = await client.post(
url, json={"remove_holidays": ["Christmas", "Thanksgiving"]}
)
assert resp.status == HTTPStatus.OK
data = await resp.json()
assert data["type"] == "create_entry"
await hass.async_block_till_done()
state = hass.states.get("binary_sensor.workday_sensor")
assert state
await ws_client.send_json({"id": 2, "type": "repairs/list_issues"})
msg = await ws_client.receive_json()
assert msg["success"]
issue = None
for i in msg["result"]["issues"]:
if i["issue_id"] == "bad_named_holiday-1-not_a_holiday":
issue = i
assert not issue
async def test_other_fixable_issues(
hass: HomeAssistant,
hass_client: ClientSessionGenerator,