def test_prepare_download__log(caplog, url, headers, from_cache, expected): caplog.set_level(logging.INFO) resp = MockResponse(b'') resp.url = url resp.headers = headers if from_cache: resp.from_cache = from_cache link = Link(url) _prepare_download(resp, link, progress_bar="on") assert len(caplog.records) == 1 record = caplog.records[0] assert record.levelname == 'INFO' assert expected in record.message
def test_prepare_download__log( caplog: pytest.LogCaptureFixture, url: str, headers: Dict[str, str], from_cache: bool, expected: str, ) -> None: caplog.set_level(logging.INFO) resp = MockResponse(b"") resp.url = url resp.headers = headers if from_cache: resp.from_cache = from_cache link = Link(url) _prepare_download(resp, link, progress_bar="on") assert len(caplog.records) == 1 record = caplog.records[0] assert record.levelname == "INFO" assert expected in record.message