tests/mockserver/simple_https.py - master - githubmirror/scrapy

tests/mockserver/simple_https.py

65 строк2 KB

Adrian

Type tests.mockserver and other resource-defining files (#7865)

03 авг 2026, 21:44

03 авг 2026, 21:44a7385d6

# This is only used by tests.test_downloader_handlers_http_base.TestSimpleHttpsBase

from __future__ import annotations

from typing import TYPE_CHECKING

from twisted.web.static import Data

from .http_base import BaseMockServer, main_factory
from .http_resources import BaseResource, put_child

if TYPE_CHECKING:
    from twisted.web.server import Request


class Root(BaseResource):
    def __init__(self) -> None:
        super().__init__()
        put_child(self, b"file", Data(b"0123456789", "text/plain"))

    def getChild(self, path: bytes, request: Request) -> Root:
        return self


class SimpleMockServer(BaseMockServer):
    listen_http = False
    module_name = "tests.mockserver.simple_https"

    def __init__(
        self,
        keyfile: str,
        certfile: str,
        *,
        cipher_string: str | None = None,
        tls_min_version: str | None = None,
        tls_max_version: str | None = None,
    ) -> None:
        super().__init__()
        self.keyfile = keyfile
        self.certfile = certfile
        self.cipher_string = cipher_string
        self.tls_min_version = tls_min_version
        self.tls_max_version = tls_max_version

    def get_additional_args(self) -> list[str]:
        args = [
            "--keyfile",
            self.keyfile,
            "--certfile",
            self.certfile,
        ]
        if self.cipher_string is not None:
            args.extend(["--cipher-string", self.cipher_string])
        if self.tls_min_version is not None:
            args.extend(["--tls-min-version", self.tls_min_version])
        if self.tls_max_version is not None:
            args.extend(["--tls-max-version", self.tls_max_version])
        return args


main = main_factory(Root, listen_http=False)


if __name__ == "__main__":
    main()