cli: show msg if game to be deleted is not in DB

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'])