1 files changed, 21 insertions, 0 deletions
diff --git a/tests/test_vgstash_cli.py b/tests/test_vgstash_cli.py
index 0e24859..e24c5a4 100644
--- a/tests/test_vgstash_cli.py
+++ b/tests/test_vgstash_cli.py
@@ -231,6 +231,27 @@ def test_notes_invalid():
     assert result.output == 'Game not found. Please try again.\n'
 
 
+def test_notes_unmodified():
+    runner = CliRunner()
+    print("--- WHAT EDITOR SEES ---")
+    result = runner.invoke(vgstash_cli.cli, ['notes', 'Vectorman 2', 'Genesis', '-e'])
+    print("--- END WHAT EDITOR SEES ---")
+    if verbose:
+        print(result.output)
+    assert result.exit_code == 0
+    assert result.output == 'Notes for Vectorman 2 on Genesis left unchanged.\n'
+
+
+def test_notes_bad_editor():
+    os.environ['EDITOR'] = "zlurp"
+    runner = CliRunner()
+    result = runner.invoke(vgstash_cli.cli, ['notes', 'Vectorman 2', 'Genesis', '-e'])
+    if verbose:
+        print(result.output)
+    assert result.exit_code == 0
+    assert result.output == "Could not run editor. Is it set correctly?\n"
+
+
 def test_notes_empty():
     runner = CliRunner()
     result = runner.invoke(vgstash_cli.cli, ['notes', 'Super Mario Bros.', 'NES'])