815f07d6c1
Also: Allow override of database name and user in tests (important for me as I would have to mess with my PSQL and MySQL database users otherwise) Use dict.items instead of six.iteritems as it sporadically caused RuntimeError: dictionary changed size during iteration in Python 2.6 tests. Fix typo DNS to DSN Adds Python 3.5 to tox.ini Added an .editorconfig Import babel.dates in sqlalchemy_utils.i18n as an exception would be raised when using the latest versions of babel.
72 lines
1.9 KiB
Python
72 lines
1.9 KiB
Python
import pytest
|
|
import six
|
|
|
|
from sqlalchemy_utils import Country, i18n
|
|
|
|
|
|
@pytest.fixture
|
|
def set_get_locale():
|
|
i18n.get_locale = lambda: i18n.babel.Locale('en')
|
|
|
|
|
|
@pytest.mark.skipif('i18n.babel is None')
|
|
@pytest.mark.usefixtures('set_get_locale')
|
|
class TestCountry(object):
|
|
|
|
def test_init(self):
|
|
assert Country(u'FI') == Country(Country(u'FI'))
|
|
|
|
def test_constructor_with_wrong_type(self):
|
|
with pytest.raises(TypeError) as e:
|
|
Country(None)
|
|
assert str(e.value) == (
|
|
"Country() argument must be a string or a country, not 'NoneType'"
|
|
)
|
|
|
|
def test_constructor_with_invalid_code(self):
|
|
with pytest.raises(ValueError) as e:
|
|
Country('SomeUnknownCode')
|
|
assert str(e.value) == (
|
|
'Could not convert string to country code: SomeUnknownCode'
|
|
)
|
|
|
|
@pytest.mark.parametrize(
|
|
'code',
|
|
(
|
|
'FI',
|
|
'US',
|
|
)
|
|
)
|
|
def test_validate_with_valid_codes(self, code):
|
|
Country.validate(code)
|
|
|
|
def test_validate_with_invalid_code(self):
|
|
with pytest.raises(ValueError) as e:
|
|
Country.validate('SomeUnknownCode')
|
|
assert str(e.value) == (
|
|
'Could not convert string to country code: SomeUnknownCode'
|
|
)
|
|
|
|
def test_equality_operator(self):
|
|
assert Country(u'FI') == u'FI'
|
|
assert u'FI' == Country(u'FI')
|
|
assert Country(u'FI') == Country(u'FI')
|
|
|
|
def test_non_equality_operator(self):
|
|
assert Country(u'FI') != u'sv'
|
|
assert not (Country(u'FI') != u'FI')
|
|
|
|
def test_hash(self):
|
|
return hash(Country('FI')) == hash('FI')
|
|
|
|
def test_repr(self):
|
|
return repr(Country('FI')) == "Country('FI')"
|
|
|
|
def test_unicode(self):
|
|
country = Country('FI')
|
|
assert six.text_type(country) == u'Finland'
|
|
|
|
def test_str(self):
|
|
country = Country('FI')
|
|
assert str(country) == 'Finland'
|