Add indexes_ tests

Author: WaVEV

.github/workflows/runtests.py

@@ -29,6 +29,7 @@
     "generic_relations",
     "generic_relations_regress",
     "indexes",
+    "indexes_",
     "introspection",
     "known_related_objects",
     "lookup",

tests/indexes_/__init__.py

@@ -0,0 +1,12 @@
+from django.db import models
+
+
+class Article(models.Model):
+    headline = models.CharField(max_length=100)
+    pub_date = models.DateTimeField()
+    published = models.BooleanField(default=False)
+    number = models.IntegerField()
+    text = models.CharField(max_length=100)
+
+    class Meta:
+        indexes = [models.Index(fields=["headline", "pub_date"])]

tests/indexes_/models.py

@@ -0,0 +1,69 @@
+from django.db import NotSupportedError, connection
+from django.db.models import (
+    Index,
+    Q,
+)
+from django.test import (
+    TestCase,
+)
+
+from .models import Article
+
+
+class PartialIndexTests(TestCase):
+    def test_not_supported(self):
+        with connection.schema_editor() as editor:
+            index = Index(
+                name="not_supported",
+                # This is changed
+                fields=["headline"],
+                condition=Q(pk__isnull=True) & Q(pk__contains="test1"),
+            )
+            self.assertEqual(
+                "{}",
+                str(index._get_condition_mql(Article, schema_editor=editor)),
+            )
+            editor.add_index(index=index, model=Article)
+            with connection.cursor() as cursor:
+                self.assertIn(
+                    index.name,
+                    connection.introspection.get_constraints(
+                        cursor=cursor,
+                        table_name=Article._meta.db_table,
+                    ),
+                )
+            editor.remove_index(index=index, model=Article)
+
+    def test_raises_on_negated(self):
+        with self.assertRaises(NotSupportedError), connection.schema_editor() as editor:
+            Index(
+                name="raises_on_negated",
+                # This is changed
+                fields=["headline"],
+                condition=~Q(pk=True),
+            )._get_condition_mql(Article, schema_editor=editor)
+
+    def test_composite_index(self):
+        with connection.schema_editor() as editor:
+            index = Index(
+                name="composite_proposition",
+                # This is changed
+                fields=["headline"],
+                condition=Q(number__gte=3) & (Q(text__gt="test1") | Q(text__in=["A", "B"])),
+            )
+            index._get_condition_mql(Article, schema_editor=editor)
+            self.assertEqual(
+                "{'$and': [{'number': {'$gte': 3}}, {'$or': [{'text': {'$gt': 'test1'}}, "
+                "{'text': {'$in': ['A', 'B']}}]}]}",
+                str(index._get_condition_mql(Article, schema_editor=editor)),
+            )
+            editor.add_index(index=index, model=Article)
+            with connection.cursor() as cursor:
+                self.assertIn(
+                    index.name,
+                    connection.introspection.get_constraints(
+                        cursor=cursor,
+                        table_name=Article._meta.db_table,
+                    ),
+                )
+            editor.remove_index(index=index, model=Article)