Spaces:
Build error
Build error
File size: 1,431 Bytes
3932407 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
import http.server
import pytest
import socketserver
import threading
import os
HTTP_SERVER_HOST = os.environ.get("HTTP_SERVER_HOST", "localhost")
@pytest.fixture(params=[False, True], scope="module")
def on_disk_vectors(request):
return request.param
@pytest.fixture(params=[False, True], scope="module")
def on_disk_payload(request):
return request.param
@pytest.fixture
def http_server(tmpdir):
"""
Starts a HTTP server serving files from a temporary directory.
Yields a tuple (tmpdir, url).
"""
class Handler(http.server.SimpleHTTPRequestHandler):
def __init__(self, *args, **kwargs):
# Serve files from the temporary directory
super().__init__(*args, directory=str(tmpdir), **kwargs)
def log_request(self, *args, **kwargs):
# Silence logging
pass
with socketserver.TCPServer(("0.0.0.0", 0), Handler) as httpd:
httpd.allow_reuse_address = True
thread = threading.Thread(
target=httpd.serve_forever,
# Lower the shutdown poll interval to speed up tests
kwargs={"poll_interval": 0.1},
)
thread.start()
yield (tmpdir, f"http://{HTTP_SERVER_HOST}:{httpd.server_address[1]}")
httpd.shutdown()
thread.join()
@pytest.fixture(scope='module', autouse=True)
def collection_name(request):
return request.node.name.removesuffix(".py")
|