You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
wting_autojump/tests/autojump_utils_test.py

72 lines
2.0 KiB

#!/usr/bin/env python2
# -*- coding: utf-8 -*-
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 run
import autojump_utils
from autojump_utils import decode
from autojump_utils import first
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
class StringTestCase(TestCase):
def test_decode(self):
assert_equal(decode(r'blah'), u'blah')
assert_equal(decode(r'日本語'), u'日本語')
@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 IterationTestCase(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 EnvironmentalVariableTestCase(TestCase):
def test_in_bash(self):
os.environ['SHELL'] = '/bin/bash'
assert_true(in_bash())
os.environ['SHELL'] = '/usr/bin/zsh'
assert_false(in_bash())
if __name__ == "__main__":
run()