mirror of
https://github.com/jrnl-org/jrnl.git
synced 2025-05-10 08:38:32 +02:00
c043f5058f
* Fixed error related to display_format in config file Now _display_search_results tries to source the export arg from the config file before dispaying search results. * Add BDD test for original bug * update unit test Co-authored-by: Micah Jerome Ellison <micah.jerome.ellison@gmail.com> Co-authored-by: Jonathan Wren <jonathan@nowandwren.com>
48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
from unittest import mock
|
|
|
|
import pytest
|
|
|
|
import random
|
|
import string
|
|
import jrnl
|
|
from jrnl.jrnl import _display_search_results
|
|
from jrnl.args import parse_args
|
|
|
|
|
|
@pytest.fixture
|
|
def random_string():
|
|
return "".join(random.choices(string.ascii_uppercase + string.digits, k=25))
|
|
|
|
|
|
@pytest.mark.parametrize("export_format", ["pretty", "short"])
|
|
@mock.patch("builtins.print")
|
|
@mock.patch("jrnl.Journal.Journal.pprint")
|
|
def test_display_search_results_pretty_short(mock_pprint, mock_print, export_format):
|
|
mock_args = parse_args(["--format", export_format])
|
|
test_journal = mock.Mock(wraps=jrnl.Journal.Journal)
|
|
|
|
_display_search_results(mock_args, test_journal)
|
|
|
|
mock_print.assert_called_once_with(mock_pprint.return_value)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"export_format", ["markdown", "json", "xml", "yaml", "fancy", "dates"]
|
|
)
|
|
@mock.patch("jrnl.plugins.get_exporter")
|
|
@mock.patch("builtins.print")
|
|
def test_display_search_results_builtin_plugins(
|
|
mock_print, mock_exporter, export_format, random_string
|
|
):
|
|
test_filename = random_string
|
|
mock_args = parse_args(["--format", export_format, "--file", test_filename])
|
|
|
|
test_journal = mock.Mock(wraps=jrnl.Journal.Journal)
|
|
mock_export = mock.Mock()
|
|
mock_exporter.return_value.export = mock_export
|
|
|
|
_display_search_results(mock_args, test_journal)
|
|
|
|
mock_exporter.assert_called_once_with(export_format)
|
|
mock_export.assert_called_once_with(test_journal, test_filename)
|
|
mock_print.assert_called_once_with(mock_export.return_value)
|