[yt-dlp.git] / test / conftest.py

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)
Commit	Line	Data
fcd6a76a	1	import functools
	2	import inspect
	3
	4	import pytest
	5
	6	from yt_dlp.networking import RequestHandler
	7	from yt_dlp.networking.common import _REQUEST_HANDLERS
	8	from yt_dlp.utils._utils import _YDLLogger as FakeLogger
	9
	10
	11	@pytest.fixture
	12	def handler(request):
	13	RH_KEY = request.param
	14	if inspect.isclass(RH_KEY) and issubclass(RH_KEY, RequestHandler):
	15	handler = RH_KEY
	16	elif RH_KEY in _REQUEST_HANDLERS:
	17	handler = _REQUEST_HANDLERS[RH_KEY]
	18	else:
	19	pytest.skip(f'{RH_KEY} request handler is not available')
	20
	21	return functools.partial(handler, logger=FakeLogger)
ccfd70f4	22
	23
	24	def validate_and_send(rh, req):
	25	rh.validate(req)
	26	return rh.send(req)