From 7b5d3fdaf9527b24eae97faa66ceb9f8ed339512 Mon Sep 17 00:00:00 2001
From: Jonathan Wren <jonathan@nowandwren.com>
Date: Sat, 6 May 2023 16:19:21 -0700
Subject: [PATCH] update tests

---
 jrnl/main.py      | 10 ++++++++--
 pyproject.toml    |  2 +-
 tests/test_run.py | 15 +++++++++------
 3 files changed, 18 insertions(+), 9 deletions(-)

diff --git a/jrnl/main.py b/jrnl/main.py
index b68a7c9d..d39dae77 100644
--- a/jrnl/main.py
+++ b/jrnl/main.py
@@ -1,7 +1,13 @@
 # Copyright © 2012-2023 jrnl contributors
 # License: https://www.gnu.org/licenses/gpl-3.0.html
 from ruamel import yaml
+import sys
 
-def run():
-    my_dict = dict(a='a' * 80, b=dict(c=1, d=2))
+def run(num = None):
+    if num is None:
+        num = sys.argv[1]
+
+    my_str = "a" * int(num)
+    my_dict = dict(a=my_str, b=dict(c=1, d=2))
     return yaml.dump(my_dict)
+
diff --git a/pyproject.toml b/pyproject.toml
index 97dd840f..fcd0b54f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -33,6 +33,6 @@ isolated_build = True
 deps =
   pytest >= 6.2
 
-commands = pytest -vvv {posargs}
+commands = pytest -v {posargs}
 passenv = HOME
 """
diff --git a/tests/test_run.py b/tests/test_run.py
index 218ca1a8..d7718900 100644
--- a/tests/test_run.py
+++ b/tests/test_run.py
@@ -4,9 +4,12 @@
 from jrnl.main import run
 
 
-def test_whatever():
-    expected_output = """
-        a: aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa
-        b: {c: 1, d: 2}
-    """.strip()
-    assert run() == expected_output
+def test_passes():
+    num = 80
+    assert run(num) == "a: " + ("a" * num) +  "\nb: {c: 1, d: 2}\n"
+
+
+def test_fails():
+    num = 81
+    assert run(num) == "a: " + ("a" * num) +  "\nb: {c: 1, d: 2}\n"
+