def test_index_repr(self):
"""index creates descriptive representation string"""
index = PgPartialIndex(
fields=['has_events'],
where={'has_events': True},
)
self.assertEqual(
repr(index),
"<PgPartialIndex: fields='has_events', where='has_events=True'>")
index = PgPartialIndex(
fields=['has_events', 'is_hidden'],
where={'has_events': True},
)
self.assertIn("fields='has_events, is_hidden',", repr(index))
self.assertIn(", where='has_events=True'", repr(index))
index = PgPartialIndex(
fields=['has_events', 'is_hidden', 'is_closed'],
where={
'has_events': True,
'is_closed': False,
'replies__gte': 5,
},
)
self.assertIn("fields='has_events, is_hidden, is_closed',",
repr(index))
self.assertIn(
", where='has_events=True, is_closed=False, replies__gte=5'",
repr(index))
def test_set_name_with_model(self):
"""valid index name is autogenerated"""
index = PgPartialIndex(
fields=['has_events', 'is_hidden'],
where={'has_events': True},
)
index.set_name_with_model(Thread)
self.assertEqual(index.name, 'misago_thre_has_eve_1b05b8_part')
index = PgPartialIndex(
fields=['has_events', 'is_hidden', 'is_closed'],
where={'has_events': True},
)
index.set_name_with_model(Thread)
self.assertEqual(index.name, 'misago_thre_has_eve_eaab5e_part')
index = PgPartialIndex(
fields=['has_events', 'is_hidden', 'is_closed'],
where={
'has_events': True,
'is_closed': False,
},
)
index.set_name_with_model(Thread)
self.assertEqual(index.name, 'misago_thre_has_eve_e738fe_part')
class Meta:
indexes = [
PgPartialIndex(
fields=['is_staff'],
where={'is_staff': True},
),
PgPartialIndex(
fields=['requires_activation'],
where={'requires_activation__gt': 0},
),
]
def test_multiple_where_clauses(self):
"""where clause with multiple conditions generates correctly"""
with connection.schema_editor() as editor:
sql = PgPartialIndex(
fields=['has_events'],
name='test_partial',
where={
'has_events__gte': 42,
'is_hidden': True,
},
).create_sql(Thread, editor)
self.assertTrue(sql.endswith('WHERE "has_events" >= 42 AND "is_hidden" = true'))
def test_multiple_where_clauses(self):
"""where clause with multiple conditions generates correctly"""
with connection.schema_editor() as editor:
sql = PgPartialIndex(
fields=['has_events'],
name='test_partial',
where={
'has_events__gte': 42,
'is_hidden': True,
},
).create_sql(Thread, editor)
self.assertTrue(
sql.endswith(
'WHERE "has_events" >= 42 AND "is_hidden" = true'))
class Meta:
indexes = [
PgPartialIndex(
fields=['has_open_reports'],
where={'has_open_reports': True},
),
PgPartialIndex(
fields=['is_hidden'],
where={'is_hidden': False},
),
GinIndex(fields=['search_vector']),
]
index_together = [
('thread', 'id'), # speed up threadview for team members
('is_event', 'is_hidden'),
('poster', 'posted_on'),
]
def test_multiple_fields(self):
"""multiple fields are supported"""
with connection.schema_editor() as editor:
sql = PgPartialIndex(
fields=['has_events', 'is_hidden'],
name='test_partial',
where={
'has_events': True
},
).create_sql(Thread, editor)
self.assertIn(
'CREATE INDEX "test_partial" ON "misago_threads_thread"', sql)
self.assertIn(
'ON "misago_threads_thread" ("has_events", "is_hidden")', sql)
class Meta:
indexes = [
PgPartialIndex(
fields=['weight'],
where={'weight': 2},
),
PgPartialIndex(
fields=['weight'],
where={'weight': 1},
),
PgPartialIndex(
fields=['weight'],
where={'weight': 0},
),
PgPartialIndex(
fields=['weight'],
where={'weight__lt': 2},
),
PgPartialIndex(
fields=['has_reported_posts'],
where={'has_reported_posts': True},
),
PgPartialIndex(
fields=['has_unapproved_posts'],
where={'has_unapproved_posts': True},
),
PgPartialIndex(
fields=['is_hidden'],
where={'is_hidden': False},
),
]
index_together = [
['category', 'id'],
['category', 'last_post_on'],
['category', 'replies'],
]
def test_where_clauses(self):
"""where clauses generate correctly"""
with connection.schema_editor() as editor:
sql = PgPartialIndex(
fields=['has_events'],
name='test_partial',
where={
'has_events': True
},
).create_sql(Thread, editor)
self.assertTrue(sql.endswith('WHERE "has_events" = true'))
sql = PgPartialIndex(
fields=['has_events'],
name='test_partial',
where={
'has_events': False
},
).create_sql(Thread, editor)
self.assertTrue(sql.endswith('WHERE "has_events" = false'))
sql = PgPartialIndex(
fields=['has_events'],
name='test_partial',
where={
'has_events': 42
},
).create_sql(Thread, editor)
self.assertTrue(sql.endswith('WHERE "has_events" = 42'))
sql = PgPartialIndex(
fields=['has_events'],
name='test_partial',
where={
'has_events__lt': 42
},
).create_sql(Thread, editor)
self.assertTrue(sql.endswith('WHERE "has_events" < 42'))
sql = PgPartialIndex(
fields=['has_events'],
name='test_partial',
where={
'has_events__gt': 42
},
).create_sql(Thread, editor)
self.assertTrue(sql.endswith('WHERE "has_events" > 42'))
sql = PgPartialIndex(
fields=['has_events'],
name='test_partial',
where={
'has_events__lte': 42
},
).create_sql(Thread, editor)
self.assertTrue(sql.endswith('WHERE "has_events" <= 42'))
sql = PgPartialIndex(
fields=['has_events'],
name='test_partial',
where={
'has_events__gte': 42
},
).create_sql(Thread, editor)
self.assertTrue(sql.endswith('WHERE "has_events" >= 42'))
def test_set_name_with_model(self):
"""valid index name is autogenerated"""
index = PgPartialIndex(
fields=['has_events', 'is_hidden'],
where={'has_events': True},
)
index.set_name_with_model(Thread)
self.assertEqual(index.name, 'misago_thre_has_eve_1b05b8_part')
index = PgPartialIndex(
fields=['has_events', 'is_hidden', 'is_closed'],
where={'has_events': True},
)
index.set_name_with_model(Thread)
self.assertEqual(index.name, 'misago_thre_has_eve_eaab5e_part')
index = PgPartialIndex(
fields=['has_events', 'is_hidden', 'is_closed'],
where={
'has_events': True,
'is_closed': False,
},
)
index.set_name_with_model(Thread)
self.assertEqual(index.name, 'misago_thre_has_eve_e738fe_part')
def test_where_clauses(self):
"""where clauses generate correctly"""
with connection.schema_editor() as editor:
sql = PgPartialIndex(
fields=['has_events'],
name='test_partial',
where={'has_events': True},
).create_sql(Thread, editor)
self.assertTrue(sql.endswith('WHERE "has_events" = true'))
sql = PgPartialIndex(
fields=['has_events'],
name='test_partial',
where={'has_events': False},
).create_sql(Thread, editor)
self.assertTrue(sql.endswith('WHERE "has_events" = false'))
sql = PgPartialIndex(
fields=['has_events'],
name='test_partial',
where={'has_events': 42},
).create_sql(Thread, editor)
self.assertTrue(sql.endswith('WHERE "has_events" = 42'))
sql = PgPartialIndex(
fields=['has_events'],
name='test_partial',
where={'has_events__lt': 42},
).create_sql(Thread, editor)
self.assertTrue(sql.endswith('WHERE "has_events" < 42'))
sql = PgPartialIndex(
fields=['has_events'],
name='test_partial',
where={'has_events__gt': 42},
).create_sql(Thread, editor)
self.assertTrue(sql.endswith('WHERE "has_events" > 42'))
sql = PgPartialIndex(
fields=['has_events'],
name='test_partial',
where={'has_events__lte': 42},
).create_sql(Thread, editor)
self.assertTrue(sql.endswith('WHERE "has_events" <= 42'))
sql = PgPartialIndex(
fields=['has_events'],
name='test_partial',
where={'has_events__gte': 42},
).create_sql(Thread, editor)
self.assertTrue(sql.endswith('WHERE "has_events" >= 42'))