diff --git a/tests/components/nzbget/__init__.py b/tests/components/nzbget/__init__.py index d3216b62ef3..d8fa2f87233 100644 --- a/tests/components/nzbget/__init__.py +++ b/tests/components/nzbget/__init__.py @@ -59,14 +59,9 @@ MOCK_HISTORY = [ ] -async def init_integration( - hass, - *, - data: dict = ENTRY_CONFIG, - options: dict = ENTRY_OPTIONS, -) -> MockConfigEntry: +async def init_integration(hass) -> MockConfigEntry: """Set up the NZBGet integration in Home Assistant.""" - entry = MockConfigEntry(domain=DOMAIN, data=data, options=options) + entry = MockConfigEntry(domain=DOMAIN, data=ENTRY_CONFIG, options=ENTRY_OPTIONS) entry.add_to_hass(hass) await hass.config_entries.async_setup(entry.entry_id) @@ -82,17 +77,17 @@ def _patch_async_setup_entry(return_value=True): ) -def _patch_history(return_value=MOCK_HISTORY): +def _patch_history(): return patch( "homeassistant.components.nzbget.coordinator.NZBGetAPI.history", - return_value=return_value, + return_value=MOCK_HISTORY, ) -def _patch_status(return_value=MOCK_STATUS): +def _patch_status(): return patch( "homeassistant.components.nzbget.coordinator.NZBGetAPI.status", - return_value=return_value, + return_value=MOCK_STATUS, )