mirror of
https://github.com/home-assistant/core.git
synced 2025-11-13 04:50:17 +00:00
45 lines
1.1 KiB
Python
45 lines
1.1 KiB
Python
"""Configuration for Sentry tests."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from unittest.mock import AsyncMock, patch
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components.smarla.config_flow import Connection
|
|
from homeassistant.components.smarla.const import DOMAIN
|
|
from homeassistant.config_entries import SOURCE_USER
|
|
|
|
from . import MOCK_SERIAL_NUMBER
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_config_entry() -> MockConfigEntry:
|
|
"""Create a mock config entry."""
|
|
return MockConfigEntry(
|
|
domain=DOMAIN,
|
|
unique_id=MOCK_SERIAL_NUMBER,
|
|
source=SOURCE_USER,
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_refresh_token_success():
|
|
"""Patch Connection.refresh_token to return True."""
|
|
with patch.object(Connection, "refresh_token", new=AsyncMock(return_value=True)):
|
|
yield
|
|
|
|
|
|
@pytest.fixture
|
|
def malformed_token_patch():
|
|
"""Patch Connection to raise exception."""
|
|
return patch.object(Connection, "__init__", side_effect=ValueError)
|
|
|
|
|
|
@pytest.fixture
|
|
def invalid_auth_patch():
|
|
"""Patch Connection.refresh_token to return False."""
|
|
return patch.object(Connection, "refresh_token", new=AsyncMock(return_value=False))
|