28 lines
711 B
Python
28 lines
711 B
Python
import sqlalchemy as sa
|
|
from sqlalchemy_utils import EmailType
|
|
from tests import TestCase
|
|
|
|
|
|
class TestEmailType(TestCase):
|
|
def create_models(self):
|
|
class User(self.Base):
|
|
__tablename__ = 'user'
|
|
id = sa.Column(sa.Integer, primary_key=True)
|
|
email = sa.Column(EmailType)
|
|
|
|
def __repr__(self):
|
|
return 'User(%r)' % self.id
|
|
|
|
self.User = User
|
|
|
|
def test_saves_email_as_lowercased(self):
|
|
user = self.User(
|
|
email=u'Someone@example.com'
|
|
)
|
|
|
|
self.session.add(user)
|
|
self.session.commit()
|
|
|
|
user = self.session.query(self.User).first()
|
|
assert user.email == u'someone@example.com'
|