mirror of
https://github.com/home-assistant/core.git
synced 2026-09-26 03:35:14 +01:00
69 lines
1.9 KiB
Python
69 lines
1.9 KiB
Python
"""Test fixtures for caldav."""
|
|
|
|
from unittest.mock import Mock, patch
|
|
|
|
from caldav.lib.url import URL
|
|
import pytest
|
|
|
|
from homeassistant.components.caldav.const import DOMAIN
|
|
from homeassistant.const import (
|
|
CONF_PASSWORD,
|
|
CONF_URL,
|
|
CONF_USERNAME,
|
|
CONF_VERIFY_SSL,
|
|
Platform,
|
|
)
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
TEST_URL = "https://example.com/url-1"
|
|
TEST_USERNAME = "username-1"
|
|
TEST_PASSWORD = "password-1"
|
|
|
|
|
|
@pytest.fixture(name="platforms")
|
|
def mock_platforms() -> list[Platform]:
|
|
"""Fixture to specify platforms to test."""
|
|
return []
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
async def mock_patch_platforms(platforms: list[str]) -> None:
|
|
"""Fixture to set up the integration."""
|
|
with patch(f"homeassistant.components.{DOMAIN}.PLATFORMS", platforms):
|
|
yield
|
|
|
|
|
|
@pytest.fixture(name="calendars")
|
|
def mock_calendars() -> list[Mock]:
|
|
"""Fixture to provide calendars returned by CalDAV client."""
|
|
return []
|
|
|
|
|
|
@pytest.fixture(name="dav_client", autouse=True)
|
|
def mock_dav_client(calendars: list[Mock]) -> Mock:
|
|
"""Fixture to mock the DAVClient."""
|
|
with (
|
|
patch("homeassistant.components.caldav.DAVClient") as mock_client,
|
|
patch("homeassistant.components.caldav.calendar.DAVClient", mock_client),
|
|
patch("homeassistant.components.caldav.config_flow.DAVClient", mock_client),
|
|
):
|
|
mock_client.url = URL(TEST_URL)
|
|
mock_client.return_value.url = URL(TEST_URL)
|
|
mock_client.return_value.get_principal.return_value.get_calendars.return_value = calendars
|
|
yield mock_client
|
|
|
|
|
|
@pytest.fixture(name="config_entry")
|
|
def mock_config_entry() -> MockConfigEntry:
|
|
"""Fixture for a config entry."""
|
|
return MockConfigEntry(
|
|
domain=DOMAIN,
|
|
data={
|
|
CONF_URL: TEST_URL,
|
|
CONF_USERNAME: TEST_USERNAME,
|
|
CONF_PASSWORD: TEST_PASSWORD,
|
|
CONF_VERIFY_SSL: True,
|
|
},
|
|
)
|