Fix solarlog test RuntimeWarning (#125504)

This commit is contained in:
Marc Mueller 2024-09-08 11:43:50 +02:00 committed by GitHub
parent 1f80b803f7
commit d3badb88ef
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1,7 +1,7 @@
"""Test helpers."""
from collections.abc import Generator
from unittest.mock import AsyncMock, patch
from unittest.mock import AsyncMock, MagicMock, patch
import pytest
from solarlog_cli.solarlog_models import InverterData, SolarlogData
@ -53,6 +53,7 @@ def mock_solarlog_connector():
data.inverter_data = INVERTER_DATA
mock_solarlog_api = AsyncMock()
mock_solarlog_api.set_enabled_devices = MagicMock()
mock_solarlog_api.test_connection.return_value = True
mock_solarlog_api.update_data.return_value = data
mock_solarlog_api.update_device_list.return_value = INVERTER_DATA