From dad6d54d20aac8c48324029e9f9eb54010cf5664 Mon Sep 17 00:00:00 2001
From: Adam Goldsmith <adam@adamgoldsmith.name>
Date: Fri, 6 Nov 2015 16:23:12 -0500
Subject: [PATCH] Add two test cases for editSub

basic edit and unset EDITOR
---
 test.py | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/test.py b/test.py
index df12cf1..3fb8405 100755
--- a/test.py
+++ b/test.py
@@ -71,5 +71,30 @@ class TestAdd(unittest.TestCase):
         fileExists = os.path.isfile(os.path.join(os.getenv("TODO_DIR"), out[0], out[1]))
         self.assertTrue(fileExists)
 
+class TestEdit(unittest.TestCase):
+    def test_edit(self):
+        from unittest.mock import MagicMock
+        import subprocess
+
+        subprocess.call = MagicMock()
+        setup_environment("test1")
+        os.environ["EDITOR"] = "test"
+
+        sub.editSub(1)
+        subName = sub.getSub(sub.getItem(1))
+        subprocess.call.assert_called_with(["test", os.path.join(TEST_DATA_DIR, "test1", subName[0], subName[1])])
+
+    def test_edit_EDITOR_unset(self):
+        from unittest.mock import MagicMock
+        import subprocess
+
+        subprocess.call = MagicMock()
+        setup_environment("test1")
+        os.environ["EDITOR"] = ""
+
+        subName = sub.getSub(sub.getItem(1))
+        with self.assertRaises(SystemExit):
+            sub.editSub(1)
+
 if __name__ == '__main__':
     unittest.main()