mirror of
https://github.com/wting/autojump
synced 2024-10-27 20:34:07 +00:00
162 lines
4.3 KiB
Python
162 lines
4.3 KiB
Python
#!/usr/bin/env python2
|
|
# -*- coding: utf-8 -*-
|
|
from random import randrange
|
|
from shutil import rmtree
|
|
from tempfile import gettempdir
|
|
from tempfile import mkdtemp
|
|
import os
|
|
|
|
import mock
|
|
from testify import TestCase
|
|
from testify import assert_equal
|
|
from testify import assert_false
|
|
from testify import assert_true
|
|
from testify import class_setup
|
|
from testify import class_teardown
|
|
from testify import run
|
|
from testify import setup
|
|
from testify import teardown
|
|
|
|
import autojump_utils
|
|
from autojump_utils import create_dir
|
|
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
|
|
from autojump_utils import move_file
|
|
from autojump_utils import sanitize
|
|
from autojump_utils import second
|
|
from autojump_utils import surround_quotes
|
|
from autojump_utils import take
|
|
|
|
|
|
class StringUnitTests(TestCase):
|
|
def test_decode(self):
|
|
assert_equal(decode(r'blah'), u'blah')
|
|
assert_equal(decode(r'日本語'), u'日本語')
|
|
|
|
def test_has_uppercase(self):
|
|
assert_true(has_uppercase('Foo'))
|
|
assert_true(has_uppercase('foO'))
|
|
assert_false(has_uppercase('foo'))
|
|
assert_false(has_uppercase(''))
|
|
|
|
@mock.patch.object(autojump_utils, 'in_bash', return_value=True)
|
|
def test_surround_quotes_in_bash(self, _):
|
|
assert_equal(surround_quotes('foo'), '"foo"')
|
|
|
|
@mock.patch.object(autojump_utils, 'in_bash', return_value=False)
|
|
def test_dont_surround_quotes_not_in_bash(self, _):
|
|
assert_equal(surround_quotes('foo'), 'foo')
|
|
|
|
def test_sanitize(self):
|
|
assert_equal(sanitize([]), [])
|
|
assert_equal(sanitize([r'/foo/bar/', r'/']), [u'/foo/bar', u'/'])
|
|
|
|
|
|
class IterationUnitTests(TestCase):
|
|
def test_first(self):
|
|
assert_equal(first((0, 1)), 0)
|
|
assert_equal(first(()), None)
|
|
|
|
def test_second(self):
|
|
assert_equal(second((0, 1)), 1)
|
|
assert_equal(second((0,)), None)
|
|
|
|
def test_last(self):
|
|
assert_equal(last((1, 2, 3)), 3)
|
|
assert_equal(last(()), None)
|
|
|
|
def test_take(self):
|
|
xs = [1, 2, 3]
|
|
assert_equal(list(take(1, xs)), [1])
|
|
assert_equal(list(take(2, xs)), [1, 2])
|
|
assert_equal(list(take(4, xs)), [1, 2, 3])
|
|
assert_equal(list(take(10, [])), [])
|
|
|
|
|
|
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."
|
|
|
|
|
|
class FileSystemIntegrationTests(TestCase):
|
|
@class_setup
|
|
def init(self):
|
|
self.tmp_dir = os.path.join(gettempdir(), 'autojump')
|
|
os.makedirs(self.tmp_dir)
|
|
|
|
@class_teardown
|
|
def cleanup(self):
|
|
try:
|
|
rmtree(self.tmp_dir)
|
|
except OSError:
|
|
pass
|
|
|
|
def get_random_path(self):
|
|
path = gettempdir()
|
|
|
|
while os.path.exists(path):
|
|
random_string = '%30x' % randrange(16 ** 30)
|
|
path = os.path.join(self.tmp_dir, random_string)
|
|
|
|
return path
|
|
|
|
def get_random_file(self):
|
|
path = self.get_random_path()
|
|
with open(path, 'w+') as f:
|
|
f.write('filler\n')
|
|
|
|
return path
|
|
|
|
def test_create_dir(self):
|
|
path = self.get_random_path()
|
|
create_dir(path)
|
|
assert_true(os.path.exists(path))
|
|
|
|
# should not raise OSError if directory already exists
|
|
create_dir(path)
|
|
assert_true(os.path.exists(path))
|
|
|
|
def test_move_file(self):
|
|
src = self.get_random_file()
|
|
dst = self.get_random_path()
|
|
assert_true(os.path.exists(src))
|
|
assert_false(os.path.exists(dst))
|
|
move_file(src, dst)
|
|
assert_true(os.path.exists(dst))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
run()
|