address formatting

This commit is contained in:
sriniv27 2021-02-13 19:54:53 -05:00
parent 9b2633a7c5
commit edc8cd93e3

View file

@ -5,39 +5,38 @@ from unittest import mock
from jrnl.jrnl import _export_journal from jrnl.jrnl import _export_journal
@pytest.mark.parametrize( # fmt: off
"export_format", # see: https://github.com/psf/black/issues/664
[ @pytest.mark.parametrize("export_format", [ "pretty", "short",])
"pretty", #fmt: on
"short", @mock.patch.object(argparse, "Namespace", return_value={"export": None, "filename": None})
],
)
@mock.patch.object(
argparse, "Namespace", return_value={"export": None, "filename": None}
)
def test_export_format(mock_args, export_format): def test_export_format(mock_args, export_format):
test_journal = jrnl.Journal.Journal test_journal = jrnl.Journal.Journal
mock_args.export = export_format mock_args.export = export_format
with mock.patch("builtins.print") as print_spy, mock.patch( # fmt: off
"jrnl.Journal.Journal.pprint" # see: https://github.com/psf/black/issues/664
) as mock_pprint: with mock.patch("builtins.print") as print_spy, mock.patch("jrnl.Journal.Journal.pprint") as mock_pprint:
_export_journal(mock_args, test_journal) _export_journal(mock_args, test_journal)
print_spy.call_args_list = mock_pprint print_spy.call_args_list = mock_pprint
# fmt: on
@mock.patch.object( @mock.patch.object(argparse, "Namespace", return_value={"export": None, "filename": None})
argparse, "Namespace", return_value={"export": None, "filename": None}
)
def test_export_plugin(mock_args): def test_export_plugin(mock_args):
export_format = "markdown" export_format = "markdown"
test_journal = jrnl.Journal.Journal test_journal = jrnl.Journal.Journal
mock_args.export = export_format mock_args.export = export_format
mock_args.filename = "foo.jrnl" mock_args.filename = "foo.jrnl"
with mock.patch("builtins.print") as print_spy, mock.patch(
"jrnl.plugins.get_exporter" # fmt: off
) as mock_get_exporter, mock.patch("jrnl.Journal.Journal.pprint") as mock_pprint: # see: https://github.com/psf/black/issues/664
with mock.patch("builtins.print") as print_spy, \
mock.patch("jrnl.plugins.get_exporter") as mock_get_exporter, \
mock.patch("jrnl.Journal.Journal.pprint") as mock_pprint:
_export_journal(mock_args, test_journal) _export_journal(mock_args, test_journal)
# fmt: on
print_spy.call_args_list = mock_pprint print_spy.call_args_list = mock_pprint
mock_get_exporter.assert_called_once_with(export_format) mock_get_exporter.assert_called_once_with(export_format)