cc4ebc925c
Co-authored-by: Paulus Schoutsen <balloob@gmail.com> Co-authored-by: Martin Hjelmare <marhje52@gmail.com> Co-authored-by: Franck Nijhof <git@frenck.dev> Co-authored-by: Pascal Vizeli <pvizeli@syshack.ch>
34 lines
803 B
Python
34 lines
803 B
Python
"""Tests for the HTTP component."""
|
|
from aiohttp import web
|
|
|
|
# Relic from the past. Kept here so we can run negative tests.
|
|
HTTP_HEADER_HA_AUTH = "X-HA-access"
|
|
|
|
|
|
def mock_real_ip(app):
|
|
"""Inject middleware to mock real IP.
|
|
|
|
Returns a function to set the real IP.
|
|
"""
|
|
ip_to_mock = None
|
|
|
|
def set_ip_to_mock(value):
|
|
nonlocal ip_to_mock
|
|
ip_to_mock = value
|
|
|
|
@web.middleware
|
|
async def mock_real_ip(request, handler):
|
|
"""Mock Real IP middleware."""
|
|
nonlocal ip_to_mock
|
|
|
|
request = request.clone(remote=ip_to_mock)
|
|
|
|
return await handler(request)
|
|
|
|
async def real_ip_startup(app):
|
|
"""Startup of real ip."""
|
|
app.middlewares.insert(0, mock_real_ip)
|
|
|
|
app.on_startup.append(real_ip_startup)
|
|
|
|
return set_ip_to_mock
|