60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
"""Global fixtures for Roborock integration."""
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components.roborock.const import (
|
|
CONF_BASE_URL,
|
|
CONF_USER_DATA,
|
|
DOMAIN,
|
|
)
|
|
from homeassistant.const import CONF_USERNAME
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
from .mock_data import BASE_URL, HOME_DATA, PROP, USER_DATA, USER_EMAIL
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
@pytest.fixture(name="bypass_api_fixture")
|
|
def bypass_api_fixture() -> None:
|
|
"""Skip calls to the API."""
|
|
with patch("homeassistant.components.roborock.RoborockMqttClient.connect"), patch(
|
|
"homeassistant.components.roborock.RoborockMqttClient.send_command"
|
|
), patch(
|
|
"homeassistant.components.roborock.coordinator.RoborockLocalClient.get_prop",
|
|
return_value=PROP,
|
|
):
|
|
yield
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_roborock_entry(hass: HomeAssistant) -> MockConfigEntry:
|
|
"""Create a Roborock Entry that has not been setup."""
|
|
mock_entry = MockConfigEntry(
|
|
domain=DOMAIN,
|
|
title=USER_EMAIL,
|
|
data={
|
|
CONF_USERNAME: USER_EMAIL,
|
|
CONF_USER_DATA: USER_DATA.as_dict(),
|
|
CONF_BASE_URL: BASE_URL,
|
|
},
|
|
)
|
|
mock_entry.add_to_hass(hass)
|
|
return mock_entry
|
|
|
|
|
|
@pytest.fixture
|
|
async def setup_entry(
|
|
hass: HomeAssistant, mock_roborock_entry: MockConfigEntry
|
|
) -> MockConfigEntry:
|
|
"""Set up the Roborock platform."""
|
|
with patch(
|
|
"homeassistant.components.roborock.RoborockApiClient.get_home_data",
|
|
return_value=HOME_DATA,
|
|
), patch("homeassistant.components.roborock.RoborockMqttClient.get_networking"):
|
|
assert await async_setup_component(hass, DOMAIN, {})
|
|
await hass.async_block_till_done()
|
|
return mock_roborock_entry
|