mirror of
https://github.com/jrnl-org/jrnl.git
synced 2025-05-17 11:38:32 +02:00
48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
import argparse
|
|
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)
|