import argparse
import jrnl
import pytest
import mock
from jrnl.jrnl import _export_journal


@pytest.mark.parametrize(
    "export_format",
    [
        "pretty",
        "short",
    ],
)
@mock.patch.object(
    argparse, "Namespace", return_value={"export": None, "filename": None}
)
def test_export_format(mock_args, export_format):

    test_journal = jrnl.Journal.Journal
    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"

    test_journal = jrnl.Journal.Journal
    mock_args.export = export_format
    mock_args.filename = "foo.jrnl"
    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)
    print_spy.call_args_list = mock_pprint
    mock_get_exporter.assert_called_once_with(export_format)