self-hosted/home-assistant-core
self-hosted/home-assistant-core
1
0
Fork 0
mirror of https://github.com/home-assistant/core synced 2024-10-03 01:07:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

Ensure github tests add config entry before updating it (#110430)

Browse source
This commit is contained in:
J. Nick Koston 2024-02-12 22:00:44 -06:00 committed by GitHub
parent 704ad67917
commit 0e1cdc1f79
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 20 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/components/github/common.py
View file

@ -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()

2
tests/components/github/test_config_flow.py
View file

@ -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()

5
tests/components/github/test_diagnostics.py
View file

@ -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,

15
tests/components/github/test_init.py
View file

@ -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