tests: add Index test case

tortoise · Dec 6, 2024 · 5de31d7 · 5de31d7
1 parent ff22def
commit 5de31d7
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 1 deletion.
diff --git a/tests/models.py b/tests/models.py
@@ -3,6 +3,7 @@
 from enum import IntEnum
 
 from tortoise import Model, fields
+from tortoise.indexes import Index
 
 
 class ProductType(IntEnum):
@@ -33,6 +34,9 @@ class User(Model):
 
     products: fields.ManyToManyRelation["Product"]
 
+    class Meta:
+        indexes = [Index(fields=("username",))]
+
 
 class Email(Model):
     email_id = fields.IntField(primary_key=True)

diff --git a/tests/old_models.py b/tests/old_models.py
@@ -2,6 +2,7 @@
 from enum import IntEnum
 
 from tortoise import Model, fields
+from tortoise.indexes import Index
 
 
 class ProductType(IntEnum):
@@ -31,6 +32,9 @@ class User(Model):
     intro = fields.TextField(default="")
     longitude = fields.DecimalField(max_digits=12, decimal_places=9)
 
+    class Meta:
+        indexes = [Index(fields=("username",))]
+
 
 class Email(Model):
     email = fields.CharField(max_length=200)

diff --git a/tests/test_migrate.py b/tests/test_migrate.py
@@ -4,6 +4,7 @@
 import pytest
 import tortoise
 from pytest_mock import MockerFixture
+from tortoise.indexes import Index
 
 from aerich.ddl.mysql import MysqlDDL
 from aerich.ddl.postgres import PostgresDDL
@@ -565,7 +566,7 @@
         "description": None,
         "docstring": None,
         "unique_together": [],
-        "indexes": [],
+        "indexes": [Index(fields=("username",))],
         "pk_field": {
             "name": "id",
             "field_type": "IntField",