Fixed #29614 -- Added BTreeIndex to django.contrib.postres.

1 files changed, 43 insertions, 1 deletions

diff --git a/tests/postgres_tests/test_indexes.py b/tests/postgres_tests/test_indexes.py
index 9bca4510fe..96ff829807 100644
--- a/tests/postgres_tests/test_indexes.py
+++ b/tests/postgres_tests/test_indexes.py
@@ -1,5 +1,5 @@
 from django.contrib.postgres.indexes import (
-    BrinIndex, GinIndex, GistIndex, HashIndex, SpGistIndex,
+    BrinIndex, BTreeIndex, GinIndex, GistIndex, HashIndex, SpGistIndex,
 )
 from django.db import connection
 from django.test import skipUnlessDBFeature
@@ -47,6 +47,20 @@ class BrinIndexTests(IndexTestMixin, PostgreSQLTestCase):
             BrinIndex(fields=['title'], name='test_title_brin', pages_per_range=0)
 
 
+class BTreeIndexTests(IndexTestMixin, PostgreSQLTestCase):
+    index_class = BTreeIndex
+
+    def test_suffix(self):
+        self.assertEqual(BTreeIndex.suffix, 'btree')
+
+    def test_deconstruction(self):
+        index = BTreeIndex(fields=['title'], name='test_title_btree', fillfactor=80)
+        path, args, kwargs = index.deconstruct()
+        self.assertEqual(path, 'django.contrib.postgres.indexes.BTreeIndex')
+        self.assertEqual(args, ())
+        self.assertEqual(kwargs, {'fields': ['title'], 'name': 'test_title_btree', 'fillfactor': 80})
+
+
 class GinIndexTests(IndexTestMixin, PostgreSQLTestCase):
     index_class = GinIndex
 
@@ -194,6 +208,34 @@ class SchemaTests(PostgreSQLTestCase):
             editor.remove_index(CharFieldModel, index)
         self.assertNotIn(index_name, self.get_constraints(CharFieldModel._meta.db_table))
 
+    def test_btree_index(self):
+        # Ensure the table is there and doesn't have an index.
+        self.assertNotIn('field', self.get_constraints(CharFieldModel._meta.db_table))
+        # Add the index.
+        index_name = 'char_field_model_field_btree'
+        index = BTreeIndex(fields=['field'], name=index_name)
+        with connection.schema_editor() as editor:
+            editor.add_index(CharFieldModel, index)
+        constraints = self.get_constraints(CharFieldModel._meta.db_table)
+        # The index was added.
+        self.assertEqual(constraints[index_name]['type'], BTreeIndex.suffix)
+        # Drop the index.
+        with connection.schema_editor() as editor:
+            editor.remove_index(CharFieldModel, index)
+        self.assertNotIn(index_name, self.get_constraints(CharFieldModel._meta.db_table))
+
+    def test_btree_parameters(self):
+        index_name = 'integer_array_btree_fillfactor'
+        index = BTreeIndex(fields=['field'], name=index_name, fillfactor=80)
+        with connection.schema_editor() as editor:
+            editor.add_index(CharFieldModel, index)
+        constraints = self.get_constraints(CharFieldModel._meta.db_table)
+        self.assertEqual(constraints[index_name]['type'], BTreeIndex.suffix)
+        self.assertEqual(constraints[index_name]['options'], ['fillfactor=80'])
+        with connection.schema_editor() as editor:
+            editor.remove_index(CharFieldModel, index)
+        self.assertNotIn(index_name, self.get_constraints(CharFieldModel._meta.db_table))
+
     def test_gist_index(self):
         # Ensure the table is there and doesn't have an index.
         self.assertNotIn('field', self.get_constraints(CharFieldModel._meta.db_table))