core/tests/components/webostv/__init__.py

"""Tests for the WebOS TV integration."""

from homeassistant.components.webostv.const import DOMAIN
from homeassistant.const import CONF_CLIENT_SECRET, CONF_HOST
from homeassistant.setup import async_setup_component

from .const import CLIENT_KEY, FAKE_UUID, HOST, TV_NAME

from tests.common import MockConfigEntry


async def setup_webostv(hass, unique_id=FAKE_UUID):
    """Initialize webostv and media_player for tests."""
    entry = MockConfigEntry(
        domain=DOMAIN,
        data={
            CONF_HOST: HOST,
            CONF_CLIENT_SECRET: CLIENT_KEY,
        },
        title=TV_NAME,
        unique_id=unique_id,
    )
    entry.add_to_hass(hass)

    await async_setup_component(
        hass,
        DOMAIN,
        {DOMAIN: {CONF_HOST: HOST}},
    )
    await hass.async_block_till_done()

    return entry