1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60
|
import pytest
import sqlalchemy as sa
from sqlalchemy_utils import EmailType
@pytest.fixture
def User(Base):
class User(Base):
__tablename__ = 'user'
id = sa.Column(sa.Integer, primary_key=True)
email = sa.Column(EmailType)
def __repr__(self):
return 'Building(%r)' % self.id
return User
@pytest.fixture
def init_models(User):
pass
class TestCaseInsensitiveComparator:
def test_supports_equals(self, session, User):
query = (
session.query(User)
.filter(User.email == 'email@example.com')
)
assert 'user.email = lower(?)' in str(query)
def test_supports_in_(self, session, User):
query = (
session.query(User)
.filter(User.email.in_(['email@example.com', 'a']))
)
assert (
'user.email IN (lower(?), lower(?))'
in str(query)
)
def test_supports_notin_(self, session, User):
query = (
session.query(User)
.filter(User.email.notin_(['email@example.com', 'a']))
)
assert (
'user.email NOT IN ('
in str(query)
)
def test_does_not_apply_lower_to_types_that_are_already_lowercased(
self,
User
):
assert str(User.email == User.email) == (
'"user".email = "user".email'
)
|