mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2024-12-22 06:00:00 +00:00
ccfd70f4c2
* Adds a basic WebSocket framework * Introduces new minimum `websockets` version of 12.0 * Deprecates `WebSocketsWrapper` Fixes https://github.com/yt-dlp/yt-dlp/issues/8439 Authored by: coletdjnz
26 lines
663 B
Python
26 lines
663 B
Python
import functools
|
|
import inspect
|
|
|
|
import pytest
|
|
|
|
from yt_dlp.networking import RequestHandler
|
|
from yt_dlp.networking.common import _REQUEST_HANDLERS
|
|
from yt_dlp.utils._utils import _YDLLogger as FakeLogger
|
|
|
|
|
|
@pytest.fixture
|
|
def handler(request):
|
|
RH_KEY = request.param
|
|
if inspect.isclass(RH_KEY) and issubclass(RH_KEY, RequestHandler):
|
|
handler = RH_KEY
|
|
elif RH_KEY in _REQUEST_HANDLERS:
|
|
handler = _REQUEST_HANDLERS[RH_KEY]
|
|
else:
|
|
pytest.skip(f'{RH_KEY} request handler is not available')
|
|
|
|
return functools.partial(handler, logger=FakeLogger)
|
|
|
|
|
|
def validate_and_send(rh, req):
|
|
rh.validate(req)
|
|
return rh.send(req)
|