Rewrite panel_iframe unittest tests to pytest style tests (#41368)

This commit is contained in:
Alexander Pitkin 2020-10-07 01:40:28 +03:00 committed by GitHub
parent ca59c4a645
commit 5b0282e56d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -1,41 +1,28 @@
"""The tests for the panel_iframe component."""
import unittest
import pytest
from homeassistant import setup
from homeassistant.components import frontend
from tests.async_mock import patch
from tests.common import get_test_home_assistant
from homeassistant.setup import async_setup_component
class TestPanelIframe(unittest.TestCase):
"""Test the panel_iframe component."""
def setUp(self):
"""Set up things to be run when tests are started."""
self.hass = get_test_home_assistant()
self.addCleanup(self.hass.stop)
def test_wrong_config(self):
"""Test setup with wrong configuration."""
to_try = [
@pytest.mark.parametrize(
"config_to_try",
(
{"invalid space": {"url": "https://home-assistant.io"}},
{"router": {"url": "not-a-url"}},
]
for conf in to_try:
with patch(
"homeassistant.components.http.start_http_server_and_save_config"
):
assert not setup.setup_component(
self.hass, "panel_iframe", {"panel_iframe": conf}
),
)
async def test_wrong_config(hass, config_to_try):
"""Test setup with wrong configuration."""
assert not await async_setup_component(
hass, "panel_iframe", {"panel_iframe": config_to_try}
)
def test_correct_config(self):
async def test_correct_config(hass):
"""Test correct config."""
with patch("homeassistant.components.http.start_http_server_and_save_config"):
assert setup.setup_component(
self.hass,
assert await async_setup_component(
hass,
"panel_iframe",
{
"panel_iframe": {
@ -61,7 +48,7 @@ class TestPanelIframe(unittest.TestCase):
},
)
panels = self.hass.data[frontend.DATA_PANELS]
panels = hass.data[frontend.DATA_PANELS]
assert panels.get("router").to_response() == {
"component_name": "iframe",