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
|
import sqlalchemy as sa
from pytest import raises
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy_utils import has_index
class TestHasIndex(object):
def setup_method(self, method):
Base = declarative_base()
class ArticleTranslation(Base):
__tablename__ = 'article_translation'
id = sa.Column(sa.Integer, primary_key=True)
locale = sa.Column(sa.String(10), primary_key=True)
title = sa.Column(sa.String(100))
is_published = sa.Column(sa.Boolean, index=True)
is_deleted = sa.Column(sa.Boolean)
is_archived = sa.Column(sa.Boolean)
__table_args__ = (
sa.Index('my_index', is_deleted, is_archived),
)
self.table = ArticleTranslation.__table__
def test_column_that_belongs_to_an_alias(self):
alias = sa.orm.aliased(self.table)
with raises(TypeError):
assert has_index(alias.c.id)
def test_compound_primary_key(self):
assert has_index(self.table.c.id)
assert not has_index(self.table.c.locale)
def test_single_column_index(self):
assert has_index(self.table.c.is_published)
def test_compound_column_index(self):
assert has_index(self.table.c.is_deleted)
assert not has_index(self.table.c.is_archived)
def test_table_without_primary_key(self):
article = sa.Table(
'article',
sa.MetaData(),
sa.Column('name', sa.String)
)
assert not has_index(article.c.name)
|
