core/tests/components/drop_connect/conftest.py

"""Define fixtures available for all tests."""
import pytest

from homeassistant.components.drop_connect.const import (
    CONF_COMMAND_TOPIC,
    CONF_DATA_TOPIC,
    CONF_DEVICE_DESC,
    CONF_DEVICE_ID,
    CONF_DEVICE_NAME,
    CONF_DEVICE_OWNER_ID,
    CONF_DEVICE_TYPE,
    CONF_HUB_ID,
    DOMAIN,
)
from homeassistant.core import HomeAssistant

from tests.common import MockConfigEntry


@pytest.fixture
def config_entry_hub(hass: HomeAssistant):
    """Config entry version 1 fixture."""
    return MockConfigEntry(
        domain=DOMAIN,
        unique_id="DROP-1_C0FFEE_255",
        data={
            CONF_COMMAND_TOPIC: "drop_connect/DROP-1_C0FFEE/255/cmd",
            CONF_DATA_TOPIC: "drop_connect/DROP-1_C0FFEE/255/#",
            CONF_DEVICE_DESC: "Hub",
            CONF_DEVICE_ID: 255,
            CONF_DEVICE_NAME: "Hub DROP-1_C0FFEE",
            CONF_DEVICE_TYPE: "hub",
            CONF_HUB_ID: "DROP-1_C0FFEE",
            CONF_DEVICE_OWNER_ID: "DROP-1_C0FFEE_255",
        },
        version=1,
    )


@pytest.fixture
def config_entry_salt(hass: HomeAssistant):
    """Config entry version 1 fixture."""
    return MockConfigEntry(
        domain=DOMAIN,
        unique_id="DROP-1_C0FFEE_8",
        data={
            CONF_COMMAND_TOPIC: "drop_connect/DROP-1_C0FFEE/8/cmd",
            CONF_DATA_TOPIC: "drop_connect/DROP-1_C0FFEE/8/#",
            CONF_DEVICE_DESC: "Salt Sensor",
            CONF_DEVICE_ID: 8,
            CONF_DEVICE_NAME: "Salt Sensor",
            CONF_DEVICE_TYPE: "salt",
            CONF_HUB_ID: "DROP-1_C0FFEE",
            CONF_DEVICE_OWNER_ID: "DROP-1_C0FFEE_255",
        },
        version=1,
    )


@pytest.fixture
def config_entry_leak(hass: HomeAssistant):
    """Config entry version 1 fixture."""
    return MockConfigEntry(
        domain=DOMAIN,
        unique_id="DROP-1_C0FFEE_20",
        data={
            CONF_COMMAND_TOPIC: "drop_connect/DROP-1_C0FFEE/20/cmd",
            CONF_DATA_TOPIC: "drop_connect/DROP-1_C0FFEE/20/#",
            CONF_DEVICE_DESC: "Leak Detector",
            CONF_DEVICE_ID: 20,
            CONF_DEVICE_NAME: "Leak Detector",
            CONF_DEVICE_TYPE: "leak",
            CONF_HUB_ID: "DROP-1_C0FFEE",
            CONF_DEVICE_OWNER_ID: "DROP-1_C0FFEE_255",
        },
        version=1,
    )


@pytest.fixture
def config_entry_softener(hass: HomeAssistant):
    """Config entry version 1 fixture."""
    return MockConfigEntry(
        domain=DOMAIN,
        unique_id="DROP-1_C0FFEE_0",
        data={
            CONF_COMMAND_TOPIC: "drop_connect/DROP-1_C0FFEE/0/cmd",
            CONF_DATA_TOPIC: "drop_connect/DROP-1_C0FFEE/0/#",
            CONF_DEVICE_DESC: "Softener",
            CONF_DEVICE_ID: 0,
            CONF_DEVICE_NAME: "Softener",
            CONF_DEVICE_TYPE: "soft",
            CONF_HUB_ID: "DROP-1_C0FFEE",
            CONF_DEVICE_OWNER_ID: "DROP-1_C0FFEE_255",
        },
        version=1,
    )


@pytest.fixture
def config_entry_filter(hass: HomeAssistant):
    """Config entry version 1 fixture."""
    return MockConfigEntry(
        domain=DOMAIN,
        unique_id="DROP-1_C0FFEE_4",
        data={
            CONF_COMMAND_TOPIC: "drop_connect/DROP-1_C0FFEE/4/cmd",
            CONF_DATA_TOPIC: "drop_connect/DROP-1_C0FFEE/4/#",
            CONF_DEVICE_DESC: "Filter",
            CONF_DEVICE_ID: 4,
            CONF_DEVICE_NAME: "Filter",
            CONF_DEVICE_TYPE: "filt",
            CONF_HUB_ID: "DROP-1_C0FFEE",
            CONF_DEVICE_OWNER_ID: "DROP-1_C0FFEE_255",
        },
        version=1,
    )


@pytest.fixture
def config_entry_protection_valve(hass: HomeAssistant):
    """Config entry version 1 fixture."""
    return MockConfigEntry(
        domain=DOMAIN,
        unique_id="DROP-1_C0FFEE_78",
        data={
            CONF_COMMAND_TOPIC: "drop_connect/DROP-1_C0FFEE/78/cmd",
            CONF_DATA_TOPIC: "drop_connect/DROP-1_C0FFEE/78/#",
            CONF_DEVICE_DESC: "Protection Valve",
            CONF_DEVICE_ID: 78,
            CONF_DEVICE_NAME: "Protection Valve",
            CONF_DEVICE_TYPE: "pv",
            CONF_HUB_ID: "DROP-1_C0FFEE",
            CONF_DEVICE_OWNER_ID: "DROP-1_C0FFEE_255",
        },
        version=1,
    )


@pytest.fixture
def config_entry_pump_controller(hass: HomeAssistant):
    """Config entry version 1 fixture."""
    return MockConfigEntry(
        domain=DOMAIN,
        unique_id="DROP-1_C0FFEE_83",
        data={
            CONF_COMMAND_TOPIC: "drop_connect/DROP-1_C0FFEE/83/cmd",
            CONF_DATA_TOPIC: "drop_connect/DROP-1_C0FFEE/83/#",
            CONF_DEVICE_DESC: "Pump Controller",
            CONF_DEVICE_ID: 83,
            CONF_DEVICE_NAME: "Pump Controller",
            CONF_DEVICE_TYPE: "pc",
            CONF_HUB_ID: "DROP-1_C0FFEE",
            CONF_DEVICE_OWNER_ID: "DROP-1_C0FFEE_255",
        },
        version=1,
    )


@pytest.fixture
def config_entry_ro_filter(hass: HomeAssistant):
    """Config entry version 1 fixture."""
    return MockConfigEntry(
        domain=DOMAIN,
        unique_id="DROP-1_C0FFEE_255",
        data={
            CONF_COMMAND_TOPIC: "drop_connect/DROP-1_C0FFEE/95/cmd",
            CONF_DATA_TOPIC: "drop_connect/DROP-1_C0FFEE/95/#",
            CONF_DEVICE_DESC: "RO Filter",
            CONF_DEVICE_ID: 95,
            CONF_DEVICE_NAME: "RO Filter",
            CONF_DEVICE_TYPE: "ro",
            CONF_HUB_ID: "DROP-1_C0FFEE",
            CONF_DEVICE_OWNER_ID: "DROP-1_C0FFEE_255",
        },
        version=1,
    )