cli: Tell the user when a game lacks notes

2 files changed, 15 insertions, 3 deletions

diff --git a/src/vgstash_cli.py b/src/vgstash_cli.py
index db5853b..c8e8d24 100644
--- a/src/vgstash_cli.py
+++ b/src/vgstash_cli.py
@@ -196,9 +196,12 @@ def notes(title, system, edit):
         else:
             click.echo("Couldn't find an editor for notes. Check the EDITOR environment variable and try again.")
     else:
-        click.echo("Notes for {} on {}:".format(target_game.title, target_game.system))
-        click.echo()
-        click.echo(target_game.notes)
+        if len(target_game.notes) > 0:
+            click.echo("Notes for {} on {}:".format(target_game.title, target_game.system))
+            click.echo()
+            click.echo(target_game.notes)
+        else:
+            click.echo("No notes for {} on {}.".format(target_game.title, target_game.system))
 
 
 @cli.command("import")
diff --git a/tests/test_vgstash_cli.py b/tests/test_vgstash_cli.py
index f34f2d1..831fe80 100644
--- a/tests/test_vgstash_cli.py
+++ b/tests/test_vgstash_cli.py
@@ -206,6 +206,15 @@ def test_notes():
     ))
 
 
+def test_notes_empty():
+    runner = CliRunner()
+    result = runner.invoke(vgstash_cli.cli, ['notes', 'Super Mario Bros.', 'NES'])
+    if verbose:
+        print(result.output)
+    assert result.exit_code == 0
+    assert result.output == "No notes for Super Mario Bros. on NES.\n"
+
+
 def test_notes_edit():
     if not interactive:
         return