mirrors/django
1
0
Fork 0
mirror of https://github.com/django/django.git synced 2025-01-03 06:55:47 +00:00
Code Issues 30 Releases Wiki Activity

Added SchemaTests._add_ci_collation() hook.

Browse Source
This commit is contained in:
Mariusz Felisiak 2023-04-21 09:59:01 +02:00
parent 8a6c0203c4
commit 765b96734c
1 changed files with 17 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
tests/schema/tests.py
View File

@ -1293,6 +1293,21 @@ class SchemaTests(TransactionTestCase):
with self.assertRaisesMessage(DataError, msg): with self.assertRaisesMessage(DataError, msg):
editor.alter_field(ArrayModel, old_field, new_field, strict=True) editor.alter_field(ArrayModel, old_field, new_field, strict=True)
def _add_ci_collation(self):
ci_collation = "case_insensitive"
def drop_collation():
with connection.cursor() as cursor:
cursor.execute(f"DROP COLLATION IF EXISTS {ci_collation}")
with connection.cursor() as cursor:
cursor.execute(
f"CREATE COLLATION IF NOT EXISTS {ci_collation} (provider=icu, "
f"locale='und-u-ks-level2', deterministic=false)"
)
self.addCleanup(drop_collation)
return ci_collation
@isolate_apps("schema") @isolate_apps("schema")
@unittest.skipUnless(connection.vendor == "postgresql", "PostgreSQL specific") @unittest.skipUnless(connection.vendor == "postgresql", "PostgreSQL specific")
@skipUnlessDBFeature( @skipUnlessDBFeature(
@ -1302,20 +1317,9 @@ class SchemaTests(TransactionTestCase):
def test_db_collation_arrayfield(self): def test_db_collation_arrayfield(self):
from django.contrib.postgres.fields import ArrayField from django.contrib.postgres.fields import ArrayField
ci_collation = "case_insensitive" ci_collation = self._add_ci_collation()
cs_collation = "en-x-icu" cs_collation = "en-x-icu"
def drop_collation():
with connection.cursor() as cursor:
cursor.execute(f"DROP COLLATION IF EXISTS {ci_collation}")
with connection.cursor() as cursor:
cursor.execute(
f"CREATE COLLATION IF NOT EXISTS {ci_collation} (provider = icu, "
f"locale = 'und-u-ks-level2', deterministic = false)"
)
self.addCleanup(drop_collation)
class ArrayModel(Model): class ArrayModel(Model):
field = ArrayField(CharField(max_length=16, db_collation=ci_collation)) field = ArrayField(CharField(max_length=16, db_collation=ci_collation))
@ -1349,18 +1353,7 @@ class SchemaTests(TransactionTestCase):
"supports_non_deterministic_collations", "supports_non_deterministic_collations",
) )
def test_unique_with_collation_charfield(self): def test_unique_with_collation_charfield(self):
ci_collation = "case_insensitive" ci_collation = self._add_ci_collation()
def drop_collation():
with connection.cursor() as cursor:
cursor.execute(f"DROP COLLATION IF EXISTS {ci_collation}")
with connection.cursor() as cursor:
cursor.execute(
f"CREATE COLLATION IF NOT EXISTS {ci_collation} (provider = icu, "
f"locale = 'und-u-ks-level2', deterministic = false)"
)
self.addCleanup(drop_collation)
class CiCharModel(Model): class CiCharModel(Model):
field = CharField(max_length=16, db_collation=ci_collation, unique=True) field = CharField(max_length=16, db_collation=ci_collation, unique=True)