diff options
| -rw-r--r-- | src/vgstash_cli.py | 2 | ||||
| -rw-r--r-- | tests/test_vgstash_cli.py | 12 |
2 files changed, 12 insertions, 2 deletions
diff --git a/src/vgstash_cli.py b/src/vgstash_cli.py index c8e8d24..7996086 100644 --- a/src/vgstash_cli.py +++ b/src/vgstash_cli.py @@ -145,6 +145,8 @@ def delete_game(title, system): target_game = vgstash.Game(title, system) if db.delete_game(target_game): click.echo("Removed {} for {} from your collection.".format(title, system)) + else: + click.echo("That game does not exist in your collection. Please try again.") @cli.command('update') diff --git a/tests/test_vgstash_cli.py b/tests/test_vgstash_cli.py index 831fe80..c3b60d6 100644 --- a/tests/test_vgstash_cli.py +++ b/tests/test_vgstash_cli.py @@ -6,8 +6,6 @@ import vgstash_cli from click.testing import CliRunner -# TODO: Setup a testing directory for test data and integrate with pytest - verbose = True interactive = False @@ -99,6 +97,7 @@ def test_list_filter(): 'The Legend of Zelda|NES|2|2|\n', )) + def test_list_filter_invalid(): runner = CliRunner() result = runner.invoke(vgstash_cli.cli, ['list', '-r', 'derp']) @@ -170,6 +169,15 @@ def test_delete(): assert result.output == "Removed Vectorman for Genesis from your collection.\n" +def test_delete_invalid(): + runner = CliRunner() + result = runner.invoke(vgstash_cli.cli, ['delete', 'Vectorman 3', 'Genesis']) + if verbose: + print(result.output) + assert result.exit_code == 0 + assert result.output == "That game does not exist in your collection. Please try again.\n" + + def test_update(): runner = CliRunner() result = runner.invoke(vgstash_cli.cli, ['update', 'Super Mario Bros.', 'NES', 'progress', 'c']) |