|
|
|
@@ -1,7 +1,6 @@
|
|
|
|
|
"""The tests for the Home Assistant HTTP component."""
|
|
|
|
|
from ipaddress import ip_network
|
|
|
|
|
import logging
|
|
|
|
|
import unittest
|
|
|
|
|
|
|
|
|
|
import pytest
|
|
|
|
|
|
|
|
|
@@ -12,6 +11,32 @@ from homeassistant.util.ssl import server_context_intermediate, server_context_m
|
|
|
|
|
from tests.async_mock import Mock, patch
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@pytest.fixture
|
|
|
|
|
def mock_stack():
|
|
|
|
|
"""Mock extract stack."""
|
|
|
|
|
with patch(
|
|
|
|
|
"homeassistant.components.http.extract_stack",
|
|
|
|
|
return_value=[
|
|
|
|
|
Mock(
|
|
|
|
|
filename="/home/paulus/core/homeassistant/core.py",
|
|
|
|
|
lineno="23",
|
|
|
|
|
line="do_something()",
|
|
|
|
|
),
|
|
|
|
|
Mock(
|
|
|
|
|
filename="/home/paulus/core/homeassistant/components/hue/light.py",
|
|
|
|
|
lineno="23",
|
|
|
|
|
line="self.light.is_on",
|
|
|
|
|
),
|
|
|
|
|
Mock(
|
|
|
|
|
filename="/home/paulus/core/homeassistant/components/http/__init__.py",
|
|
|
|
|
lineno="157",
|
|
|
|
|
line="base_url",
|
|
|
|
|
),
|
|
|
|
|
],
|
|
|
|
|
):
|
|
|
|
|
yield
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class TestView(http.HomeAssistantView):
|
|
|
|
|
"""Test the HTTP views."""
|
|
|
|
|
|
|
|
|
@@ -36,110 +61,73 @@ async def test_registering_view_while_running(
|
|
|
|
|
hass.http.register_view(TestView)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class TestApiConfig(unittest.TestCase):
|
|
|
|
|
"""Test API configuration methods."""
|
|
|
|
|
|
|
|
|
|
def test_api_base_url_with_domain(hass):
|
|
|
|
|
"""Test setting API URL with domain."""
|
|
|
|
|
api_config = http.ApiConfig("127.0.0.1", "example.com")
|
|
|
|
|
assert api_config.base_url == "http://example.com:8123"
|
|
|
|
|
|
|
|
|
|
def test_api_base_url_with_ip(hass):
|
|
|
|
|
"""Test setting API URL with IP."""
|
|
|
|
|
api_config = http.ApiConfig("127.0.0.1", "1.1.1.1")
|
|
|
|
|
assert api_config.base_url == "http://1.1.1.1:8123"
|
|
|
|
|
|
|
|
|
|
def test_api_base_url_with_ip_and_port(hass):
|
|
|
|
|
"""Test setting API URL with IP and port."""
|
|
|
|
|
api_config = http.ApiConfig("127.0.0.1", "1.1.1.1", 8124)
|
|
|
|
|
assert api_config.base_url == "http://1.1.1.1:8124"
|
|
|
|
|
|
|
|
|
|
def test_api_base_url_with_protocol(hass):
|
|
|
|
|
"""Test setting API URL with protocol."""
|
|
|
|
|
api_config = http.ApiConfig("127.0.0.1", "https://example.com")
|
|
|
|
|
assert api_config.base_url == "https://example.com:8123"
|
|
|
|
|
|
|
|
|
|
def test_api_base_url_with_protocol_and_port(hass):
|
|
|
|
|
"""Test setting API URL with protocol and port."""
|
|
|
|
|
api_config = http.ApiConfig("127.0.0.1", "https://example.com", 433)
|
|
|
|
|
assert api_config.base_url == "https://example.com:433"
|
|
|
|
|
|
|
|
|
|
def test_api_base_url_with_ssl_enable(hass):
|
|
|
|
|
"""Test setting API URL with use_ssl enabled."""
|
|
|
|
|
api_config = http.ApiConfig("127.0.0.1", "example.com", use_ssl=True)
|
|
|
|
|
assert api_config.base_url == "https://example.com:8123"
|
|
|
|
|
|
|
|
|
|
def test_api_base_url_with_ssl_enable_and_port(hass):
|
|
|
|
|
"""Test setting API URL with use_ssl enabled and port."""
|
|
|
|
|
api_config = http.ApiConfig("127.0.0.1", "1.1.1.1", use_ssl=True, port=8888)
|
|
|
|
|
assert api_config.base_url == "https://1.1.1.1:8888"
|
|
|
|
|
|
|
|
|
|
def test_api_base_url_with_protocol_and_ssl_enable(hass):
|
|
|
|
|
"""Test setting API URL with specific protocol and use_ssl enabled."""
|
|
|
|
|
api_config = http.ApiConfig("127.0.0.1", "http://example.com", use_ssl=True)
|
|
|
|
|
assert api_config.base_url == "http://example.com:8123"
|
|
|
|
|
|
|
|
|
|
def test_api_base_url_removes_trailing_slash(hass):
|
|
|
|
|
"""Test a trialing slash is removed when setting the API URL."""
|
|
|
|
|
api_config = http.ApiConfig("127.0.0.1", "http://example.com/")
|
|
|
|
|
assert api_config.base_url == "http://example.com:8123"
|
|
|
|
|
|
|
|
|
|
def test_api_local_ip(hass):
|
|
|
|
|
"""Test a trialing slash is removed when setting the API URL."""
|
|
|
|
|
api_config = http.ApiConfig("127.0.0.1", "http://example.com/")
|
|
|
|
|
assert api_config.local_ip == "127.0.0.1"
|
|
|
|
|
def test_api_base_url_with_domain(mock_stack):
|
|
|
|
|
"""Test setting API URL with domain."""
|
|
|
|
|
api_config = http.ApiConfig("127.0.0.1", "example.com")
|
|
|
|
|
assert api_config.base_url == "http://example.com:8123"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
async def test_api_base_url_with_domain(hass):
|
|
|
|
|
"""Test setting API URL."""
|
|
|
|
|
result = await async_setup_component(
|
|
|
|
|
hass, "http", {"http": {"base_url": "example.com"}}
|
|
|
|
|
)
|
|
|
|
|
assert result
|
|
|
|
|
assert hass.config.api.base_url == "http://example.com"
|
|
|
|
|
def test_api_base_url_with_ip(mock_stack):
|
|
|
|
|
"""Test setting API URL with IP."""
|
|
|
|
|
api_config = http.ApiConfig("127.0.0.1", "1.1.1.1")
|
|
|
|
|
assert api_config.base_url == "http://1.1.1.1:8123"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
async def test_api_base_url_with_ip(hass):
|
|
|
|
|
"""Test setting api url."""
|
|
|
|
|
result = await async_setup_component(
|
|
|
|
|
hass, "http", {"http": {"server_host": "1.1.1.1"}}
|
|
|
|
|
)
|
|
|
|
|
assert result
|
|
|
|
|
assert hass.config.api.base_url == "http://1.1.1.1:8123"
|
|
|
|
|
def test_api_base_url_with_ip_and_port(mock_stack):
|
|
|
|
|
"""Test setting API URL with IP and port."""
|
|
|
|
|
api_config = http.ApiConfig("127.0.0.1", "1.1.1.1", 8124)
|
|
|
|
|
assert api_config.base_url == "http://1.1.1.1:8124"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
async def test_api_base_url_with_ip_port(hass):
|
|
|
|
|
"""Test setting api url."""
|
|
|
|
|
result = await async_setup_component(
|
|
|
|
|
hass, "http", {"http": {"base_url": "1.1.1.1:8124"}}
|
|
|
|
|
)
|
|
|
|
|
assert result
|
|
|
|
|
assert hass.config.api.base_url == "http://1.1.1.1:8124"
|
|
|
|
|
def test_api_base_url_with_protocol(mock_stack):
|
|
|
|
|
"""Test setting API URL with protocol."""
|
|
|
|
|
api_config = http.ApiConfig("127.0.0.1", "https://example.com")
|
|
|
|
|
assert api_config.base_url == "https://example.com:8123"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
async def test_api_no_base_url(hass):
|
|
|
|
|
def test_api_base_url_with_protocol_and_port(mock_stack):
|
|
|
|
|
"""Test setting API URL with protocol and port."""
|
|
|
|
|
api_config = http.ApiConfig("127.0.0.1", "https://example.com", 433)
|
|
|
|
|
assert api_config.base_url == "https://example.com:433"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_api_base_url_with_ssl_enable(mock_stack):
|
|
|
|
|
"""Test setting API URL with use_ssl enabled."""
|
|
|
|
|
api_config = http.ApiConfig("127.0.0.1", "example.com", use_ssl=True)
|
|
|
|
|
assert api_config.base_url == "https://example.com:8123"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_api_base_url_with_ssl_enable_and_port(mock_stack):
|
|
|
|
|
"""Test setting API URL with use_ssl enabled and port."""
|
|
|
|
|
api_config = http.ApiConfig("127.0.0.1", "1.1.1.1", use_ssl=True, port=8888)
|
|
|
|
|
assert api_config.base_url == "https://1.1.1.1:8888"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_api_base_url_with_protocol_and_ssl_enable(mock_stack):
|
|
|
|
|
"""Test setting API URL with specific protocol and use_ssl enabled."""
|
|
|
|
|
api_config = http.ApiConfig("127.0.0.1", "http://example.com", use_ssl=True)
|
|
|
|
|
assert api_config.base_url == "http://example.com:8123"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_api_base_url_removes_trailing_slash(mock_stack):
|
|
|
|
|
"""Test a trialing slash is removed when setting the API URL."""
|
|
|
|
|
api_config = http.ApiConfig("127.0.0.1", "http://example.com/")
|
|
|
|
|
assert api_config.base_url == "http://example.com:8123"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_api_local_ip(mock_stack):
|
|
|
|
|
"""Test a trialing slash is removed when setting the API URL."""
|
|
|
|
|
api_config = http.ApiConfig("127.0.0.1", "http://example.com/")
|
|
|
|
|
assert api_config.local_ip == "127.0.0.1"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
async def test_api_no_base_url(hass, mock_stack):
|
|
|
|
|
"""Test setting api url."""
|
|
|
|
|
result = await async_setup_component(hass, "http", {"http": {}})
|
|
|
|
|
assert result
|
|
|
|
|
assert hass.config.api.base_url == "http://127.0.0.1:8123"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
async def test_api_local_ip(hass):
|
|
|
|
|
"""Test setting api url."""
|
|
|
|
|
result = await async_setup_component(hass, "http", {"http": {}})
|
|
|
|
|
assert result
|
|
|
|
|
assert hass.config.api.local_ip == "127.0.0.1"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
async def test_api_base_url_removes_trailing_slash(hass):
|
|
|
|
|
"""Test setting api url."""
|
|
|
|
|
result = await async_setup_component(
|
|
|
|
|
hass, "http", {"http": {"base_url": "https://example.com/"}}
|
|
|
|
|
)
|
|
|
|
|
assert result
|
|
|
|
|
assert hass.config.api.base_url == "https://example.com"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
async def test_not_log_password(hass, aiohttp_client, caplog, legacy_auth):
|
|
|
|
|
"""Test access with password doesn't get logged."""
|
|
|
|
|
assert await async_setup_component(hass, "api", {"http": {}})
|
|
|
|
|