jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-19 11:17:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Clean up Whois tests use entity_registry_enabled_by_default fixture (#68113)

Browse Source
This commit is contained in:
Franck Nijhof 2022-03-14 11:41:56 +01:00 committed by GitHub
parent fb703ddc5d
commit a29afc3731
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/components/whois/conftest.py
View File

@ -124,13 +124,3 @@ async def init_integration_missing_some_attrs(
await hass.async_block_till_done() await hass.async_block_till_done()
return mock_config_entry return mock_config_entry
@pytest.fixture
def enable_all_entities() -> Generator[AsyncMock, None, None]:
"""Test fixture that ensures all entities are enabled in the registry."""
with patch(
"homeassistant.helpers.entity.Entity.entity_registry_enabled_default",
return_value=True,
) as mock_entity_registry_enabled_by_default:
yield mock_entity_registry_enabled_by_default

6
tests/components/whois/test_sensor.py
View File

@ -22,7 +22,7 @@ from tests.common import MockConfigEntry, async_fire_time_changed
@pytest.mark.freeze_time("2022-01-01 12:00:00", tz_offset=0) @pytest.mark.freeze_time("2022-01-01 12:00:00", tz_offset=0)
async def test_whois_sensors( async def test_whois_sensors(
hass: HomeAssistant, hass: HomeAssistant,
enable_all_entities: AsyncMock, entity_registry_enabled_by_default: AsyncMock,
init_integration: MockConfigEntry, init_integration: MockConfigEntry,
) -> None: ) -> None:
"""Test the Whois sensors.""" """Test the Whois sensors."""
@ -146,7 +146,7 @@ async def test_whois_sensors(
@pytest.mark.freeze_time("2022-01-01 12:00:00", tz_offset=0) @pytest.mark.freeze_time("2022-01-01 12:00:00", tz_offset=0)
async def test_whois_sensors_missing_some_attrs( async def test_whois_sensors_missing_some_attrs(
hass: HomeAssistant, hass: HomeAssistant,
enable_all_entities: AsyncMock, entity_registry_enabled_by_default: AsyncMock,
init_integration_missing_some_attrs: MockConfigEntry, init_integration_missing_some_attrs: MockConfigEntry,
) -> None: ) -> None:
"""Test the Whois sensors with owner and reseller missing.""" """Test the Whois sensors with owner and reseller missing."""
@ -213,7 +213,7 @@ async def test_disabled_by_default_sensors(
async def test_no_data( async def test_no_data(
hass: HomeAssistant, hass: HomeAssistant,
mock_whois: MagicMock, mock_whois: MagicMock,
enable_all_entities: AsyncMock, entity_registry_enabled_by_default: AsyncMock,
init_integration: MockConfigEntry, init_integration: MockConfigEntry,
entity_id: str, entity_id: str,
) -> None: ) -> None: