Don't use MagicMock() as a dummy logger in pytest

This commit is contained in:
James R. Barlow
2019-05-11 12:44:17 -07:00
parent 15a988b999
commit 482cb788ed
2 changed files with 9 additions and 7 deletions
+6 -5
View File
@@ -285,7 +285,7 @@ def test_kodak_toc(resources, outpdf, spoof_tesseract_noop):
assert isinstance(p.root.Outlines.First, pikepdf.Dictionary)
def test_metadata_fixup_warning(resources, outdir):
def test_metadata_fixup_warning(resources, outdir, caplog):
from ocrmypdf._pipeline import metadata_fixup
input_files = [
@@ -296,7 +296,7 @@ def test_metadata_fixup_warning(resources, outdir):
for f in input_files:
copyfile(resources / 'graph.pdf', f)
log = MagicMock()
log = logging.getLogger()
context = MagicMock()
metadata_fixup(
input_files_groups=input_files,
@@ -304,7 +304,8 @@ def test_metadata_fixup_warning(resources, outdir):
log=log,
context=context,
)
log.warning.assert_not_called()
for record in caplog.records:
assert record.levelname != 'WARNING'
# Now add some metadata that will not be copyable
graph = pikepdf.open(outdir / 'graph.repaired.pdf')
@@ -312,7 +313,7 @@ def test_metadata_fixup_warning(resources, outdir):
meta['prism2:publicationName'] = 'OCRmyPDF Test'
graph.save(outdir / 'graph.repaired.pdf')
log = MagicMock()
log = logging.getLogger()
context = MagicMock()
metadata_fixup(
input_files_groups=input_files,
@@ -320,7 +321,7 @@ def test_metadata_fixup_warning(resources, outdir):
log=log,
context=context,
)
log.warning.assert_called_once()
assert any(record.levelname == 'WARNING' for record in caplog.records)
def test_prevent_gs_invalid_xml(resources, outdir):
+3 -2
View File
@@ -16,9 +16,10 @@
# along with OCRmyPDF. If not, see <http://www.gnu.org/licenses/>.
import argparse
import logging
from os import fspath
from pathlib import Path
from unittest.mock import MagicMock, patch
from unittest.mock import patch
import pytest
@@ -57,7 +58,7 @@ def test_no_unpaper(resources, no_outpdf):
with patch("ocrmypdf.exec.unpaper.version") as mock_unpaper_version:
mock_unpaper_version.side_effect = FileNotFoundError("unpaper")
with pytest.raises(SystemExit):
main.check_options(options, log=MagicMock())
main.check_options(options, log=logging.getLogger())
def test_old_unpaper(spoof_unpaper_oldversion, resources, no_outpdf):