32 lines
838 B
Python
32 lines
838 B
Python
|
"""Fixtures for helpers."""
|
||
|
|
from unittest.mock import Mock, patch
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def mock_integration_frame():
|
||
|
|
"""Mock as if we're calling code from inside an integration."""
|
||
|
|
correct_frame = Mock(
|
||
|
|
filename="/home/paulus/homeassistant/components/hue/light.py",
|
||
|
|
lineno="23",
|
||
|
|
line="self.light.is_on",
|
||
|
|
)
|
||
|
|
with patch(
|
||
|
|
"homeassistant.helpers.frame.extract_stack",
|
||
|
|
return_value=[
|
||
|
|
Mock(
|
||
|
|
filename="/home/paulus/homeassistant/core.py",
|
||
|
|
lineno="23",
|
||
|
|
line="do_something()",
|
||
|
|
),
|
||
|
|
correct_frame,
|
||
|
|
Mock(
|
||
|
|
filename="/home/paulus/aiohue/lights.py",
|
||
|
|
lineno="2",
|
||
|
|
line="something()",
|
||
|
|
),
|
||
|
|
],
|
||
|
|
):
|
||
|
|
yield correct_frame
|