mirror of
https://github.com/home-assistant/core.git
synced 2025-07-19 11:17:21 +00:00
Add tests for buttons in Habitica integration (#128194)
* Add tests for button platform * update tests * Add skill buttons * Assert state, add fixtures/parametrization * entity as list
This commit is contained in:
Manu
GitHub
parent
ffc0651d89
commit
e602a464db
@ -0,0 +1,39 @@
|
|||||||
|
{
|
||||||
|
"data": {
|
||||||
|
"api_user": "test-api-user",
|
||||||
|
"profile": { "name": "test-user" },
|
||||||
|
"stats": {
|
||||||
|
"buffs": {
|
||||||
|
"str": 0,
|
||||||
|
"int": 0,
|
||||||
|
"per": 0,
|
||||||
|
"con": 0,
|
||||||
|
"stealth": 0,
|
||||||
|
"streaks": true,
|
||||||
|
"seafoam": false,
|
||||||
|
"shinySeed": false,
|
||||||
|
"snowball": false,
|
||||||
|
"spookySparkles": false
|
||||||
|
},
|
||||||
|
"hp": 50,
|
||||||
|
"mp": 50,
|
||||||
|
"exp": 737,
|
||||||
|
"gp": 0,
|
||||||
|
"lvl": 5,
|
||||||
|
"class": "wizard",
|
||||||
|
"maxHealth": 50,
|
||||||
|
"maxMP": 166,
|
||||||
|
"toNextLevel": 880,
|
||||||
|
"points": 0
|
||||||
|
},
|
||||||
|
"preferences": {
|
||||||
|
"sleep": false,
|
||||||
|
"automaticAllocation": false,
|
||||||
|
"disableClasses": false
|
||||||
|
},
|
||||||
|
"flags": {
|
||||||
|
"classSelected": true
|
||||||
|
},
|
||||||
|
"needsCron": false
|
||||||
|
}
|
||||||
|
}
|
||||||
40
tests/components/habitica/fixtures/healer_fixture.json
Normal file
40
tests/components/habitica/fixtures/healer_fixture.json
Normal file
@ -0,0 +1,40 @@
|
|||||||
|
{
|
||||||
|
"data": {
|
||||||
|
"api_user": "test-api-user",
|
||||||
|
"profile": { "name": "test-user" },
|
||||||
|
"stats": {
|
||||||
|
"buffs": {
|
||||||
|
"str": 0,
|
||||||
|
"int": 0,
|
||||||
|
"per": 0,
|
||||||
|
"con": 0,
|
||||||
|
"stealth": 0,
|
||||||
|
"streaks": false,
|
||||||
|
"seafoam": false,
|
||||||
|
"shinySeed": false,
|
||||||
|
"snowball": false,
|
||||||
|
"spookySparkles": false
|
||||||
|
},
|
||||||
|
"hp": 45,
|
||||||
|
"mp": 50.89999999999998,
|
||||||
|
"exp": 737,
|
||||||
|
"gp": 137.62587214609795,
|
||||||
|
"lvl": 38,
|
||||||
|
"class": "healer",
|
||||||
|
"maxHealth": 50,
|
||||||
|
"maxMP": 166,
|
||||||
|
"toNextLevel": 880,
|
||||||
|
"points": 5
|
||||||
|
},
|
||||||
|
"preferences": {
|
||||||
|
"sleep": false,
|
||||||
|
"automaticAllocation": true,
|
||||||
|
"disableClasses": false
|
||||||
|
},
|
||||||
|
"flags": {
|
||||||
|
"classSelected": true
|
||||||
|
},
|
||||||
|
"needsCron": true,
|
||||||
|
"lastCron": "2024-09-21T22:01:55.586Z"
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,39 @@
|
|||||||
|
{
|
||||||
|
"data": {
|
||||||
|
"api_user": "test-api-user",
|
||||||
|
"profile": { "name": "test-user" },
|
||||||
|
"stats": {
|
||||||
|
"buffs": {
|
||||||
|
"str": 0,
|
||||||
|
"int": 0,
|
||||||
|
"per": 0,
|
||||||
|
"con": 0,
|
||||||
|
"stealth": 0,
|
||||||
|
"streaks": false,
|
||||||
|
"seafoam": false,
|
||||||
|
"shinySeed": false,
|
||||||
|
"snowball": false,
|
||||||
|
"spookySparkles": false
|
||||||
|
},
|
||||||
|
"hp": 50,
|
||||||
|
"mp": 10,
|
||||||
|
"exp": 737,
|
||||||
|
"gp": 0,
|
||||||
|
"lvl": 34,
|
||||||
|
"class": "healer",
|
||||||
|
"maxHealth": 50,
|
||||||
|
"maxMP": 166,
|
||||||
|
"toNextLevel": 880,
|
||||||
|
"points": 0
|
||||||
|
},
|
||||||
|
"preferences": {
|
||||||
|
"sleep": false,
|
||||||
|
"automaticAllocation": false,
|
||||||
|
"disableClasses": false
|
||||||
|
},
|
||||||
|
"flags": {
|
||||||
|
"classSelected": true
|
||||||
|
},
|
||||||
|
"needsCron": false
|
||||||
|
}
|
||||||
|
}
|
||||||
40
tests/components/habitica/fixtures/rogue_fixture.json
Normal file
40
tests/components/habitica/fixtures/rogue_fixture.json
Normal file
@ -0,0 +1,40 @@
|
|||||||
|
{
|
||||||
|
"data": {
|
||||||
|
"api_user": "test-api-user",
|
||||||
|
"profile": { "name": "test-user" },
|
||||||
|
"stats": {
|
||||||
|
"buffs": {
|
||||||
|
"str": 0,
|
||||||
|
"int": 0,
|
||||||
|
"per": 0,
|
||||||
|
"con": 0,
|
||||||
|
"stealth": 0,
|
||||||
|
"streaks": false,
|
||||||
|
"seafoam": false,
|
||||||
|
"shinySeed": false,
|
||||||
|
"snowball": false,
|
||||||
|
"spookySparkles": false
|
||||||
|
},
|
||||||
|
"hp": 0,
|
||||||
|
"mp": 50.89999999999998,
|
||||||
|
"exp": 737,
|
||||||
|
"gp": 137.62587214609795,
|
||||||
|
"lvl": 38,
|
||||||
|
"class": "rogue",
|
||||||
|
"maxHealth": 50,
|
||||||
|
"maxMP": 166,
|
||||||
|
"toNextLevel": 880,
|
||||||
|
"points": 5
|
||||||
|
},
|
||||||
|
"preferences": {
|
||||||
|
"sleep": false,
|
||||||
|
"automaticAllocation": true,
|
||||||
|
"disableClasses": false
|
||||||
|
},
|
||||||
|
"flags": {
|
||||||
|
"classSelected": true
|
||||||
|
},
|
||||||
|
"needsCron": true,
|
||||||
|
"lastCron": "2024-09-21T22:01:55.586Z"
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,39 @@
|
|||||||
|
{
|
||||||
|
"data": {
|
||||||
|
"api_user": "test-api-user",
|
||||||
|
"profile": { "name": "test-user" },
|
||||||
|
"stats": {
|
||||||
|
"buffs": {
|
||||||
|
"str": 0,
|
||||||
|
"int": 0,
|
||||||
|
"per": 0,
|
||||||
|
"con": 0,
|
||||||
|
"stealth": 0,
|
||||||
|
"streaks": true,
|
||||||
|
"seafoam": false,
|
||||||
|
"shinySeed": false,
|
||||||
|
"snowball": false,
|
||||||
|
"spookySparkles": false
|
||||||
|
},
|
||||||
|
"hp": 50,
|
||||||
|
"mp": 20,
|
||||||
|
"exp": 737,
|
||||||
|
"gp": 0,
|
||||||
|
"lvl": 38,
|
||||||
|
"class": "rogue",
|
||||||
|
"maxHealth": 50,
|
||||||
|
"maxMP": 166,
|
||||||
|
"toNextLevel": 880,
|
||||||
|
"points": 0
|
||||||
|
},
|
||||||
|
"preferences": {
|
||||||
|
"sleep": false,
|
||||||
|
"automaticAllocation": false,
|
||||||
|
"disableClasses": false
|
||||||
|
},
|
||||||
|
"flags": {
|
||||||
|
"classSelected": true
|
||||||
|
},
|
||||||
|
"needsCron": false
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,39 @@
|
|||||||
|
{
|
||||||
|
"data": {
|
||||||
|
"api_user": "test-api-user",
|
||||||
|
"profile": { "name": "test-user" },
|
||||||
|
"stats": {
|
||||||
|
"buffs": {
|
||||||
|
"str": 0,
|
||||||
|
"int": 0,
|
||||||
|
"per": 0,
|
||||||
|
"con": 0,
|
||||||
|
"stealth": 4,
|
||||||
|
"streaks": false,
|
||||||
|
"seafoam": false,
|
||||||
|
"shinySeed": false,
|
||||||
|
"snowball": false,
|
||||||
|
"spookySparkles": false
|
||||||
|
},
|
||||||
|
"hp": 50,
|
||||||
|
"mp": 50,
|
||||||
|
"exp": 737,
|
||||||
|
"gp": 0,
|
||||||
|
"lvl": 38,
|
||||||
|
"class": "rogue",
|
||||||
|
"maxHealth": 50,
|
||||||
|
"maxMP": 166,
|
||||||
|
"toNextLevel": 880,
|
||||||
|
"points": 0
|
||||||
|
},
|
||||||
|
"preferences": {
|
||||||
|
"sleep": false,
|
||||||
|
"automaticAllocation": false,
|
||||||
|
"disableClasses": false
|
||||||
|
},
|
||||||
|
"flags": {
|
||||||
|
"classSelected": true
|
||||||
|
},
|
||||||
|
"needsCron": false
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -3,6 +3,18 @@
|
|||||||
"api_user": "test-api-user",
|
"api_user": "test-api-user",
|
||||||
"profile": { "name": "test-user" },
|
"profile": { "name": "test-user" },
|
||||||
"stats": {
|
"stats": {
|
||||||
|
"buffs": {
|
||||||
|
"str": 0,
|
||||||
|
"int": 0,
|
||||||
|
"per": 0,
|
||||||
|
"con": 0,
|
||||||
|
"stealth": 0,
|
||||||
|
"streaks": false,
|
||||||
|
"seafoam": false,
|
||||||
|
"shinySeed": false,
|
||||||
|
"snowball": false,
|
||||||
|
"spookySparkles": false
|
||||||
|
},
|
||||||
"hp": 0,
|
"hp": 0,
|
||||||
"mp": 50.89999999999998,
|
"mp": 50.89999999999998,
|
||||||
"exp": 737,
|
"exp": 737,
|
||||||
@ -16,7 +28,11 @@
|
|||||||
},
|
},
|
||||||
"preferences": {
|
"preferences": {
|
||||||
"sleep": false,
|
"sleep": false,
|
||||||
"automaticAllocation": true
|
"automaticAllocation": true,
|
||||||
|
"disableClasses": false
|
||||||
|
},
|
||||||
|
"flags": {
|
||||||
|
"classSelected": true
|
||||||
},
|
},
|
||||||
"needsCron": true,
|
"needsCron": true,
|
||||||
"lastCron": "2024-09-21T22:01:55.586Z"
|
"lastCron": "2024-09-21T22:01:55.586Z"
|
||||||
|
|||||||
40
tests/components/habitica/fixtures/warrior_fixture.json
Normal file
40
tests/components/habitica/fixtures/warrior_fixture.json
Normal file
@ -0,0 +1,40 @@
|
|||||||
|
{
|
||||||
|
"data": {
|
||||||
|
"api_user": "test-api-user",
|
||||||
|
"profile": { "name": "test-user" },
|
||||||
|
"stats": {
|
||||||
|
"buffs": {
|
||||||
|
"str": 0,
|
||||||
|
"int": 0,
|
||||||
|
"per": 0,
|
||||||
|
"con": 0,
|
||||||
|
"stealth": 0,
|
||||||
|
"streaks": false,
|
||||||
|
"seafoam": false,
|
||||||
|
"shinySeed": false,
|
||||||
|
"snowball": false,
|
||||||
|
"spookySparkles": false
|
||||||
|
},
|
||||||
|
"hp": 50,
|
||||||
|
"mp": 50.89999999999998,
|
||||||
|
"exp": 737,
|
||||||
|
"gp": 137.62587214609795,
|
||||||
|
"lvl": 38,
|
||||||
|
"class": "warrior",
|
||||||
|
"maxHealth": 50,
|
||||||
|
"maxMP": 166,
|
||||||
|
"toNextLevel": 880,
|
||||||
|
"points": 5
|
||||||
|
},
|
||||||
|
"preferences": {
|
||||||
|
"sleep": false,
|
||||||
|
"automaticAllocation": true,
|
||||||
|
"disableClasses": false
|
||||||
|
},
|
||||||
|
"flags": {
|
||||||
|
"classSelected": true
|
||||||
|
},
|
||||||
|
"needsCron": true,
|
||||||
|
"lastCron": "2024-09-21T22:01:55.586Z"
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,39 @@
|
|||||||
|
{
|
||||||
|
"data": {
|
||||||
|
"api_user": "test-api-user",
|
||||||
|
"profile": { "name": "test-user" },
|
||||||
|
"stats": {
|
||||||
|
"buffs": {
|
||||||
|
"str": 0,
|
||||||
|
"int": 0,
|
||||||
|
"per": 0,
|
||||||
|
"con": 0,
|
||||||
|
"stealth": 0,
|
||||||
|
"streaks": false,
|
||||||
|
"seafoam": false,
|
||||||
|
"shinySeed": false,
|
||||||
|
"snowball": false,
|
||||||
|
"spookySparkles": false
|
||||||
|
},
|
||||||
|
"hp": 50,
|
||||||
|
"mp": 10,
|
||||||
|
"exp": 737,
|
||||||
|
"gp": 0,
|
||||||
|
"lvl": 34,
|
||||||
|
"class": "warrior",
|
||||||
|
"maxHealth": 50,
|
||||||
|
"maxMP": 166,
|
||||||
|
"toNextLevel": 880,
|
||||||
|
"points": 0
|
||||||
|
},
|
||||||
|
"preferences": {
|
||||||
|
"sleep": false,
|
||||||
|
"automaticAllocation": false,
|
||||||
|
"disableClasses": false
|
||||||
|
},
|
||||||
|
"flags": {
|
||||||
|
"classSelected": true
|
||||||
|
},
|
||||||
|
"needsCron": false
|
||||||
|
}
|
||||||
|
}
|
||||||
40
tests/components/habitica/fixtures/wizard_fixture.json
Normal file
40
tests/components/habitica/fixtures/wizard_fixture.json
Normal file
@ -0,0 +1,40 @@
|
|||||||
|
{
|
||||||
|
"data": {
|
||||||
|
"api_user": "test-api-user",
|
||||||
|
"profile": { "name": "test-user" },
|
||||||
|
"stats": {
|
||||||
|
"buffs": {
|
||||||
|
"str": 0,
|
||||||
|
"int": 0,
|
||||||
|
"per": 0,
|
||||||
|
"con": 0,
|
||||||
|
"stealth": 0,
|
||||||
|
"streaks": false,
|
||||||
|
"seafoam": false,
|
||||||
|
"shinySeed": false,
|
||||||
|
"snowball": false,
|
||||||
|
"spookySparkles": false
|
||||||
|
},
|
||||||
|
"hp": 50,
|
||||||
|
"mp": 50.89999999999998,
|
||||||
|
"exp": 737,
|
||||||
|
"gp": 137.62587214609795,
|
||||||
|
"lvl": 38,
|
||||||
|
"class": "wizard",
|
||||||
|
"maxHealth": 50,
|
||||||
|
"maxMP": 166,
|
||||||
|
"toNextLevel": 880,
|
||||||
|
"points": 5
|
||||||
|
},
|
||||||
|
"preferences": {
|
||||||
|
"sleep": false,
|
||||||
|
"automaticAllocation": true,
|
||||||
|
"disableClasses": false
|
||||||
|
},
|
||||||
|
"flags": {
|
||||||
|
"classSelected": true
|
||||||
|
},
|
||||||
|
"needsCron": true,
|
||||||
|
"lastCron": "2024-09-21T22:01:55.586Z"
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,39 @@
|
|||||||
|
{
|
||||||
|
"data": {
|
||||||
|
"api_user": "test-api-user",
|
||||||
|
"profile": { "name": "test-user" },
|
||||||
|
"stats": {
|
||||||
|
"buffs": {
|
||||||
|
"str": 0,
|
||||||
|
"int": 0,
|
||||||
|
"per": 0,
|
||||||
|
"con": 0,
|
||||||
|
"stealth": 0,
|
||||||
|
"streaks": true,
|
||||||
|
"seafoam": false,
|
||||||
|
"shinySeed": false,
|
||||||
|
"snowball": false,
|
||||||
|
"spookySparkles": false
|
||||||
|
},
|
||||||
|
"hp": 50,
|
||||||
|
"mp": 50,
|
||||||
|
"exp": 737,
|
||||||
|
"gp": 0,
|
||||||
|
"lvl": 34,
|
||||||
|
"class": "wizard",
|
||||||
|
"maxHealth": 50,
|
||||||
|
"maxMP": 166,
|
||||||
|
"toNextLevel": 880,
|
||||||
|
"points": 0
|
||||||
|
},
|
||||||
|
"preferences": {
|
||||||
|
"sleep": false,
|
||||||
|
"automaticAllocation": false,
|
||||||
|
"disableClasses": false
|
||||||
|
},
|
||||||
|
"flags": {
|
||||||
|
"classSelected": true
|
||||||
|
},
|
||||||
|
"needsCron": false
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,39 @@
|
|||||||
|
{
|
||||||
|
"data": {
|
||||||
|
"api_user": "test-api-user",
|
||||||
|
"profile": { "name": "test-user" },
|
||||||
|
"stats": {
|
||||||
|
"buffs": {
|
||||||
|
"str": 0,
|
||||||
|
"int": 0,
|
||||||
|
"per": 0,
|
||||||
|
"con": 0,
|
||||||
|
"stealth": 0,
|
||||||
|
"streaks": false,
|
||||||
|
"seafoam": false,
|
||||||
|
"shinySeed": false,
|
||||||
|
"snowball": false,
|
||||||
|
"spookySparkles": false
|
||||||
|
},
|
||||||
|
"hp": 50,
|
||||||
|
"mp": 10,
|
||||||
|
"exp": 737,
|
||||||
|
"gp": 0,
|
||||||
|
"lvl": 34,
|
||||||
|
"class": "wizard",
|
||||||
|
"maxHealth": 50,
|
||||||
|
"maxMP": 166,
|
||||||
|
"toNextLevel": 880,
|
||||||
|
"points": 0
|
||||||
|
},
|
||||||
|
"preferences": {
|
||||||
|
"sleep": false,
|
||||||
|
"automaticAllocation": false,
|
||||||
|
"disableClasses": false
|
||||||
|
},
|
||||||
|
"flags": {
|
||||||
|
"classSelected": true
|
||||||
|
},
|
||||||
|
"needsCron": false
|
||||||
|
}
|
||||||
|
}
|
||||||
1631
tests/components/habitica/snapshots/test_button.ambr
Normal file
1631
tests/components/habitica/snapshots/test_button.ambr
Normal file
File diff suppressed because it is too large
Load Diff
332
tests/components/habitica/test_button.py
Normal file
332
tests/components/habitica/test_button.py
Normal file
@ -0,0 +1,332 @@
|
|||||||
|
"""Tests for Habitica button platform."""
|
||||||
|
|
||||||
|
from collections.abc import Generator
|
||||||
|
from http import HTTPStatus
|
||||||
|
import re
|
||||||
|
from unittest.mock import patch
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
from syrupy.assertion import SnapshotAssertion
|
||||||
|
|
||||||
|
from homeassistant.components.button import DOMAIN as BUTTON_DOMAIN, SERVICE_PRESS
|
||||||
|
from homeassistant.components.habitica.const import DEFAULT_URL, DOMAIN
|
||||||
|
from homeassistant.config_entries import ConfigEntryState
|
||||||
|
from homeassistant.const import ATTR_ENTITY_ID, STATE_UNAVAILABLE, Platform
|
||||||
|
from homeassistant.core import HomeAssistant
|
||||||
|
from homeassistant.exceptions import HomeAssistantError, ServiceValidationError
|
||||||
|
from homeassistant.helpers import entity_registry as er
|
||||||
|
|
||||||
|
from .conftest import mock_called_with
|
||||||
|
|
||||||
|
from tests.common import MockConfigEntry, load_json_object_fixture, snapshot_platform
|
||||||
|
from tests.test_util.aiohttp import AiohttpClientMocker
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture(autouse=True)
|
||||||
|
def button_only() -> Generator[None]:
|
||||||
|
"""Enable only the button platform."""
|
||||||
|
with patch(
|
||||||
|
"homeassistant.components.habitica.PLATFORMS",
|
||||||
|
[Platform.BUTTON],
|
||||||
|
):
|
||||||
|
yield
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
"fixture",
|
||||||
|
[
|
||||||
|
"wizard_fixture",
|
||||||
|
"rogue_fixture",
|
||||||
|
"warrior_fixture",
|
||||||
|
"healer_fixture",
|
||||||
|
],
|
||||||
|
)
|
||||||
|
async def test_buttons(
|
||||||
|
hass: HomeAssistant,
|
||||||
|
config_entry: MockConfigEntry,
|
||||||
|
aioclient_mock: AiohttpClientMocker,
|
||||||
|
snapshot: SnapshotAssertion,
|
||||||
|
entity_registry: er.EntityRegistry,
|
||||||
|
fixture: str,
|
||||||
|
) -> None:
|
||||||
|
"""Test button entities."""
|
||||||
|
aioclient_mock.get(
|
||||||
|
f"{DEFAULT_URL}/api/v3/user",
|
||||||
|
json=load_json_object_fixture(f"{fixture}.json", DOMAIN),
|
||||||
|
)
|
||||||
|
aioclient_mock.get(
|
||||||
|
f"{DEFAULT_URL}/api/v3/tasks/user",
|
||||||
|
params={"type": "completedTodos"},
|
||||||
|
json=load_json_object_fixture("completed_todos.json", DOMAIN),
|
||||||
|
)
|
||||||
|
aioclient_mock.get(
|
||||||
|
f"{DEFAULT_URL}/api/v3/tasks/user",
|
||||||
|
json=load_json_object_fixture("tasks.json", DOMAIN),
|
||||||
|
)
|
||||||
|
config_entry.add_to_hass(hass)
|
||||||
|
await hass.config_entries.async_setup(config_entry.entry_id)
|
||||||
|
await hass.async_block_till_done()
|
||||||
|
|
||||||
|
assert config_entry.state is ConfigEntryState.LOADED
|
||||||
|
|
||||||
|
await snapshot_platform(hass, entity_registry, snapshot, config_entry.entry_id)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
("entity_id", "api_url", "fixture"),
|
||||||
|
[
|
||||||
|
("button.test_user_allocate_all_stat_points", "user/allocate-now", "user"),
|
||||||
|
("button.test_user_buy_a_health_potion", "user/buy-health-potion", "user"),
|
||||||
|
("button.test_user_revive_from_death", "user/revive", "user"),
|
||||||
|
("button.test_user_start_my_day", "cron", "user"),
|
||||||
|
(
|
||||||
|
"button.test_user_chilling_frost",
|
||||||
|
"user/class/cast/frost",
|
||||||
|
"wizard_fixture",
|
||||||
|
),
|
||||||
|
(
|
||||||
|
"button.test_user_earthquake",
|
||||||
|
"user/class/cast/earth",
|
||||||
|
"wizard_fixture",
|
||||||
|
),
|
||||||
|
(
|
||||||
|
"button.test_user_ethereal_surge",
|
||||||
|
"user/class/cast/mpheal",
|
||||||
|
"wizard_fixture",
|
||||||
|
),
|
||||||
|
(
|
||||||
|
"button.test_user_stealth",
|
||||||
|
"user/class/cast/stealth",
|
||||||
|
"rogue_fixture",
|
||||||
|
),
|
||||||
|
(
|
||||||
|
"button.test_user_tools_of_the_trade",
|
||||||
|
"user/class/cast/toolsOfTrade",
|
||||||
|
"rogue_fixture",
|
||||||
|
),
|
||||||
|
(
|
||||||
|
"button.test_user_defensive_stance",
|
||||||
|
"user/class/cast/defensiveStance",
|
||||||
|
"warrior_fixture",
|
||||||
|
),
|
||||||
|
(
|
||||||
|
"button.test_user_intimidating_gaze",
|
||||||
|
"user/class/cast/intimidate",
|
||||||
|
"warrior_fixture",
|
||||||
|
),
|
||||||
|
(
|
||||||
|
"button.test_user_valorous_presence",
|
||||||
|
"user/class/cast/valorousPresence",
|
||||||
|
"warrior_fixture",
|
||||||
|
),
|
||||||
|
(
|
||||||
|
"button.test_user_healing_light",
|
||||||
|
"user/class/cast/heal",
|
||||||
|
"healer_fixture",
|
||||||
|
),
|
||||||
|
(
|
||||||
|
"button.test_user_protective_aura",
|
||||||
|
"user/class/cast/protectAura",
|
||||||
|
"healer_fixture",
|
||||||
|
),
|
||||||
|
(
|
||||||
|
"button.test_user_searing_brightness",
|
||||||
|
"user/class/cast/brightness",
|
||||||
|
"healer_fixture",
|
||||||
|
),
|
||||||
|
(
|
||||||
|
"button.test_user_blessing",
|
||||||
|
"user/class/cast/healAll",
|
||||||
|
"healer_fixture",
|
||||||
|
),
|
||||||
|
],
|
||||||
|
)
|
||||||
|
async def test_button_press(
|
||||||
|
hass: HomeAssistant,
|
||||||
|
config_entry: MockConfigEntry,
|
||||||
|
aioclient_mock: AiohttpClientMocker,
|
||||||
|
entity_id: str,
|
||||||
|
api_url: str,
|
||||||
|
fixture: str,
|
||||||
|
) -> None:
|
||||||
|
"""Test button press method."""
|
||||||
|
aioclient_mock.get(
|
||||||
|
f"{DEFAULT_URL}/api/v3/user",
|
||||||
|
json=load_json_object_fixture(f"{fixture}.json", DOMAIN),
|
||||||
|
)
|
||||||
|
aioclient_mock.get(
|
||||||
|
f"{DEFAULT_URL}/api/v3/tasks/user",
|
||||||
|
params={"type": "completedTodos"},
|
||||||
|
json=load_json_object_fixture("completed_todos.json", DOMAIN),
|
||||||
|
)
|
||||||
|
aioclient_mock.get(
|
||||||
|
f"{DEFAULT_URL}/api/v3/tasks/user",
|
||||||
|
json=load_json_object_fixture("tasks.json", DOMAIN),
|
||||||
|
)
|
||||||
|
config_entry.add_to_hass(hass)
|
||||||
|
await hass.config_entries.async_setup(config_entry.entry_id)
|
||||||
|
await hass.async_block_till_done()
|
||||||
|
|
||||||
|
assert config_entry.state is ConfigEntryState.LOADED
|
||||||
|
|
||||||
|
aioclient_mock.post(f"{DEFAULT_URL}/api/v3/{api_url}", json={"data": None})
|
||||||
|
|
||||||
|
await hass.services.async_call(
|
||||||
|
BUTTON_DOMAIN,
|
||||||
|
SERVICE_PRESS,
|
||||||
|
{ATTR_ENTITY_ID: entity_id},
|
||||||
|
blocking=True,
|
||||||
|
)
|
||||||
|
|
||||||
|
assert mock_called_with(aioclient_mock, "post", f"{DEFAULT_URL}/api/v3/{api_url}")
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
("entity_id", "api_url"),
|
||||||
|
[
|
||||||
|
("button.test_user_allocate_all_stat_points", "user/allocate-now"),
|
||||||
|
("button.test_user_buy_a_health_potion", "user/buy-health-potion"),
|
||||||
|
("button.test_user_revive_from_death", "user/revive"),
|
||||||
|
("button.test_user_start_my_day", "cron"),
|
||||||
|
("button.test_user_chilling_frost", "user/class/cast/frost"),
|
||||||
|
("button.test_user_earthquake", "user/class/cast/earth"),
|
||||||
|
("button.test_user_ethereal_surge", "user/class/cast/mpheal"),
|
||||||
|
],
|
||||||
|
ids=[
|
||||||
|
"allocate-points",
|
||||||
|
"health-potion",
|
||||||
|
"revive",
|
||||||
|
"run-cron",
|
||||||
|
"chilling frost",
|
||||||
|
"earthquake",
|
||||||
|
"ethereal surge",
|
||||||
|
],
|
||||||
|
)
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
("status_code", "msg", "exception"),
|
||||||
|
[
|
||||||
|
(
|
||||||
|
HTTPStatus.TOO_MANY_REQUESTS,
|
||||||
|
"Currently rate limited",
|
||||||
|
ServiceValidationError,
|
||||||
|
),
|
||||||
|
(
|
||||||
|
HTTPStatus.BAD_REQUEST,
|
||||||
|
"Unable to connect to Habitica, try again later",
|
||||||
|
HomeAssistantError,
|
||||||
|
),
|
||||||
|
(
|
||||||
|
HTTPStatus.UNAUTHORIZED,
|
||||||
|
"Unable to carry out this action",
|
||||||
|
ServiceValidationError,
|
||||||
|
),
|
||||||
|
],
|
||||||
|
)
|
||||||
|
async def test_button_press_exceptions(
|
||||||
|
hass: HomeAssistant,
|
||||||
|
config_entry: MockConfigEntry,
|
||||||
|
mock_habitica: AiohttpClientMocker,
|
||||||
|
entity_id: str,
|
||||||
|
api_url: str,
|
||||||
|
status_code: HTTPStatus,
|
||||||
|
msg: str,
|
||||||
|
exception: Exception,
|
||||||
|
) -> None:
|
||||||
|
"""Test button press exceptions."""
|
||||||
|
|
||||||
|
config_entry.add_to_hass(hass)
|
||||||
|
await hass.config_entries.async_setup(config_entry.entry_id)
|
||||||
|
await hass.async_block_till_done()
|
||||||
|
|
||||||
|
assert config_entry.state is ConfigEntryState.LOADED
|
||||||
|
|
||||||
|
mock_habitica.post(
|
||||||
|
f"{DEFAULT_URL}/api/v3/{api_url}",
|
||||||
|
status=status_code,
|
||||||
|
json={"data": None},
|
||||||
|
)
|
||||||
|
|
||||||
|
with pytest.raises(exception, match=msg):
|
||||||
|
await hass.services.async_call(
|
||||||
|
BUTTON_DOMAIN,
|
||||||
|
SERVICE_PRESS,
|
||||||
|
{ATTR_ENTITY_ID: entity_id},
|
||||||
|
blocking=True,
|
||||||
|
)
|
||||||
|
|
||||||
|
assert mock_called_with(mock_habitica, "post", f"{DEFAULT_URL}/api/v3/{api_url}")
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
("fixture", "entity_ids"),
|
||||||
|
[
|
||||||
|
(
|
||||||
|
"common_buttons_unavailable",
|
||||||
|
[
|
||||||
|
"button.test_user_allocate_all_stat_points",
|
||||||
|
"button.test_user_revive_from_death",
|
||||||
|
"button.test_user_buy_a_health_potion",
|
||||||
|
"button.test_user_start_my_day",
|
||||||
|
],
|
||||||
|
),
|
||||||
|
(
|
||||||
|
"wizard_skills_unavailable",
|
||||||
|
[
|
||||||
|
"button.test_user_chilling_frost",
|
||||||
|
"button.test_user_earthquake",
|
||||||
|
"button.test_user_ethereal_surge",
|
||||||
|
],
|
||||||
|
),
|
||||||
|
("wizard_frost_unavailable", ["button.test_user_chilling_frost"]),
|
||||||
|
(
|
||||||
|
"rogue_skills_unavailable",
|
||||||
|
["button.test_user_tools_of_the_trade", "button.test_user_stealth"],
|
||||||
|
),
|
||||||
|
("rogue_stealth_unavailable", ["button.test_user_stealth"]),
|
||||||
|
(
|
||||||
|
"warrior_skills_unavailable",
|
||||||
|
[
|
||||||
|
"button.test_user_defensive_stance",
|
||||||
|
"button.test_user_intimidating_gaze",
|
||||||
|
"button.test_user_valorous_presence",
|
||||||
|
],
|
||||||
|
),
|
||||||
|
(
|
||||||
|
"healer_skills_unavailable",
|
||||||
|
[
|
||||||
|
"button.test_user_healing_light",
|
||||||
|
"button.test_user_protective_aura",
|
||||||
|
"button.test_user_searing_brightness",
|
||||||
|
"button.test_user_blessing",
|
||||||
|
],
|
||||||
|
),
|
||||||
|
],
|
||||||
|
)
|
||||||
|
async def test_button_unavailable(
|
||||||
|
hass: HomeAssistant,
|
||||||
|
config_entry: MockConfigEntry,
|
||||||
|
aioclient_mock: AiohttpClientMocker,
|
||||||
|
fixture: str,
|
||||||
|
entity_ids: list[str],
|
||||||
|
) -> None:
|
||||||
|
"""Test buttons are unavailable if conditions are not met."""
|
||||||
|
|
||||||
|
aioclient_mock.get(
|
||||||
|
f"{DEFAULT_URL}/api/v3/user",
|
||||||
|
json=load_json_object_fixture(f"{fixture}.json", DOMAIN),
|
||||||
|
)
|
||||||
|
aioclient_mock.get(
|
||||||
|
f"{DEFAULT_URL}/api/v3/tasks/user",
|
||||||
|
json=load_json_object_fixture("tasks.json", DOMAIN),
|
||||||
|
)
|
||||||
|
aioclient_mock.get(re.compile(r".*"), json={"data": []})
|
||||||
|
|
||||||
|
config_entry.add_to_hass(hass)
|
||||||
|
await hass.config_entries.async_setup(config_entry.entry_id)
|
||||||
|
await hass.async_block_till_done()
|
||||||
|
|
||||||
|
assert config_entry.state is ConfigEntryState.LOADED
|
||||||
|
|
||||||
|
for entity_id in entity_ids:
|
||||||
|
assert (state := hass.states.get(entity_id))
|
||||||
|
assert state.state == STATE_UNAVAILABLE
|
||||||
Loading…
x
Reference in New Issue
Block a user