Fix tests after refactor

2025-08-03 11:43:53 +02:00
parent f1d10458c6
commit 64e1ca124f
5 changed files with 60 additions and 48 deletions
--- a/tests/test_maintenance.py
+++ b/tests/test_maintenance.py
@@ -43,8 +43,10 @@ class TestMaintenanceCommands:

    def test_list_maintenances_api_error(self, mock_client, capsys):
        """Test maintenance listing with API error"""
+        from uptime_kuma_api import UptimeKumaException
+
        # Setup
-        mock_client.api.get_maintenances.side_effect = Exception("API Error")
+        mock_client.api.get_maintenances.side_effect = UptimeKumaException("API Error")

        maintenance_commands = MaintenanceCommands(mock_client)