49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
|
"""Common fixtures for the Schlage tests."""
|
||
|
|
from collections.abc import Generator
|
||
|
|
from unittest.mock import AsyncMock, patch
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from homeassistant.components.schlage.const import DOMAIN
|
||
|
|
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
|
||
|
|
|
||
|
|
from tests.common import MockConfigEntry
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def mock_config_entry() -> MockConfigEntry:
|
||
|
|
"""Mock ConfigEntry."""
|
||
|
|
return MockConfigEntry(
|
||
|
|
title="asdf@asdf.com",
|
||
|
|
domain=DOMAIN,
|
||
|
|
data={
|
||
|
|
CONF_USERNAME: "asdf@asdf.com",
|
||
|
|
CONF_PASSWORD: "hunter2",
|
||
|
|
},
|
||
|
|
unique_id="abc123",
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def mock_setup_entry() -> Generator[AsyncMock, None, None]:
|
||
|
|
"""Override async_setup_entry."""
|
||
|
|
with patch(
|
||
|
|
"homeassistant.components.schlage.async_setup_entry", return_value=True
|
||
|
|
) as mock_setup_entry:
|
||
|
|
yield mock_setup_entry
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def mock_schlage():
|
||
|
|
"""Mock pyschlage.Schlage."""
|
||
|
|
with patch("pyschlage.Schlage", autospec=True) as mock_schlage:
|
||
|
|
yield mock_schlage.return_value
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def mock_pyschlage_auth():
|
||
|
|
"""Mock pyschlage.Auth."""
|
||
|
|
with patch("pyschlage.Auth", autospec=True) as mock_auth:
|
||
|
|
mock_auth.return_value.user_id = "abc123"
|
||
|
|
yield mock_auth.return_value
|