mirror of
https://github.com/jamesroberts/fastwsgi.git
synced 2026-01-06 06:09:47 -06:00
29 lines
680 B
Python
29 lines
680 B
Python
import pytest
|
|
import fastwsgi
|
|
import time
|
|
from multiprocessing import Process, set_start_method
|
|
|
|
|
|
HOST = "127.0.0.1"
|
|
PORT = 8080
|
|
|
|
|
|
class ServerProcess:
|
|
def __init__(self, application, host=HOST, port=PORT) -> None:
|
|
self.process = Process(target=fastwsgi.run, args=(application, host, port))
|
|
self.endpoint = f"http://{host}:{port}"
|
|
|
|
def __enter__(self):
|
|
self.process.start()
|
|
time.sleep(1) # Allow server to start
|
|
return self
|
|
|
|
def __exit__(self, exc_type, exc_value, exc_tb):
|
|
self.process.kill()
|
|
|
|
|
|
@pytest.fixture(autouse=True, scope="session")
|
|
def server_process():
|
|
set_start_method("fork")
|
|
return ServerProcess
|