204aba376d
- Make babel dependent primitive types to use Locale('en') for data
validation instead of current locale. Using current locale leads to
infinite recursion in cases where the loaded data has dependency to
the loaded object's locale.
166 lines
4.9 KiB
Python
166 lines
4.9 KiB
Python
import pytest
|
|
import six
|
|
from flexmock import flexmock
|
|
|
|
from sqlalchemy_utils import i18n
|
|
from sqlalchemy_utils.primitives import WeekDay, WeekDays
|
|
|
|
|
|
@pytest.mark.skipif('i18n.babel is None')
|
|
class TestWeekDay(object):
|
|
def setup_method(self, method):
|
|
i18n.get_locale = lambda: i18n.babel.Locale('fi')
|
|
|
|
def test_constructor_with_valid_index(self):
|
|
day = WeekDay(1)
|
|
assert day.index == 1
|
|
|
|
@pytest.mark.parametrize('index', [-1, 7])
|
|
def test_constructor_with_invalid_index(self, index):
|
|
with pytest.raises(ValueError):
|
|
WeekDay(index)
|
|
|
|
def test_equality_with_equal_week_day(self):
|
|
day = WeekDay(1)
|
|
day2 = WeekDay(1)
|
|
assert day == day2
|
|
|
|
def test_equality_with_unequal_week_day(self):
|
|
day = WeekDay(1)
|
|
day2 = WeekDay(2)
|
|
assert day != day2
|
|
|
|
def test_equality_with_unsupported_comparison(self):
|
|
day = WeekDay(1)
|
|
assert day != 'foobar'
|
|
|
|
def test_hash_is_equal_to_index_hash(self):
|
|
day = WeekDay(1)
|
|
assert hash(day) == hash(day.index)
|
|
|
|
def test_representation(self):
|
|
day = WeekDay(1)
|
|
assert repr(day) == "WeekDay(1)"
|
|
|
|
@pytest.mark.parametrize(
|
|
('index', 'first_week_day', 'position'),
|
|
[
|
|
(0, 0, 0),
|
|
(1, 0, 1),
|
|
(6, 0, 6),
|
|
(0, 6, 1),
|
|
(1, 6, 2),
|
|
(6, 6, 0),
|
|
]
|
|
)
|
|
def test_position(self, index, first_week_day, position):
|
|
i18n.get_locale = flexmock(first_week_day=first_week_day)
|
|
day = WeekDay(index)
|
|
assert day.position == position
|
|
|
|
def test_get_name_returns_localized_week_day_name(self):
|
|
day = WeekDay(0)
|
|
assert day.get_name() == u'maanantaina'
|
|
|
|
def test_override_get_locale_as_class_method(self):
|
|
day = WeekDay(0)
|
|
assert day.get_name() == u'maanantaina'
|
|
|
|
def test_name_delegates_to_get_name(self):
|
|
day = WeekDay(0)
|
|
flexmock(day).should_receive('get_name').and_return(u'maanantaina')
|
|
assert day.name == u'maanantaina'
|
|
|
|
def test_unicode(self):
|
|
day = WeekDay(0)
|
|
flexmock(day).should_receive('name').and_return(u'maanantaina')
|
|
assert six.text_type(day) == u'maanantaina'
|
|
|
|
def test_str(self):
|
|
day = WeekDay(0)
|
|
flexmock(day).should_receive('name').and_return(u'maanantaina')
|
|
assert str(day) == 'maanantaina'
|
|
|
|
|
|
@pytest.mark.skipif('i18n.babel is None')
|
|
class TestWeekDays(object):
|
|
def test_constructor_with_valid_bit_string(self):
|
|
days = WeekDays('1000100')
|
|
assert days._days == set([WeekDay(0), WeekDay(4)])
|
|
|
|
@pytest.mark.parametrize(
|
|
'bit_string',
|
|
[
|
|
'000000', # too short
|
|
'00000000', # too long
|
|
]
|
|
)
|
|
def test_constructor_with_bit_string_of_invalid_length(self, bit_string):
|
|
with pytest.raises(ValueError):
|
|
WeekDays(bit_string)
|
|
|
|
def test_constructor_with_bit_string_containing_invalid_characters(self):
|
|
with pytest.raises(ValueError):
|
|
WeekDays('foobarz')
|
|
|
|
def test_constructor_with_another_week_days_object(self):
|
|
days = WeekDays('0000000')
|
|
another_days = WeekDays(days)
|
|
assert days._days == another_days._days
|
|
|
|
def test_representation(self):
|
|
days = WeekDays('0000000')
|
|
assert repr(days) == "WeekDays('0000000')"
|
|
|
|
@pytest.mark.parametrize(
|
|
'bit_string',
|
|
[
|
|
'0000000',
|
|
'1000000',
|
|
'0000001',
|
|
'0101000',
|
|
'1111111',
|
|
]
|
|
)
|
|
def test_as_bit_string(self, bit_string):
|
|
days = WeekDays(bit_string)
|
|
assert days.as_bit_string() == bit_string
|
|
|
|
def test_equality_with_equal_week_days_object(self):
|
|
days = WeekDays('0001000')
|
|
days2 = WeekDays('0001000')
|
|
assert days == days2
|
|
|
|
def test_equality_with_unequal_week_days_object(self):
|
|
days = WeekDays('0001000')
|
|
days2 = WeekDays('1000000')
|
|
assert days != days2
|
|
|
|
def test_equality_with_equal_bit_string(self):
|
|
days = WeekDays('0001000')
|
|
assert days == '0001000'
|
|
|
|
def test_equality_with_unequal_bit_string(self):
|
|
days = WeekDays('0001000')
|
|
assert days != '0101000'
|
|
|
|
def test_equality_with_unsupported_comparison(self):
|
|
days = WeekDays('0001000')
|
|
assert days != 0
|
|
|
|
def test_iterator_starts_from_locales_first_week_day(self):
|
|
i18n.get_locale = lambda: flexmock(first_week_day=1)
|
|
days = WeekDays('1111111')
|
|
indices = list(day.index for day in days)
|
|
assert indices == [1, 2, 3, 4, 5, 6, 0]
|
|
|
|
def test_unicode(self):
|
|
i18n.get_locale = lambda: i18n.babel.Locale('fi')
|
|
days = WeekDays('1000100')
|
|
assert six.text_type(days) == u'maanantaina, perjantaina'
|
|
|
|
def test_str(self):
|
|
i18n.get_locale = lambda: i18n.babel.Locale('fi')
|
|
days = WeekDays('1000100')
|
|
assert str(days) == 'maanantaina, perjantaina'
|