wting_autojump/tests/autojump_utils_test.py

#!/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)