mirror of
https://github.com/home-assistant/core.git
synced 2026-08-07 13:55:32 +01:00
26 lines
811 B
Python
26 lines
811 B
Python
"""Tests for the LiteLLM integration."""
|
|
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
async def setup_integration(hass: HomeAssistant, config_entry: MockConfigEntry) -> None:
|
|
"""Fixture for setting up the component."""
|
|
config_entry.add_to_hass(hass)
|
|
|
|
await hass.config_entries.async_setup(config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
|
|
def get_subentry_id(mock_config_entry: MockConfigEntry, subentry_type: str) -> str:
|
|
"""Get the subentry ID for a given type."""
|
|
ids = [
|
|
subentry_id
|
|
for subentry_id, subentry in mock_config_entry.subentries.items()
|
|
if subentry.subentry_type == subentry_type
|
|
]
|
|
if not ids:
|
|
raise ValueError(f"No subentry found for type {subentry_type}")
|
|
return ids[0]
|