mirror of
https://github.com/neondatabase/neon.git
synced 2025-12-26 15:49:58 +00:00
54 lines
1.4 KiB
Python
54 lines
1.4 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
import pytest
|
|
from pytest_httpserver import HTTPServer
|
|
|
|
if TYPE_CHECKING:
|
|
from collections.abc import Iterator
|
|
from ssl import SSLContext
|
|
|
|
from fixtures.port_distributor import PortDistributor
|
|
|
|
ListenAddress = tuple[str, int]
|
|
|
|
# compared to the fixtures from pytest_httpserver with same names, these are
|
|
# always function scoped, so you can check and stop the server in tests.
|
|
|
|
|
|
@pytest.fixture(scope="function")
|
|
def httpserver_ssl_context() -> Iterator[SSLContext | None]:
|
|
yield None
|
|
|
|
|
|
@pytest.fixture(scope="function")
|
|
def make_httpserver(
|
|
httpserver_listen_address: ListenAddress, httpserver_ssl_context: SSLContext | None
|
|
) -> Iterator[HTTPServer]:
|
|
host, port = httpserver_listen_address
|
|
if not host:
|
|
host = HTTPServer.DEFAULT_LISTEN_HOST
|
|
if not port:
|
|
port = HTTPServer.DEFAULT_LISTEN_PORT
|
|
|
|
server = HTTPServer(host=host, port=port, ssl_context=httpserver_ssl_context)
|
|
server.start()
|
|
yield server
|
|
server.clear()
|
|
if server.is_running():
|
|
server.stop()
|
|
|
|
|
|
@pytest.fixture(scope="function")
|
|
def httpserver(make_httpserver: HTTPServer) -> Iterator[HTTPServer]:
|
|
server = make_httpserver
|
|
yield server
|
|
server.clear()
|
|
|
|
|
|
@pytest.fixture(scope="function")
|
|
def httpserver_listen_address(port_distributor: PortDistributor) -> ListenAddress:
|
|
port = port_distributor.get_port()
|
|
return ("localhost", port)
|