Simplified indexes tests.

This commit is contained in:
Aymeric Augustin 2014-05-07 21:49:45 +02:00
parent d667263169
commit 43a80f4812
1 changed files with 2 additions and 5 deletions

View File

@ -1,7 +1,7 @@
from unittest import skipUnless from unittest import skipUnless
from django.core.management.color import no_style from django.core.management.color import no_style
from django.db import connections, DEFAULT_DB_ALIAS from django.db import connection
from django.test import TestCase from django.test import TestCase
from .models import Article, IndexTogetherSingleList from .models import Article, IndexTogetherSingleList
@ -9,22 +9,19 @@ from .models import Article, IndexTogetherSingleList
class IndexesTests(TestCase): class IndexesTests(TestCase):
def test_index_together(self): def test_index_together(self):
connection = connections[DEFAULT_DB_ALIAS]
index_sql = connection.creation.sql_indexes_for_model(Article, no_style()) index_sql = connection.creation.sql_indexes_for_model(Article, no_style())
self.assertEqual(len(index_sql), 1) self.assertEqual(len(index_sql), 1)
def test_index_together_single_list(self): def test_index_together_single_list(self):
# Test for using index_together with a single list (#22172) # Test for using index_together with a single list (#22172)
connection = connections[DEFAULT_DB_ALIAS]
index_sql = connection.creation.sql_indexes_for_model(IndexTogetherSingleList, no_style()) index_sql = connection.creation.sql_indexes_for_model(IndexTogetherSingleList, no_style())
self.assertEqual(len(index_sql), 1) self.assertEqual(len(index_sql), 1)
@skipUnless(connections[DEFAULT_DB_ALIAS].vendor == 'postgresql', @skipUnless(connection.vendor == 'postgresql',
"This is a postgresql-specific issue") "This is a postgresql-specific issue")
def test_postgresql_text_indexes(self): def test_postgresql_text_indexes(self):
"""Test creation of PostgreSQL-specific text indexes (#12234)""" """Test creation of PostgreSQL-specific text indexes (#12234)"""
from .models import IndexedArticle from .models import IndexedArticle
connection = connections[DEFAULT_DB_ALIAS]
index_sql = connection.creation.sql_indexes_for_model(IndexedArticle, no_style()) index_sql = connection.creation.sql_indexes_for_model(IndexedArticle, no_style())
self.assertEqual(len(index_sql), 5) self.assertEqual(len(index_sql), 5)
self.assertIn('("headline" varchar_pattern_ops)', index_sql[1]) self.assertIn('("headline" varchar_pattern_ops)', index_sql[1])