36 lines
729 B
Python
36 lines
729 B
Python
"""Common helpers for the datetime entity component tests."""
|
|
|
|
from unittest import mock
|
|
|
|
MOCK_DATA = {
|
|
"host": "localhost",
|
|
"port": 8125,
|
|
}
|
|
|
|
MOCK_OPTIONS = {
|
|
"prefix": "hass",
|
|
"rate": 1,
|
|
}
|
|
|
|
MOCK_CONFIG = {**MOCK_DATA, **MOCK_OPTIONS}
|
|
|
|
MOCK_YAML_INVALID = {
|
|
"host": "127.0.0.1",
|
|
"port": 65535,
|
|
"prefix": "failtest",
|
|
"rate": 1,
|
|
}
|
|
|
|
|
|
CONNECTION_TEST_METRIC = "connection_test"
|
|
|
|
|
|
def create_mock_state(entity_id, state, attributes=None):
|
|
"""Helper to create a mock state object."""
|
|
mock_state = mock.MagicMock()
|
|
mock_state.entity_id = entity_id
|
|
mock_state.state = state
|
|
mock_state.domain = entity_id.split(".")[0]
|
|
mock_state.attributes = attributes or {}
|
|
return mock_state
|