fix: correct tests for filter/pump

Author: cryptk

tests/test_filter_pump.py

@@ -197,7 +197,7 @@ async def test_filter_turn_on(self, mock_omni, sample_filter_config, mock_teleme
         filter_obj._api.async_set_equipment.assert_called_once_with(
             pool_id=7,
             equipment_id=8,
-            is_on=True,
+            is_on=filter_obj.last_speed,
         )
 
     @pytest.mark.asyncio
@@ -220,44 +220,44 @@ async def test_filter_run_preset_speed_low(self, mock_omni, sample_filter_config
         """Test run_preset_speed with LOW preset."""
         sample_filter_config.bow_id = 7
         filter_obj = Filter(mock_omni, sample_filter_config, mock_telemetry)
-        filter_obj._api.async_set_filter_speed = AsyncMock()
+        filter_obj._api.async_set_equipment = AsyncMock()
 
         await filter_obj.run_preset_speed(FilterSpeedPresets.LOW)
 
-        filter_obj._api.async_set_filter_speed.assert_called_once_with(
+        filter_obj._api.async_set_equipment.assert_called_once_with(
             pool_id=7,
             equipment_id=8,
-            speed=40,
+            is_on=40,
         )
 
     @pytest.mark.asyncio
     async def test_filter_run_preset_speed_medium(self, mock_omni, sample_filter_config, mock_telemetry):
         """Test run_preset_speed with MEDIUM preset."""
         sample_filter_config.bow_id = 7
         filter_obj = Filter(mock_omni, sample_filter_config, mock_telemetry)
-        filter_obj._api.async_set_filter_speed = AsyncMock()
+        filter_obj._api.async_set_equipment = AsyncMock()
 
         await filter_obj.run_preset_speed(FilterSpeedPresets.MEDIUM)
 
-        filter_obj._api.async_set_filter_speed.assert_called_once_with(
+        filter_obj._api.async_set_equipment.assert_called_once_with(
             pool_id=7,
             equipment_id=8,
-            speed=60,
+            is_on=filter_obj.medium_speed,
         )
 
     @pytest.mark.asyncio
     async def test_filter_run_preset_speed_high(self, mock_omni, sample_filter_config, mock_telemetry):
         """Test run_preset_speed with HIGH preset."""
         sample_filter_config.bow_id = 7
         filter_obj = Filter(mock_omni, sample_filter_config, mock_telemetry)
-        filter_obj._api.async_set_filter_speed = AsyncMock()
+        filter_obj._api.async_set_equipment = AsyncMock()
 
         await filter_obj.run_preset_speed(FilterSpeedPresets.HIGH)
 
-        filter_obj._api.async_set_filter_speed.assert_called_once_with(
+        filter_obj._api.async_set_equipment.assert_called_once_with(
             pool_id=7,
             equipment_id=8,
-            speed=80,
+            is_on=filter_obj.high_speed,
         )