From ae74fdf252df515965224804a250340e65975663 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Tue, 13 Aug 2024 14:57:24 +0200
Subject: [PATCH] Improve type hints in nzbget tests (#123798)

---
 tests/components/nzbget/conftest.py    | 3 ++-
 tests/components/nzbget/test_init.py   | 4 +++-
 tests/components/nzbget/test_sensor.py | 7 ++++---
 tests/components/nzbget/test_switch.py | 8 ++++++--
 4 files changed, 15 insertions(+), 7 deletions(-)

diff --git a/tests/components/nzbget/conftest.py b/tests/components/nzbget/conftest.py
index 8f48a4306c7..8a980d3ddb0 100644
--- a/tests/components/nzbget/conftest.py
+++ b/tests/components/nzbget/conftest.py
@@ -1,5 +1,6 @@
 """Define fixtures available for all tests."""
 
+from collections.abc import Generator
 from unittest.mock import MagicMock, patch
 
 import pytest
@@ -8,7 +9,7 @@ from . import MOCK_HISTORY, MOCK_STATUS, MOCK_VERSION
 
 
 @pytest.fixture
-def nzbget_api(hass):
+def nzbget_api() -> Generator[MagicMock]:
     """Mock NZBGetApi for easier testing."""
     with patch("homeassistant.components.nzbget.coordinator.NZBGetAPI") as mock_api:
         instance = mock_api.return_value
diff --git a/tests/components/nzbget/test_init.py b/tests/components/nzbget/test_init.py
index a119bb953ce..baf0a37546d 100644
--- a/tests/components/nzbget/test_init.py
+++ b/tests/components/nzbget/test_init.py
@@ -3,6 +3,7 @@
 from unittest.mock import patch
 
 from pynzbgetapi import NZBGetAPIException
+import pytest
 
 from homeassistant.components.nzbget.const import DOMAIN
 from homeassistant.config_entries import ConfigEntryState
@@ -13,7 +14,8 @@ from . import ENTRY_CONFIG, _patch_version, init_integration
 from tests.common import MockConfigEntry
 
 
-async def test_unload_entry(hass: HomeAssistant, nzbget_api) -> None:
+@pytest.mark.usefixtures("nzbget_api")
+async def test_unload_entry(hass: HomeAssistant) -> None:
     """Test successful unload of entry."""
     entry = await init_integration(hass)
 
diff --git a/tests/components/nzbget/test_sensor.py b/tests/components/nzbget/test_sensor.py
index 30a7f262b0b..38f7d8a68c3 100644
--- a/tests/components/nzbget/test_sensor.py
+++ b/tests/components/nzbget/test_sensor.py
@@ -3,6 +3,8 @@
 from datetime import timedelta
 from unittest.mock import patch
 
+import pytest
+
 from homeassistant.components.sensor import SensorDeviceClass
 from homeassistant.const import (
     ATTR_UNIT_OF_MEASUREMENT,
@@ -16,9 +18,8 @@ from homeassistant.util import dt as dt_util
 from . import init_integration
 
 
-async def test_sensors(
-    hass: HomeAssistant, entity_registry: er.EntityRegistry, nzbget_api
-) -> None:
+@pytest.mark.usefixtures("nzbget_api")
+async def test_sensors(hass: HomeAssistant, entity_registry: er.EntityRegistry) -> None:
     """Test the creation and values of the sensors."""
     now = dt_util.utcnow().replace(microsecond=0)
     with patch("homeassistant.components.nzbget.sensor.utcnow", return_value=now):
diff --git a/tests/components/nzbget/test_switch.py b/tests/components/nzbget/test_switch.py
index 1c518486b9f..afb88a7be82 100644
--- a/tests/components/nzbget/test_switch.py
+++ b/tests/components/nzbget/test_switch.py
@@ -1,5 +1,7 @@
 """Test the NZBGet switches."""
 
+from unittest.mock import MagicMock
+
 from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN
 from homeassistant.const import (
     ATTR_ENTITY_ID,
@@ -16,7 +18,7 @@ from . import init_integration
 
 
 async def test_download_switch(
-    hass: HomeAssistant, entity_registry: er.EntityRegistry, nzbget_api
+    hass: HomeAssistant, entity_registry: er.EntityRegistry, nzbget_api: MagicMock
 ) -> None:
     """Test the creation and values of the download switch."""
     instance = nzbget_api.return_value
@@ -44,7 +46,9 @@ async def test_download_switch(
     assert state.state == STATE_OFF
 
 
-async def test_download_switch_services(hass: HomeAssistant, nzbget_api) -> None:
+async def test_download_switch_services(
+    hass: HomeAssistant, nzbget_api: MagicMock
+) -> None:
     """Test download switch services."""
     instance = nzbget_api.return_value