1
mirror of https://github.com/home-assistant/core synced 2024-08-28 03:36:46 +02:00

Add preselect_remember_me to /auth/providers (#106462)

This commit is contained in:
Robert Resch 2023-12-27 11:47:52 +01:00 committed by GitHub
parent 4f38d8cc5d
commit c19688e2d2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 57 additions and 126 deletions

View File

@ -91,6 +91,7 @@ from homeassistant.components.http.data_validator import RequestDataValidator
from homeassistant.components.http.view import HomeAssistantView
from homeassistant.core import HomeAssistant
from homeassistant.helpers.network import is_cloud_connection
from homeassistant.util.network import is_local
from . import indieauth
@ -185,7 +186,14 @@ class AuthProvidersView(HomeAssistantView):
}
)
return self.json(providers)
preselect_remember_me = not cloud_connection and is_local(remote_address)
return self.json(
{
"providers": providers,
"preselect_remember_me": preselect_remember_me,
}
)
def _prepare_result_json(

View File

@ -1,11 +1,9 @@
"""Support for tracking people."""
from __future__ import annotations
from http import HTTPStatus
import logging
from typing import Any
from aiohttp import web
import voluptuous as vol
from homeassistant.auth import EVENT_USER_REMOVED
@ -15,7 +13,6 @@ from homeassistant.components.device_tracker import (
DOMAIN as DEVICE_TRACKER_DOMAIN,
SourceType,
)
from homeassistant.components.http.view import HomeAssistantView
from homeassistant.const import (
ATTR_EDITABLE,
ATTR_ENTITY_ID,
@ -388,8 +385,6 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
hass, DOMAIN, SERVICE_RELOAD, async_reload_yaml
)
hass.http.register_view(ListPersonsView)
return True
@ -574,19 +569,3 @@ def _get_latest(prev: State | None, curr: State):
if prev is None or curr.last_updated > prev.last_updated:
return curr
return prev
class ListPersonsView(HomeAssistantView):
"""List all persons if request is made from a local network."""
requires_auth = False
url = "/api/person/list"
name = "api:person:list"
async def get(self, request: web.Request) -> web.Response:
"""Return a list of persons if request comes from a local IP."""
return self.json_message(
message="Not local",
status_code=HTTPStatus.BAD_REQUEST,
message_code="not_local",
)

View File

@ -6,7 +6,6 @@ from unittest.mock import patch
import pytest
from homeassistant.core import HomeAssistant
from homeassistant.setup import async_setup_component
from . import BASE_CONFIG, async_setup_auth
@ -26,22 +25,30 @@ _TRUSTED_NETWORKS_CONFIG = {
@pytest.mark.parametrize(
("provider_configs", "ip", "expected"),
("ip", "preselect_remember_me"),
[
("192.168.1.10", True),
("::ffff:192.168.0.10", True),
("1.2.3.4", False),
("2001:db8::1", False),
],
)
@pytest.mark.parametrize(
("provider_configs", "expected"),
[
(
BASE_CONFIG,
None,
[{"name": "Example", "type": "insecure_example", "id": None}],
),
(
[_TRUSTED_NETWORKS_CONFIG],
None,
[],
),
(
[_TRUSTED_NETWORKS_CONFIG],
"192.168.0.1",
[{"name": "Trusted Networks", "type": "trusted_networks", "id": None}],
[{"type": "homeassistant"}],
[
{
"name": "Home Assistant Local",
"type": "homeassistant",
"id": None,
}
],
),
],
)
@ -49,8 +56,9 @@ async def test_fetch_auth_providers(
hass: HomeAssistant,
aiohttp_client: ClientSessionGenerator,
provider_configs: list[dict[str, Any]],
ip: str | None,
expected: list[dict[str, Any]],
ip: str,
preselect_remember_me: bool,
) -> None:
"""Test fetching auth providers."""
client = await async_setup_auth(
@ -58,73 +66,37 @@ async def test_fetch_auth_providers(
)
resp = await client.get("/auth/providers")
assert resp.status == HTTPStatus.OK
assert await resp.json() == expected
async def _test_fetch_auth_providers_home_assistant(
hass: HomeAssistant,
aiohttp_client: ClientSessionGenerator,
ip: str,
) -> None:
"""Test fetching auth providers for homeassistant auth provider."""
client = await async_setup_auth(
hass, aiohttp_client, [{"type": "homeassistant"}], custom_ip=ip
)
expected = {
"name": "Home Assistant Local",
"type": "homeassistant",
"id": None,
assert await resp.json() == {
"providers": expected,
"preselect_remember_me": preselect_remember_me,
}
@pytest.mark.parametrize(
("ip", "expected"),
[
(
"192.168.0.1",
[{"name": "Trusted Networks", "type": "trusted_networks", "id": None}],
),
("::ffff:192.168.0.10", []),
("1.2.3.4", []),
("2001:db8::1", []),
],
)
async def test_fetch_auth_providers_trusted_network(
hass: HomeAssistant,
aiohttp_client: ClientSessionGenerator,
expected: list[dict[str, Any]],
ip: str,
) -> None:
"""Test fetching auth providers."""
client = await async_setup_auth(
hass, aiohttp_client, [_TRUSTED_NETWORKS_CONFIG], custom_ip=ip
)
resp = await client.get("/auth/providers")
assert resp.status == HTTPStatus.OK
assert await resp.json() == [expected]
@pytest.mark.parametrize(
"ip",
[
"192.168.0.10",
"::ffff:192.168.0.10",
"1.2.3.4",
"2001:db8::1",
],
)
async def test_fetch_auth_providers_home_assistant_person_not_loaded(
hass: HomeAssistant,
aiohttp_client: ClientSessionGenerator,
ip: str,
) -> None:
"""Test fetching auth providers for homeassistant auth provider, where person integration is not loaded."""
await _test_fetch_auth_providers_home_assistant(hass, aiohttp_client, ip)
@pytest.mark.parametrize(
("ip", "is_local"),
[
("192.168.0.10", True),
("::ffff:192.168.0.10", True),
("1.2.3.4", False),
("2001:db8::1", False),
],
)
async def test_fetch_auth_providers_home_assistant_person_loaded(
hass: HomeAssistant,
aiohttp_client: ClientSessionGenerator,
ip: str,
is_local: bool,
) -> None:
"""Test fetching auth providers for homeassistant auth provider, where person integration is loaded."""
domain = "person"
config = {domain: {"id": "1234", "name": "test person"}}
assert await async_setup_component(hass, domain, config)
await _test_fetch_auth_providers_home_assistant(
hass,
aiohttp_client,
ip,
)
assert (await resp.json())["providers"] == expected
async def test_fetch_auth_providers_onboarding(

View File

@ -1,5 +1,4 @@
"""The tests for the person component."""
from http import HTTPStatus
from typing import Any
from unittest.mock import patch
@ -30,7 +29,7 @@ from homeassistant.setup import async_setup_component
from .conftest import DEVICE_TRACKER, DEVICE_TRACKER_2
from tests.common import MockUser, mock_component, mock_restore_cache
from tests.typing import ClientSessionGenerator, WebSocketGenerator
from tests.typing import WebSocketGenerator
async def test_minimal_setup(hass: HomeAssistant) -> None:
@ -848,30 +847,3 @@ async def test_entities_in_person(hass: HomeAssistant) -> None:
"device_tracker.paulus_iphone",
"device_tracker.paulus_ipad",
]
async def test_list_persons(
hass: HomeAssistant,
hass_client_no_auth: ClientSessionGenerator,
hass_admin_user: MockUser,
) -> None:
"""Test listing persons from a not local ip address."""
user_id = hass_admin_user.id
admin = {"id": "1234", "name": "Admin", "user_id": user_id, "picture": "/bla"}
config = {
DOMAIN: [
admin,
{"id": "5678", "name": "Only a person"},
]
}
assert await async_setup_component(hass, DOMAIN, config)
await async_setup_component(hass, "api", {})
client = await hass_client_no_auth()
resp = await client.get("/api/person/list")
assert resp.status == HTTPStatus.BAD_REQUEST
result = await resp.json()
assert result == {"code": "not_local", "message": "Not local"}