add env vars integration tests

bin/autojump_utils.py

@@ -90,7 +90,7 @@ def get_pwd():
         return os.getcwdu()
     except OSError:
         print("Current directory no longer exists.", file=sys.stderr)
-        sys.exit(1)
+        raise
 
 
 def has_uppercase(string):

tests/autojump_utils_test.py

@@ -1,5 +1,7 @@
 #!/usr/bin/env python2
 # -*- coding: utf-8 -*-
+from shutil import rmtree
+from tempfile import mkdtemp
 import os
 
 import mock
@@ -8,10 +10,13 @@ from testify import assert_equal
 from testify import assert_false
 from testify import assert_true
 from testify import run
+from testify import setup
+from testify import teardown
 
 import autojump_utils
 from autojump_utils import decode
 from autojump_utils import first
+from autojump_utils import get_pwd
 from autojump_utils import has_uppercase
 from autojump_utils import in_bash
 from autojump_utils import last
@@ -21,7 +26,7 @@ from autojump_utils import surround_quotes
 from autojump_utils import take
 
 
-class StringTestCase(TestCase):
+class StringUnitTests(TestCase):
     def test_decode(self):
         assert_equal(decode(r'blah'), u'blah')
         assert_equal(decode(r'日本語'), u'日本語')
@@ -45,7 +50,7 @@ class StringTestCase(TestCase):
         assert_equal(sanitize([r'/foo/bar/', r'/']), [u'/foo/bar', u'/'])
 
 
-class IterationTestCase(TestCase):
+class IterationUnitTests(TestCase):
     def test_first(self):
         assert_equal(first((0, 1)), 0)
         assert_equal(first(()), None)
@@ -66,13 +71,37 @@ class IterationTestCase(TestCase):
         assert_equal(list(take(10, [])), [])
 
 
-class EnvironmentalVariableTestCase(TestCase):
+class EnvironmentalVariableIntegrationTests(TestCase):
+    @setup
+    def create_tmp_dir(self):
+        self.tmp_dir = mkdtemp()
+
+    @teardown
+    def delete_tmp_dir(self):
+        try:
+            rmtree(self.tmp_dir)
+        except OSError:
+            pass
+
     def test_in_bash(self):
         os.environ['SHELL'] = '/bin/bash'
         assert_true(in_bash())
         os.environ['SHELL'] = '/usr/bin/zsh'
         assert_false(in_bash())
 
+    def test_good_get_pwd(self):
+        os.chdir(self.tmp_dir)
+        assert_equal(get_pwd(), self.tmp_dir)
+
+    def test_bad_get_pwd(self):
+        os.chdir(self.tmp_dir)
+        rmtree(self.tmp_dir)
+        try:
+            get_pwd()
+        except OSError:
+            return
+
+        assert False, "test_bad_get_pwd() failed."
 
 if __name__ == "__main__":
     run()