43a5852561
* Fix habitica entry unload clean up * Fix service remove * Add entry setup and unload test * Fix config flow tests * Assert service
36 lines
1,016 B
Python
36 lines
1,016 B
Python
"""Test the habitica init module."""
|
|
from homeassistant.components.habitica.const import (
|
|
DEFAULT_URL,
|
|
DOMAIN,
|
|
SERVICE_API_CALL,
|
|
)
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
async def test_entry_setup_unload(hass, aioclient_mock):
|
|
"""Test integration setup and unload."""
|
|
entry = MockConfigEntry(
|
|
domain=DOMAIN,
|
|
unique_id="test-api-user",
|
|
data={
|
|
"api_user": "test-api-user",
|
|
"api_key": "test-api-key",
|
|
"url": DEFAULT_URL,
|
|
},
|
|
)
|
|
entry.add_to_hass(hass)
|
|
|
|
aioclient_mock.get(
|
|
"https://habitica.com/api/v3/user",
|
|
json={"data": {"api_user": "test-api-user", "profile": {"name": "test_user"}}},
|
|
)
|
|
|
|
assert await hass.config_entries.async_setup(entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
assert hass.services.has_service(DOMAIN, SERVICE_API_CALL)
|
|
|
|
assert await hass.config_entries.async_unload(entry.entry_id)
|
|
|
|
assert not hass.services.has_service(DOMAIN, SERVICE_API_CALL)
|