diff options
| -rw-r--r-- | src/vgstash_cli.py | 9 | ||||
| -rw-r--r-- | 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 |