* Don't expose flows that aren't initialised. If a flow init does not return immediately then there is a window where our behaviour is screwy: * Can try to configure a flow that isn't ready * Can show notifications for discoveries that might yet return an abort This moves the flow discovery events and notifications to after the flow is initialised and hides flows that don't have a cur_step from async_progress * Fix tradfri test * Black. * Lint fixes
23 lines
562 B
Python
23 lines
562 B
Python
"""Common tradfri test fixtures."""
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from tests.common import mock_coro
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_gateway_info():
|
|
"""Mock get_gateway_info."""
|
|
with patch(
|
|
"homeassistant.components.tradfri.config_flow.get_gateway_info"
|
|
) as mock_gateway:
|
|
yield mock_gateway
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_entry_setup():
|
|
"""Mock entry setup."""
|
|
with patch("homeassistant.components.tradfri.async_setup_entry") as mock_setup:
|
|
mock_setup.return_value = mock_coro(True)
|
|
yield mock_setup
|