1
0
mirror of https://github.com/django/django.git synced 2025-10-24 06:06:09 +00:00

Refs #26610 -- Added CIText mixin and CIChar/Email/TextField.

This commit is contained in:
Mads Jensen
2017-02-11 13:16:35 +01:00
committed by Tim Graham
parent fe2d288434
commit fb5bd38e3b
7 changed files with 71 additions and 36 deletions

View File

@@ -6,9 +6,9 @@ from django.db import models
try:
from django.contrib.postgres.fields import (
ArrayField, BigIntegerRangeField, CITextField, DateRangeField,
DateTimeRangeField, FloatRangeField, HStoreField, IntegerRangeField,
JSONField,
ArrayField, BigIntegerRangeField, CICharField, CIEmailField,
CITextField, DateRangeField, DateTimeRangeField, FloatRangeField,
HStoreField, IntegerRangeField, JSONField,
)
from django.contrib.postgres.search import SearchVectorField
except ImportError:
@@ -30,6 +30,8 @@ except ImportError:
ArrayField = DummyArrayField
BigIntegerRangeField = models.Field
CICharField = models.Field
CIEmailField = models.Field
CITextField = models.Field
DateRangeField = models.Field
DateTimeRangeField = models.Field

View File

@@ -2,9 +2,9 @@ from django.core.serializers.json import DjangoJSONEncoder
from django.db import migrations, models
from ..fields import (
ArrayField, BigIntegerRangeField, CITextField, DateRangeField,
DateTimeRangeField, FloatRangeField, HStoreField, IntegerRangeField,
JSONField, SearchVectorField,
ArrayField, BigIntegerRangeField, CICharField, CIEmailField, CITextField,
DateRangeField, DateTimeRangeField, FloatRangeField, HStoreField,
IntegerRangeField, JSONField, SearchVectorField,
)
from ..models import TagField
@@ -136,9 +136,11 @@ class Migration(migrations.Migration):
bases=None,
),
migrations.CreateModel(
name='CITextTestModel',
name='CITestModel',
fields=[
('name', CITextField(primary_key=True, max_length=255)),
('name', CICharField(primary_key=True, max_length=255)),
('email', CIEmailField()),
('description', CITextField()),
],
options={
'required_db_vendor': 'postgresql',

View File

@@ -2,9 +2,9 @@ from django.core.serializers.json import DjangoJSONEncoder
from django.db import models
from .fields import (
ArrayField, BigIntegerRangeField, CITextField, DateRangeField,
DateTimeRangeField, FloatRangeField, HStoreField, IntegerRangeField,
JSONField, SearchVectorField,
ArrayField, BigIntegerRangeField, CICharField, CIEmailField, CITextField,
DateRangeField, DateTimeRangeField, FloatRangeField, HStoreField,
IntegerRangeField, JSONField, SearchVectorField,
)
@@ -101,8 +101,10 @@ class Character(models.Model):
return self.name
class CITextTestModel(PostgreSQLModel):
name = CITextField(primary_key=True, max_length=255)
class CITestModel(PostgreSQLModel):
name = CICharField(primary_key=True, max_length=255)
email = CIEmailField()
description = CITextField()
def __str__(self):
return self.name

View File

@@ -6,25 +6,31 @@ modifiers to enforce use of an index.
from django.db import IntegrityError
from . import PostgreSQLTestCase
from .models import CITextTestModel
from .models import CITestModel
class CITextTestCase(PostgreSQLTestCase):
@classmethod
def setUpTestData(cls):
CITextTestModel.objects.create(name='JoHn')
cls.john = CITestModel.objects.create(
name='JoHn',
email='joHn@johN.com',
description='Average Joe named JoHn',
)
def test_equal_lowercase(self):
"""
citext removes the need for iexact as the index is case-insensitive.
"""
self.assertEqual(CITextTestModel.objects.filter(name='john').count(), 1)
self.assertEqual(CITestModel.objects.filter(name=self.john.name.lower()).count(), 1)
self.assertEqual(CITestModel.objects.filter(email=self.john.email.lower()).count(), 1)
self.assertEqual(CITestModel.objects.filter(description=self.john.description.lower()).count(), 1)
def test_fail_case(self):
def test_fail_citext_primary_key(self):
"""
Creating an entry for a citext-field which clashes with an existing
value isn't allowed.
Creating an entry for a citext field used as a primary key which
clashes with an existing value isn't allowed.
"""
with self.assertRaises(IntegrityError):
CITextTestModel.objects.create(name='John')
CITestModel.objects.create(name='John')