test/conftest.py

   1 import inspect
   2
   3 import pytest
   4
   5 from yt_dlp.networking import RequestHandler
   6 from yt_dlp.networking.common import _REQUEST_HANDLERS
   7 from yt_dlp.utils._utils import _YDLLogger as FakeLogger
   8
   9
  10 @pytest.fixture
  11 def handler(request):
  12     RH_KEY = getattr(request, 'param', None)
  13     if not RH_KEY:
  14         return
  15     if inspect.isclass(RH_KEY) and issubclass(RH_KEY, RequestHandler):
  16         handler = RH_KEY
  17     elif RH_KEY in _REQUEST_HANDLERS:
  18         handler = _REQUEST_HANDLERS[RH_KEY]
  19     else:
  20         pytest.skip(f'{RH_KEY} request handler is not available')
  21
  22     class HandlerWrapper(handler):
  23         RH_KEY = handler.RH_KEY
  24
  25         def __init__(self, **kwargs):
  26             super().__init__(logger=FakeLogger, **kwargs)
  27
  28     return HandlerWrapper
  29
  30
  31 @pytest.fixture(autouse=True)
  32 def skip_handler(request, handler):
  33     """usage: pytest.mark.skip_handler('my_handler', 'reason')"""
  34     for marker in request.node.iter_markers('skip_handler'):
  35         if marker.args[0] == handler.RH_KEY:
  36             pytest.skip(marker.args[1] if len(marker.args) > 1 else '')
  37
  38
  39 @pytest.fixture(autouse=True)
  40 def skip_handler_if(request, handler):
  41     """usage: pytest.mark.skip_handler_if('my_handler', lambda request: True, 'reason')"""
  42     for marker in request.node.iter_markers('skip_handler_if'):
  43         if marker.args[0] == handler.RH_KEY and marker.args[1](request):
  44             pytest.skip(marker.args[2] if len(marker.args) > 2 else '')
  45
  46
  47 @pytest.fixture(autouse=True)
  48 def skip_handlers_if(request, handler):
  49     """usage: pytest.mark.skip_handlers_if(lambda request, handler: True, 'reason')"""
  50     for marker in request.node.iter_markers('skip_handlers_if'):
  51         if handler and marker.args[0](request, handler):
  52             pytest.skip(marker.args[1] if len(marker.args) > 1 else '')
  53
  54
  55 def pytest_configure(config):
  56     config.addinivalue_line(
  57         'markers', 'skip_handler(handler): skip test for the given handler',
  58     )
  59     config.addinivalue_line(
  60         'markers', 'skip_handler_if(handler): skip test for the given handler if condition is true',
  61     )
  62     config.addinivalue_line(
  63         'markers', 'skip_handlers_if(handler): skip test for handlers when the condition is true',
  64     )