diff options
Diffstat (limited to 'tests/postgres_tests/test_indexes.py')
| -rw-r--r-- | tests/postgres_tests/test_indexes.py | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/tests/postgres_tests/test_indexes.py b/tests/postgres_tests/test_indexes.py index 8c15f01d47..235d72c532 100644 --- a/tests/postgres_tests/test_indexes.py +++ b/tests/postgres_tests/test_indexes.py @@ -44,13 +44,14 @@ class SchemaTests(PostgreSQLTestCase): # Ensure the table is there and doesn't have an index. self.assertNotIn('field', self.get_constraints(IntegerArrayModel._meta.db_table)) # Add the index - index = GinIndex(fields=['field'], name='integer_array_model_field_gin') + index_name = 'integer_array_model_field_gin' + index = GinIndex(fields=['field'], name=index_name) with connection.schema_editor() as editor: editor.add_index(IntegerArrayModel, index) - self.assertIn('integer_array_model_field_gin', self.get_constraints(IntegerArrayModel._meta.db_table)) constraints = self.get_constraints(IntegerArrayModel._meta.db_table) - self.assertEqual(constraints['integer_array_model_field_gin']['type'], 'gin') + # Check gin index was added + self.assertEqual(constraints[index_name]['type'], 'gin') # Drop the index with connection.schema_editor() as editor: editor.remove_index(IntegerArrayModel, index) - self.assertNotIn('integer_array_model_field_gin', self.get_constraints(IntegerArrayModel._meta.db_table)) + self.assertNotIn(index_name, self.get_constraints(IntegerArrayModel._meta.db_table)) |