Add monkeypatch type hints to shelly tests (home-assistant#121057)

* Add monkeypatch type hints to shelly tests * Improve
cereal2nd · Jul 3, 2024 · ac57eb7 · ac57eb7
1 parent c33cbf8
commit ac57eb7
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 6 deletions.
diff --git a/tests/components/shelly/bluetooth/test_scanner.py b/tests/components/shelly/bluetooth/test_scanner.py
@@ -12,7 +12,9 @@
 from .. import init_integration, inject_rpc_device_event
 
 
-async def test_scanner_v1(hass: HomeAssistant, mock_rpc_device, monkeypatch) -> None:
+async def test_scanner_v1(
+    hass: HomeAssistant, mock_rpc_device, monkeypatch: pytest.MonkeyPatch
+) -> None:
     """Test injecting data into the scanner v1."""
     await init_integration(
         hass, 2, options={CONF_BLE_SCANNER_MODE: BLEScannerMode.ACTIVE}
@@ -50,7 +52,9 @@ async def test_scanner_v1(hass: HomeAssistant, mock_rpc_device, monkeypatch) ->
     assert ble_device is None
 
 
-async def test_scanner_v2(hass: HomeAssistant, mock_rpc_device, monkeypatch) -> None:
+async def test_scanner_v2(
+    hass: HomeAssistant, mock_rpc_device, monkeypatch: pytest.MonkeyPatch
+) -> None:
     """Test injecting data into the scanner v2."""
     await init_integration(
         hass, 2, options={CONF_BLE_SCANNER_MODE: BLEScannerMode.ACTIVE}
@@ -93,7 +97,7 @@ async def test_scanner_v2(hass: HomeAssistant, mock_rpc_device, monkeypatch) ->
 
 
 async def test_scanner_ignores_non_ble_events(
-    hass: HomeAssistant, mock_rpc_device, monkeypatch
+    hass: HomeAssistant, mock_rpc_device, monkeypatch: pytest.MonkeyPatch
 ) -> None:
     """Test injecting non ble data into the scanner."""
     await init_integration(
@@ -119,7 +123,10 @@ async def test_scanner_ignores_non_ble_events(
 
 
 async def test_scanner_ignores_wrong_version_and_logs(
-    hass: HomeAssistant, mock_rpc_device, monkeypatch, caplog: pytest.LogCaptureFixture
+    hass: HomeAssistant,
+    mock_rpc_device,
+    monkeypatch: pytest.MonkeyPatch,
+    caplog: pytest.LogCaptureFixture,
 ) -> None:
     """Test injecting wrong version of ble data into the scanner."""
     await init_integration(
@@ -152,7 +159,10 @@ async def test_scanner_ignores_wrong_version_and_logs(
 
 
 async def test_scanner_warns_on_corrupt_event(
-    hass: HomeAssistant, mock_rpc_device, monkeypatch, caplog: pytest.LogCaptureFixture
+    hass: HomeAssistant,
+    mock_rpc_device,
+    monkeypatch: pytest.MonkeyPatch,
+    caplog: pytest.LogCaptureFixture,
 ) -> None:
     """Test injecting garbage ble data into the scanner."""
     await init_integration(

diff --git a/tests/components/shelly/test_switch.py b/tests/components/shelly/test_switch.py
@@ -187,7 +187,7 @@ async def test_block_device_unique_ids(
 
 
 async def test_block_set_state_connection_error(
-    hass: HomeAssistant, mock_block_device, monkeypatch
+    hass: HomeAssistant, mock_block_device, monkeypatch: pytest.MonkeyPatch
 ) -> None:
     """Test block device set state connection error."""
     monkeypatch.setattr(