Add missing mock in braviatv config flow tests (#89419)

This commit is contained in:
epenet 2023-03-09 13:05:32 +01:00 committed by GitHub
parent 9faf251321
commit b0631fed1d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 16 additions and 6 deletions

View File

@ -0,0 +1,14 @@
"""Test fixtures for Bravia TV."""
from collections.abc import Generator
from unittest.mock import AsyncMock, patch
import pytest
@pytest.fixture
def mock_setup_entry() -> Generator[AsyncMock, None, None]:
"""Override async_setup_entry."""
with patch(
"homeassistant.components.braviatv.async_setup_entry", return_value=True
) as mock_setup_entry:
yield mock_setup_entry

View File

@ -84,6 +84,8 @@ FAKE_BRAVIA_SSDP = ssdp.SsdpServiceInfo(
}, },
) )
pytestmark = pytest.mark.usefixtures("mock_setup_entry")
async def test_show_form(hass: HomeAssistant) -> None: async def test_show_form(hass: HomeAssistant) -> None:
"""Test that the form is served with no input.""" """Test that the form is served with no input."""
@ -111,8 +113,6 @@ async def test_ssdp_discovery(hass: HomeAssistant) -> None:
), patch("pybravia.BraviaClient.set_wol_mode"), patch( ), patch("pybravia.BraviaClient.set_wol_mode"), patch(
"pybravia.BraviaClient.get_system_info", "pybravia.BraviaClient.get_system_info",
return_value=BRAVIA_SYSTEM_INFO, return_value=BRAVIA_SYSTEM_INFO,
), patch(
"homeassistant.components.braviatv.async_setup_entry", return_value=True
): ):
result = await hass.config_entries.flow.async_configure( result = await hass.config_entries.flow.async_configure(
result["flow_id"], user_input={} result["flow_id"], user_input={}
@ -302,8 +302,6 @@ async def test_create_entry(hass: HomeAssistant) -> None:
), patch("pybravia.BraviaClient.set_wol_mode"), patch( ), patch("pybravia.BraviaClient.set_wol_mode"), patch(
"pybravia.BraviaClient.get_system_info", "pybravia.BraviaClient.get_system_info",
return_value=BRAVIA_SYSTEM_INFO, return_value=BRAVIA_SYSTEM_INFO,
), patch(
"homeassistant.components.braviatv.async_setup_entry", return_value=True
): ):
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, context={"source": SOURCE_USER}, data={CONF_HOST: "bravia-host"} DOMAIN, context={"source": SOURCE_USER}, data={CONF_HOST: "bravia-host"}
@ -343,8 +341,6 @@ async def test_create_entry_psk(hass: HomeAssistant) -> None:
), patch( ), patch(
"pybravia.BraviaClient.get_system_info", "pybravia.BraviaClient.get_system_info",
return_value=BRAVIA_SYSTEM_INFO, return_value=BRAVIA_SYSTEM_INFO,
), patch(
"homeassistant.components.braviatv.async_setup_entry", return_value=True
): ):
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, context={"source": SOURCE_USER}, data={CONF_HOST: "bravia-host"} DOMAIN, context={"source": SOURCE_USER}, data={CONF_HOST: "bravia-host"}