1 files changed, 10 insertions, 2 deletions
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'])