diff --git a/tests/components/zimi/conftest.py b/tests/components/zimi/conftest.py index 44d898deffb..b26c2f89784 100644 --- a/tests/components/zimi/conftest.py +++ b/tests/components/zimi/conftest.py @@ -1,6 +1,6 @@ """Test fixtures for Zimi component.""" -from unittest.mock import patch +from unittest.mock import MagicMock, patch import pytest @@ -19,6 +19,8 @@ def mock_api(): """Mock the API with defaults.""" with patch("homeassistant.components.zimi.async_connect_to_controller") as mock: mock_api = mock.return_value + mock_api.describe = MagicMock() + mock_api.disconnect = MagicMock() mock_api.connect.return_value = True mock_api.mac = INPUT_MAC mock_api.brand = API_INFO["brand"]