mirror of
https://github.com/wting/autojump
synced 2024-10-27 20:34:07 +00:00
133 lines
3.3 KiB
Python
133 lines
3.3 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
import os
|
|
import sys
|
|
|
|
import mock
|
|
import pytest
|
|
|
|
# Python 3 compatibility
|
|
if sys.version_info[0] == 3:
|
|
os.getcwdu = os.getcwd
|
|
sys.path.append(os.path.join(os.getcwd(), 'bin'))
|
|
|
|
import autojump_utils
|
|
from autojump_utils import encode_local
|
|
from autojump_utils import first
|
|
from autojump_utils import get_tab_entry_info
|
|
from autojump_utils import has_uppercase
|
|
from autojump_utils import in_bash
|
|
from autojump_utils import last
|
|
from autojump_utils import sanitize
|
|
from autojump_utils import second
|
|
from autojump_utils import surround_quotes
|
|
from autojump_utils import take
|
|
from autojump_utils import unico
|
|
|
|
|
|
# strings
|
|
@mock.patch.object(sys, 'getfilesystemencoding', return_value='ascii')
|
|
def test_encode_local_ascii(_):
|
|
assert encode_local(u'foo') == b'foo'
|
|
|
|
|
|
@pytest.mark.xfail(reason='issue #246')
|
|
def test_encode_local_ascii_fails():
|
|
with pytest.raises(UnicodeDecodeError):
|
|
with mock.patch.object(
|
|
sys,
|
|
'getfilesystemencoding',
|
|
return_value='ascii'):
|
|
encode_local(u'日本語')
|
|
|
|
|
|
@mock.patch.object(sys, 'getfilesystemencoding', return_value=None)
|
|
def test_encode_local_empty(_):
|
|
assert encode_local(b'foo') == u'foo'
|
|
|
|
|
|
@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'/']
|
|
|
|
|
|
def test_unico():
|
|
assert unico(b'blah') == u'blah'
|
|
assert unico(b'日本語') == 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)
|