From 09e2c47970a02be5a7e79a67b335f274f4c2c866 Mon Sep 17 00:00:00 2001 From: zlg Date: Thu, 18 Oct 2018 18:14:52 -0700 Subject: cli: Tell the user when a game lacks notes --- src/vgstash_cli.py | 9 ++++++--- tests/test_vgstash_cli.py | 9 +++++++++ 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 -- cgit v1.2.3-54-g00ecf