111 lines
3.4 KiB
Python
111 lines
3.4 KiB
Python
"""Define test fixtures for Enphase Envoy."""
|
|
import json
|
|
from unittest.mock import AsyncMock, Mock, patch
|
|
|
|
import pytest
|
|
|
|
from homeassistant.components.enphase_envoy import DOMAIN
|
|
from homeassistant.const import CONF_HOST, CONF_NAME, CONF_PASSWORD, CONF_USERNAME
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
from tests.common import MockConfigEntry, load_fixture
|
|
|
|
|
|
@pytest.fixture(name="config_entry")
|
|
def config_entry_fixture(hass: HomeAssistant, config, serial_number):
|
|
"""Define a config entry fixture."""
|
|
entry = MockConfigEntry(
|
|
domain=DOMAIN,
|
|
title=f"Envoy {serial_number}" if serial_number else "Envoy",
|
|
unique_id=serial_number,
|
|
data=config,
|
|
)
|
|
entry.add_to_hass(hass)
|
|
return entry
|
|
|
|
|
|
@pytest.fixture(name="config")
|
|
def config_fixture():
|
|
"""Define a config entry data fixture."""
|
|
return {
|
|
CONF_HOST: "1.1.1.1",
|
|
CONF_NAME: "Envoy 1234",
|
|
CONF_USERNAME: "test-username",
|
|
CONF_PASSWORD: "test-password",
|
|
}
|
|
|
|
|
|
@pytest.fixture(name="gateway_data", scope="package")
|
|
def gateway_data_fixture():
|
|
"""Define a fixture to return gateway data."""
|
|
return json.loads(load_fixture("data.json", "enphase_envoy"))
|
|
|
|
|
|
@pytest.fixture(name="inverters_production_data", scope="package")
|
|
def inverters_production_data_fixture():
|
|
"""Define a fixture to return inverter production data."""
|
|
return json.loads(load_fixture("inverters_production.json", "enphase_envoy"))
|
|
|
|
|
|
@pytest.fixture(name="mock_envoy_reader")
|
|
def mock_envoy_reader_fixture(
|
|
gateway_data,
|
|
mock_get_data,
|
|
mock_get_full_serial_number,
|
|
mock_inverters_production,
|
|
serial_number,
|
|
):
|
|
"""Define a mocked EnvoyReader fixture."""
|
|
mock_envoy_reader = Mock(
|
|
getData=mock_get_data,
|
|
get_full_serial_number=mock_get_full_serial_number,
|
|
inverters_production=mock_inverters_production,
|
|
)
|
|
|
|
for key, value in gateway_data.items():
|
|
setattr(mock_envoy_reader, key, AsyncMock(return_value=value))
|
|
|
|
return mock_envoy_reader
|
|
|
|
|
|
@pytest.fixture(name="mock_get_full_serial_number")
|
|
def mock_get_full_serial_number_fixture(serial_number):
|
|
"""Define a mocked EnvoyReader.get_full_serial_number fixture."""
|
|
return AsyncMock(return_value=serial_number)
|
|
|
|
|
|
@pytest.fixture(name="mock_get_data")
|
|
def mock_get_data_fixture():
|
|
"""Define a mocked EnvoyReader.getData fixture."""
|
|
return AsyncMock()
|
|
|
|
|
|
@pytest.fixture(name="mock_inverters_production")
|
|
def mock_inverters_production_fixture(inverters_production_data):
|
|
"""Define a mocked EnvoyReader.inverters_production fixture."""
|
|
return AsyncMock(return_value=inverters_production_data)
|
|
|
|
|
|
@pytest.fixture(name="setup_enphase_envoy")
|
|
async def setup_enphase_envoy_fixture(hass, config, mock_envoy_reader):
|
|
"""Define a fixture to set up Enphase Envoy."""
|
|
with patch(
|
|
"homeassistant.components.enphase_envoy.config_flow.EnvoyReader",
|
|
return_value=mock_envoy_reader,
|
|
), patch(
|
|
"homeassistant.components.enphase_envoy.EnvoyReader",
|
|
return_value=mock_envoy_reader,
|
|
), patch(
|
|
"homeassistant.components.enphase_envoy.PLATFORMS", []
|
|
):
|
|
assert await async_setup_component(hass, DOMAIN, config)
|
|
await hass.async_block_till_done()
|
|
yield
|
|
|
|
|
|
@pytest.fixture(name="serial_number")
|
|
def serial_number_fixture():
|
|
"""Define a serial number fixture."""
|
|
return "1234"
|