mirror of
https://github.com/wting/autojump
synced 2024-10-27 20:34:07 +00:00
146 lines
4.0 KiB
Python
146 lines
4.0 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
import os
|
|
import sys
|
|
|
|
import mock
|
|
import pytest
|
|
|
|
sys.path.append(os.path.join(os.getcwd(), 'bin'))
|
|
import autojump_utils # noqa
|
|
from autojump_utils import encode_local # noqa
|
|
from autojump_utils import first # noqa
|
|
from autojump_utils import get_tab_entry_info # noqa
|
|
from autojump_utils import has_uppercase # noqa
|
|
from autojump_utils import in_bash # noqa
|
|
from autojump_utils import is_python3 # noqa
|
|
from autojump_utils import last # noqa
|
|
from autojump_utils import sanitize # noqa
|
|
from autojump_utils import second # noqa
|
|
from autojump_utils import surround_quotes # noqa
|
|
from autojump_utils import take # noqa
|
|
from autojump_utils import unico # noqa
|
|
|
|
|
|
if is_python3():
|
|
os.getcwdu = os.getcwd
|
|
xrange = range
|
|
|
|
|
|
def u(string):
|
|
"""
|
|
This is a unicode() wrapper since u'string' is a Python3 compiler error.
|
|
"""
|
|
if is_python3():
|
|
return string
|
|
return unicode(string, encoding='utf-8', errors='strict')
|
|
|
|
|
|
# strings
|
|
@pytest.mark.skipif(is_python3(), reason='Unicode sucks.')
|
|
@mock.patch.object(sys, 'getfilesystemencoding', return_value='ascii')
|
|
def test_encode_local_ascii(_):
|
|
assert encode_local(u('foo')) == b'foo'
|
|
|
|
|
|
@pytest.mark.skipif(is_python3(), reason='Unicode sucks.')
|
|
@pytest.mark.xfail(reason='disabled due to pytest bug: https://bitbucket.org/hpk42/pytest/issue/534/pytest-fails-to-catch-unicodedecodeerrors') # noqa
|
|
@mock.patch.object(sys, 'getfilesystemencoding', return_value='ascii')
|
|
def test_encode_local_ascii_fails(_):
|
|
with pytest.raises(UnicodeDecodeError):
|
|
encode_local(u('日本語'))
|
|
|
|
|
|
@pytest.mark.skipif(is_python3(), reason='Unicode sucks.')
|
|
@mock.patch.object(sys, 'getfilesystemencoding', return_value=None)
|
|
def test_encode_local_empty(_):
|
|
assert encode_local(b'foo') == u('foo')
|
|
|
|
|
|
@pytest.mark.skipif(is_python3(), reason='Unicode sucks.')
|
|
@mock.patch.object(sys, 'getfilesystemencoding', return_value='utf-8')
|
|
def test_encode_local_unicode(_):
|
|
assert encode_local(b'foo') == u('foo')
|
|
assert encode_local(u('foo')) == u('foo')
|
|
|
|
|
|
def test_has_uppercase():
|
|
assert has_uppercase('Foo')
|
|
assert has_uppercase('foO')
|
|
assert not has_uppercase('foo')
|
|
assert not has_uppercase('')
|
|
|
|
|
|
@mock.patch.object(autojump_utils, 'in_bash', return_value=True)
|
|
def test_surround_quotes_in_bash(_):
|
|
assert surround_quotes('foo') == '"foo"'
|
|
|
|
|
|
@mock.patch.object(autojump_utils, 'in_bash', return_value=False)
|
|
def test_dont_surround_quotes_not_in_bash(_):
|
|
assert surround_quotes('foo') == 'foo'
|
|
|
|
|
|
def test_sanitize():
|
|
assert sanitize([]) == []
|
|
assert sanitize([r'/foo/bar/', r'/']) == [u('/foo/bar'), u('/')]
|
|
|
|
|
|
@pytest.mark.skipif(is_python3(), reason='Unicode sucks.')
|
|
def test_unico():
|
|
assert unico(str('blah')) == u('blah')
|
|
assert unico(str('日本語')) == u('日本語')
|
|
assert unico(u('でもおれは中国人だ。')) == u('でもおれは中国人だ。')
|
|
|
|
|
|
# iteration
|
|
def test_first():
|
|
assert first(xrange(5)) == 0
|
|
assert first([]) is None
|
|
|
|
|
|
def test_second():
|
|
assert second(xrange(5)) == 1
|
|
assert second([]) is None
|
|
|
|
|
|
def test_last():
|
|
assert last(xrange(4)) == 3
|
|
assert last([]) is None
|
|
|
|
|
|
def test_take():
|
|
assert list(take(1, xrange(3))) == [0]
|
|
assert list(take(2, xrange(3))) == [0, 1]
|
|
assert list(take(4, xrange(3))) == [0, 1, 2]
|
|
assert list(take(10, [])) == []
|
|
|
|
|
|
# environment variables
|
|
def test_in_bash():
|
|
for path in ['/bin/bash', '/usr/bin/bash']:
|
|
os.environ['SHELL'] = path
|
|
assert in_bash()
|
|
|
|
for path in ['/bin/zsh', '/usr/bin/zsh']:
|
|
os.environ['SHELL'] = '/usr/bin/zsh'
|
|
assert not in_bash()
|
|
|
|
|
|
# helper functions
|
|
def test_get_needle():
|
|
assert get_tab_entry_info('foo__', '__') == ('foo', None, None)
|
|
|
|
|
|
def test_get_index():
|
|
assert get_tab_entry_info('foo__2', '__') == ('foo', 2, None)
|
|
|
|
|
|
def test_get_path():
|
|
assert get_tab_entry_info('foo__3__/foo/bar', '__') \
|
|
== ('foo', 3, '/foo/bar')
|
|
|
|
|
|
def test_get_none():
|
|
assert get_tab_entry_info('gibberish content', '__') == (None, None, None)
|