diff --git a/tests/components/github/common.py b/tests/components/github/common.py index 4bd26183299..223722dccd8 100644 --- a/tests/components/github/common.py +++ b/tests/components/github/common.py @@ -18,6 +18,7 @@ async def setup_github_integration( hass: HomeAssistant, mock_config_entry: MockConfigEntry, aioclient_mock: AiohttpClientMocker, + add_entry_to_hass: bool = True, ) -> None: """Mock setting up the integration.""" headers = json.loads(load_fixture("base_headers.json", DOMAIN)) @@ -41,7 +42,8 @@ async def setup_github_integration( json=json.loads(load_fixture("graphql.json", DOMAIN)), headers=headers, ) - mock_config_entry.add_to_hass(hass) + if add_entry_to_hass: + mock_config_entry.add_to_hass(hass) setup_result = await hass.config_entries.async_setup(mock_config_entry.entry_id) await hass.async_block_till_done() diff --git a/tests/components/github/test_config_flow.py b/tests/components/github/test_config_flow.py index 67746d891d1..0edb90bee14 100644 --- a/tests/components/github/test_config_flow.py +++ b/tests/components/github/test_config_flow.py @@ -260,13 +260,13 @@ async def test_options_flow( mock_setup_entry: None, ) -> None: """Test options flow.""" + mock_config_entry.add_to_hass(hass) hass.config_entries.async_update_entry( mock_config_entry, options={ CONF_REPOSITORIES: ["homeassistant/core", "homeassistant/architecture"] }, ) - mock_config_entry.add_to_hass(hass) await hass.config_entries.async_setup(mock_config_entry.entry_id) await hass.async_block_till_done() diff --git a/tests/components/github/test_diagnostics.py b/tests/components/github/test_diagnostics.py index c358ccc9e6d..806a0ae33cc 100644 --- a/tests/components/github/test_diagnostics.py +++ b/tests/components/github/test_diagnostics.py @@ -25,6 +25,7 @@ async def test_entry_diagnostics( aioclient_mock: AiohttpClientMocker, ) -> None: """Test config entry diagnostics.""" + mock_config_entry.add_to_hass(hass) hass.config_entries.async_update_entry( mock_config_entry, options={CONF_REPOSITORIES: ["home-assistant/core"]}, @@ -43,7 +44,9 @@ async def test_entry_diagnostics( headers={"Content-Type": "application/json"}, ) - await setup_github_integration(hass, mock_config_entry, aioclient_mock) + await setup_github_integration( + hass, mock_config_entry, aioclient_mock, add_entry_to_hass=False + ) result = await get_diagnostics_for_config_entry( hass, hass_client, diff --git a/tests/components/github/test_init.py b/tests/components/github/test_init.py index f2301056ff8..03cc2e92720 100644 --- a/tests/components/github/test_init.py +++ b/tests/components/github/test_init.py @@ -21,11 +21,14 @@ async def test_device_registry_cleanup( caplog: pytest.LogCaptureFixture, ) -> None: """Test that we remove untracked repositories from the decvice registry.""" + mock_config_entry.add_to_hass(hass) hass.config_entries.async_update_entry( mock_config_entry, options={CONF_REPOSITORIES: ["home-assistant/core"]}, ) - await setup_github_integration(hass, mock_config_entry, aioclient_mock) + await setup_github_integration( + hass, mock_config_entry, aioclient_mock, add_entry_to_hass=False + ) devices = dr.async_entries_for_config_entry( registry=device_registry, @@ -62,12 +65,15 @@ async def test_subscription_setup( aioclient_mock: AiohttpClientMocker, ) -> None: """Test that we setup event subscription.""" + mock_config_entry.add_to_hass(hass) hass.config_entries.async_update_entry( mock_config_entry, options={CONF_REPOSITORIES: ["home-assistant/core"]}, pref_disable_polling=False, ) - await setup_github_integration(hass, mock_config_entry, aioclient_mock) + await setup_github_integration( + hass, mock_config_entry, aioclient_mock, add_entry_to_hass=False + ) assert ( "https://api.github.com/repos/home-assistant/core/events" in x[1] for x in aioclient_mock.mock_calls @@ -82,12 +88,15 @@ async def test_subscription_setup_polling_disabled( aioclient_mock: AiohttpClientMocker, ) -> None: """Test that we do not setup event subscription if polling is disabled.""" + mock_config_entry.add_to_hass(hass) hass.config_entries.async_update_entry( mock_config_entry, options={CONF_REPOSITORIES: ["home-assistant/core"]}, pref_disable_polling=True, ) - await setup_github_integration(hass, mock_config_entry, aioclient_mock) + await setup_github_integration( + hass, mock_config_entry, aioclient_mock, add_entry_to_hass=False + ) assert ( "https://api.github.com/repos/home-assistant/core/events" not in x[1] for x in aioclient_mock.mock_calls