make format

This commit is contained in:
Suhas 2021-02-06 21:32:40 -05:00
parent 9d60911a14
commit e0f7736a32

View file

@ -1,40 +1,43 @@
import argparse import argparse
import jrnl import jrnl
import pytest import pytest
import mock import mock
from jrnl.jrnl import _export_journal from jrnl.jrnl import _export_journal
@pytest.mark.parametrize( @pytest.mark.parametrize(
"export_format", "export_format",
[ [
"pretty", "pretty",
"short", "short",
] ],
) )
@mock.patch.object(argparse,'Namespace',return_value={"export":None,"filename":None}) @mock.patch.object(
def test_export_format(mock_args,export_format): argparse, "Namespace", return_value={"export": None, "filename": None}
)
test_journal = jrnl.Journal.Journal def test_export_format(mock_args, export_format):
mock_args.export = export_format
mock_args.filename = "foo.jrnl"
with mock.patch('builtins.print') as print_spy, \
mock.patch('jrnl.Journal.Journal.pprint') as mock_pprint:
_export_journal(mock_args,test_journal)
print_spy.call_args_list = mock_pprint
@mock.patch.object(argparse,'Namespace',return_value={"export":None,"filename":None}) test_journal = jrnl.Journal.Journal
def test_export_plugin(mock_args): mock_args.export = export_format
with mock.patch("builtins.print") as print_spy, mock.patch(
"jrnl.Journal.Journal.pprint"
) as mock_pprint:
_export_journal(mock_args, test_journal)
print_spy.call_args_list = mock_pprint
@mock.patch.object(
argparse, "Namespace", return_value={"export": None, "filename": None}
)
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, \ with mock.patch("builtins.print") as print_spy, mock.patch(
mock.patch('jrnl.plugins.get_exporter') as mock_get_exporter,\ "jrnl.plugins.get_exporter"
mock.patch('jrnl.Journal.Journal.pprint') as mock_pprint: ) 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)
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)