|
|
@ -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() |