From 694ba3bad61abc7ab08fb7fcaa8105896c1af349 Mon Sep 17 00:00:00 2001
From: Suhas <sugas182@gmail.com>
Date: Sat, 6 Feb 2021 20:48:07 -0500
Subject: [PATCH] unittest export_journal

---
 tests/test_display.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)
 create mode 100644 tests/test_display.py

diff --git a/tests/test_display.py b/tests/test_display.py
new file mode 100644
index 00000000..e0764484
--- /dev/null
+++ b/tests/test_display.py
@@ -0,0 +1,23 @@
+import argparse
+import jrnl
+import pytest 
+import mock
+from jrnl.jrnl import _export_journal, _display_search_results
+@pytest.mark.parametrize(
+    "export_format",
+    [
+        "pretty",
+        "short",
+        "markdown",
+        "json"
+    ]
+)
+@mock.patch.object(argparse,'Namespace',autospec=True)
+@mock.patch.object(jrnl,'Journal',autospec=True)
+def test_export_format(mock_journal, mock_args, export_format): 
+    mock_args.export = export_format
+    mock_args.filename = "foo.jrnl"
+    with mock.patch('builtins.print', wraps=print) as print_spy:
+        _export_journal(mock_args,mock_journal)
+    print_spy.assert_called_once()
+    
\ No newline at end of file