Improve raspberry_pi tests (#73269)

* Improve raspberry_pi tests

* Address review comments
pull/73280/head
Erik Montnemery 2022-06-09 16:12:10 +02:00 committed by GitHub
parent 542eae1cf3
commit 82afbd1d12
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 18 additions and 1 deletions

View File

@ -1,6 +1,8 @@
"""Test the Raspberry Pi integration."""
from unittest.mock import patch
import pytest
from homeassistant.components.raspberry_pi.const import DOMAIN
from homeassistant.config_entries import ConfigEntryState
from homeassistant.core import HomeAssistant
@ -8,6 +10,16 @@ from homeassistant.core import HomeAssistant
from tests.common import MockConfigEntry, MockModule, mock_integration
@pytest.fixture(autouse=True)
def mock_rpi_power():
"""Mock the rpi_power integration."""
with patch(
"homeassistant.components.rpi_power.async_setup_entry",
return_value=True,
):
yield
async def test_setup_entry(hass: HomeAssistant) -> None:
"""Test setup of a config entry."""
mock_integration(hass, MockModule("hassio"))
@ -20,14 +32,19 @@ async def test_setup_entry(hass: HomeAssistant) -> None:
title="Raspberry Pi",
)
config_entry.add_to_hass(hass)
assert not hass.config_entries.async_entries("rpi_power")
with patch(
"homeassistant.components.raspberry_pi.get_os_info",
return_value={"board": "rpi"},
) as mock_get_os_info:
) as mock_get_os_info, patch(
"homeassistant.components.rpi_power.config_flow.new_under_voltage"
):
assert await hass.config_entries.async_setup(config_entry.entry_id)
await hass.async_block_till_done()
assert len(mock_get_os_info.mock_calls) == 1
assert len(hass.config_entries.async_entries("rpi_power")) == 1
async def test_setup_entry_wrong_board(hass: HomeAssistant) -> None:
"""Test setup of a config entry with wrong board type."""