44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
import asyncio
|
|
|
|
from models.proxy import ProxyDocument
|
|
from models.service import ServiceEndpointDocument
|
|
from services.mitmproxy.health import check_proxy
|
|
|
|
|
|
def test_check_proxy_reports_upstream_and_proxy_healthy():
|
|
async def run():
|
|
upstream = await asyncio.start_server(http_handler, "127.0.0.1", 0)
|
|
proxy = await asyncio.start_server(http_handler, "127.0.0.1", 0)
|
|
try:
|
|
upstream_port = upstream.sockets[0].getsockname()[1]
|
|
proxy_port = proxy.sockets[0].getsockname()[1]
|
|
document = ProxyDocument(
|
|
proxy_port=proxy_port,
|
|
name="health",
|
|
code="",
|
|
service=ServiceEndpointDocument(port=upstream_port),
|
|
)
|
|
|
|
health = await check_proxy(document)
|
|
finally:
|
|
upstream.close()
|
|
proxy.close()
|
|
await upstream.wait_closed()
|
|
await proxy.wait_closed()
|
|
|
|
assert health["overall"] == "healthy"
|
|
assert [check["target"] for check in health["checks"]] == [
|
|
"upstream",
|
|
"proxy",
|
|
]
|
|
|
|
asyncio.run(run())
|
|
|
|
|
|
async def http_handler(reader, writer):
|
|
await reader.readline()
|
|
writer.write(b"HTTP/1.1 200 OK\r\nContent-Length: 0\r\n\r\n")
|
|
await writer.drain()
|
|
writer.close()
|
|
await writer.wait_closed()
|