1 files changed, 17 insertions, 0 deletions
diff --git a/tests/test_vgstash_cli.py b/tests/test_vgstash_cli.py
index 30eff62..0e24859 100644
--- a/tests/test_vgstash_cli.py
+++ b/tests/test_vgstash_cli.py
@@ -199,6 +199,14 @@ def test_update():
         'The Legend  |   NES    |   D |   P\n'
     ))
 
+def test_update_invalid():
+    runner = CliRunner()
+    result = runner.invoke(vgstash_cli.cli, ['update', 'Zelda: Skyward Sword', 'NES', 'progress', 'c'])
+    if verbose:
+        print(result.output)
+    assert result.exit_code == 0
+    assert result.output == 'Game not found. Please try again.\n'
+
 
 def test_notes():
     runner = CliRunner()
@@ -214,6 +222,15 @@ def test_notes():
     ))
 
 
+def test_notes_invalid():
+    runner = CliRunner()
+    result = runner.invoke(vgstash_cli.cli, ['notes', 'Zelda: Skyward Sword', 'NES', '-e'])
+    if verbose:
+        print(result.output)
+    assert result.exit_code == 0
+    assert result.output == 'Game not found. Please try again.\n'
+
+
 def test_notes_empty():
     runner = CliRunner()
     result = runner.invoke(vgstash_cli.cli, ['notes', 'Super Mario Bros.', 'NES'])