89 lines
2.0 KiB
Python
89 lines
2.0 KiB
Python
import json
|
|
from pathlib import Path
|
|
|
|
import httpx
|
|
import pytest
|
|
from jinja2 import BaseLoader, Environment
|
|
|
|
from omniread import (
|
|
# core
|
|
ContentType,
|
|
# pdf
|
|
FileSystemPDFClient,
|
|
# html
|
|
HTMLScraper,
|
|
PDFScraper,
|
|
)
|
|
|
|
MOCK_HTML_DIR = Path(__file__).parent / "mocks" / "html"
|
|
MOCK_PDF_DIR = Path(__file__).parent / "mocks" / "pdf"
|
|
|
|
|
|
def render_html(template_path, data_path) -> bytes:
|
|
template_text = Path(template_path).read_text(encoding="utf-8")
|
|
data = json.loads(Path(data_path).read_text(encoding="utf-8"))
|
|
|
|
env = Environment(
|
|
loader=BaseLoader(),
|
|
autoescape=False,
|
|
)
|
|
template = env.from_string(template_text)
|
|
|
|
rendered = template.render(**data)
|
|
return rendered.encode("utf-8")
|
|
|
|
|
|
def mock_transport(request: httpx.Request) -> httpx.Response:
|
|
"""
|
|
httpx MockTransport handler.
|
|
"""
|
|
path = request.url.path
|
|
if path not in ["/simple", "/table"]:
|
|
return httpx.Response(
|
|
status_code=404,
|
|
content=b"Not Found",
|
|
request=request,
|
|
)
|
|
endpoint = path.split("/")[-1]
|
|
content = render_html(
|
|
MOCK_HTML_DIR / f"{endpoint}.html.jinja",
|
|
MOCK_HTML_DIR / f"{endpoint}.json",
|
|
)
|
|
|
|
return httpx.Response(
|
|
status_code=200,
|
|
headers={"Content-Type": ContentType.HTML.value},
|
|
content=content,
|
|
request=request,
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def http_scraper() -> HTMLScraper:
|
|
transport = httpx.MockTransport(mock_transport)
|
|
|
|
client = httpx.Client(transport=transport)
|
|
|
|
return HTMLScraper(client=client)
|
|
|
|
|
|
class MockPDFClient(FileSystemPDFClient):
|
|
"""
|
|
Test-only PDF client that routes logical identifiers
|
|
to fixture files.
|
|
"""
|
|
|
|
def fetch(self, source: str) -> bytes:
|
|
if source in ["simple"]:
|
|
source = MOCK_PDF_DIR / f"{source}.pdf"
|
|
else:
|
|
raise FileNotFoundError(f"No mock PDF route for '{source}'")
|
|
|
|
return super().fetch(source)
|
|
|
|
|
|
@pytest.fixture
|
|
def pdf_scraper() -> PDFScraper:
|
|
client = MockPDFClient()
|
|
return PDFScraper(client=client)
|