clean up conftest

This commit is contained in:
coletdjnz 2025-05-18 11:49:45 +12:00
parent 8cb9ad3bfd
commit fdf701e8f6
No known key found for this signature in database
GPG Key ID: 91984263BB39894A

View File

@ -1,3 +1,5 @@
import collections
import pytest import pytest
from yt_dlp import YoutubeDL from yt_dlp import YoutubeDL
@ -14,27 +16,22 @@ class MockLogger(IEContentProviderLogger):
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
self.messages = {} self.messages = collections.defaultdict(list)
def trace(self, message: str): def trace(self, message: str):
self.messages.setdefault('trace', []).append(message) self.messages['trace'].append(message)
pass
def debug(self, message: str): def debug(self, message: str):
self.messages.setdefault('debug', []).append(message) self.messages['debug'].append(message)
pass
def info(self, message: str): def info(self, message: str):
self.messages.setdefault('info', []).append(message) self.messages['info'].append(message)
pass
def warning(self, message: str, *, once=False): def warning(self, message: str, *, once=False):
self.messages.setdefault('warning', []).append(message) self.messages['warning'].append(message)
pass
def error(self, message: str): def error(self, message: str):
self.messages.setdefault('error', []).append(message) self.messages['error'].append(message)
pass
@pytest.fixture @pytest.fixture