2015-01-11 02:13:28 +08:00
|
|
|
import decimal
|
2014-05-22 22:05:19 +08:00
|
|
|
import json
|
2014-03-27 00:44:21 +08:00
|
|
|
import unittest
|
2015-01-11 02:13:28 +08:00
|
|
|
import uuid
|
2014-03-27 00:44:21 +08:00
|
|
|
|
2015-01-28 20:35:27 +08:00
|
|
|
from django import forms
|
2015-02-20 19:33:02 +08:00
|
|
|
from django.core import exceptions, serializers, validators
|
2014-07-15 19:30:34 +08:00
|
|
|
from django.core.management import call_command
|
2015-01-28 20:35:27 +08:00
|
|
|
from django.db import IntegrityError, connection, models
|
2015-04-05 00:10:26 +08:00
|
|
|
from django.test import TransactionTestCase, override_settings
|
2015-11-17 13:39:28 +08:00
|
|
|
from django.test.utils import isolate_apps
|
2014-03-27 00:44:21 +08:00
|
|
|
from django.utils import timezone
|
|
|
|
|
2015-05-31 05:18:01 +08:00
|
|
|
from . import PostgreSQLTestCase
|
2015-01-11 02:13:28 +08:00
|
|
|
from .models import (
|
2015-01-28 20:35:27 +08:00
|
|
|
ArrayFieldSubclass, CharArrayModel, DateTimeArrayModel, IntegerArrayModel,
|
|
|
|
NestedIntegerArrayModel, NullableIntegerArrayModel, OtherTypesArrayModel,
|
2015-07-23 01:51:05 +08:00
|
|
|
PostgreSQLModel, Tag,
|
2015-01-11 02:13:28 +08:00
|
|
|
)
|
2014-03-27 00:44:21 +08:00
|
|
|
|
2015-04-05 00:10:26 +08:00
|
|
|
try:
|
|
|
|
from django.contrib.postgres.fields import ArrayField
|
|
|
|
from django.contrib.postgres.forms import SimpleArrayField, SplitArrayField
|
|
|
|
except ImportError:
|
|
|
|
pass
|
2014-03-27 00:44:21 +08:00
|
|
|
|
2015-04-05 00:10:26 +08:00
|
|
|
|
2015-05-31 05:18:01 +08:00
|
|
|
class TestSaveLoad(PostgreSQLTestCase):
|
2014-03-27 00:44:21 +08:00
|
|
|
|
|
|
|
def test_integer(self):
|
|
|
|
instance = IntegerArrayModel(field=[1, 2, 3])
|
|
|
|
instance.save()
|
|
|
|
loaded = IntegerArrayModel.objects.get()
|
|
|
|
self.assertEqual(instance.field, loaded.field)
|
|
|
|
|
|
|
|
def test_char(self):
|
|
|
|
instance = CharArrayModel(field=['hello', 'goodbye'])
|
|
|
|
instance.save()
|
|
|
|
loaded = CharArrayModel.objects.get()
|
|
|
|
self.assertEqual(instance.field, loaded.field)
|
|
|
|
|
|
|
|
def test_dates(self):
|
2015-01-11 02:13:28 +08:00
|
|
|
instance = DateTimeArrayModel(
|
|
|
|
datetimes=[timezone.now()],
|
|
|
|
dates=[timezone.now().date()],
|
|
|
|
times=[timezone.now().time()],
|
|
|
|
)
|
2014-03-27 00:44:21 +08:00
|
|
|
instance.save()
|
|
|
|
loaded = DateTimeArrayModel.objects.get()
|
2015-01-11 02:13:28 +08:00
|
|
|
self.assertEqual(instance.datetimes, loaded.datetimes)
|
|
|
|
self.assertEqual(instance.dates, loaded.dates)
|
|
|
|
self.assertEqual(instance.times, loaded.times)
|
2014-03-27 00:44:21 +08:00
|
|
|
|
|
|
|
def test_tuples(self):
|
|
|
|
instance = IntegerArrayModel(field=(1,))
|
|
|
|
instance.save()
|
|
|
|
loaded = IntegerArrayModel.objects.get()
|
|
|
|
self.assertSequenceEqual(instance.field, loaded.field)
|
|
|
|
|
|
|
|
def test_integers_passed_as_strings(self):
|
|
|
|
# This checks that get_prep_value is deferred properly
|
|
|
|
instance = IntegerArrayModel(field=['1'])
|
|
|
|
instance.save()
|
|
|
|
loaded = IntegerArrayModel.objects.get()
|
|
|
|
self.assertEqual(loaded.field, [1])
|
|
|
|
|
2014-11-15 22:04:34 +08:00
|
|
|
def test_default_null(self):
|
|
|
|
instance = NullableIntegerArrayModel()
|
|
|
|
instance.save()
|
|
|
|
loaded = NullableIntegerArrayModel.objects.get(pk=instance.pk)
|
2016-06-17 02:19:18 +08:00
|
|
|
self.assertIsNone(loaded.field)
|
2014-11-15 22:04:34 +08:00
|
|
|
self.assertEqual(instance.field, loaded.field)
|
|
|
|
|
2014-03-27 00:44:21 +08:00
|
|
|
def test_null_handling(self):
|
|
|
|
instance = NullableIntegerArrayModel(field=None)
|
|
|
|
instance.save()
|
|
|
|
loaded = NullableIntegerArrayModel.objects.get()
|
|
|
|
self.assertEqual(instance.field, loaded.field)
|
|
|
|
|
|
|
|
instance = IntegerArrayModel(field=None)
|
|
|
|
with self.assertRaises(IntegrityError):
|
|
|
|
instance.save()
|
|
|
|
|
|
|
|
def test_nested(self):
|
|
|
|
instance = NestedIntegerArrayModel(field=[[1, 2], [3, 4]])
|
|
|
|
instance.save()
|
|
|
|
loaded = NestedIntegerArrayModel.objects.get()
|
|
|
|
self.assertEqual(instance.field, loaded.field)
|
|
|
|
|
2015-01-11 02:13:28 +08:00
|
|
|
def test_other_array_types(self):
|
|
|
|
instance = OtherTypesArrayModel(
|
|
|
|
ips=['192.168.0.1', '::1'],
|
|
|
|
uuids=[uuid.uuid4()],
|
|
|
|
decimals=[decimal.Decimal(1.25), 1.75],
|
2015-07-23 01:51:05 +08:00
|
|
|
tags=[Tag(1), Tag(2), Tag(3)],
|
2015-01-11 02:13:28 +08:00
|
|
|
)
|
|
|
|
instance.save()
|
|
|
|
loaded = OtherTypesArrayModel.objects.get()
|
|
|
|
self.assertEqual(instance.ips, loaded.ips)
|
|
|
|
self.assertEqual(instance.uuids, loaded.uuids)
|
|
|
|
self.assertEqual(instance.decimals, loaded.decimals)
|
2015-07-23 01:51:05 +08:00
|
|
|
self.assertEqual(instance.tags, loaded.tags)
|
|
|
|
|
|
|
|
def test_null_from_db_value_handling(self):
|
|
|
|
instance = OtherTypesArrayModel.objects.create(
|
|
|
|
ips=['192.168.0.1', '::1'],
|
|
|
|
uuids=[uuid.uuid4()],
|
|
|
|
decimals=[decimal.Decimal(1.25), 1.75],
|
|
|
|
tags=None,
|
|
|
|
)
|
|
|
|
instance.refresh_from_db()
|
|
|
|
self.assertIsNone(instance.tags)
|
2015-01-11 02:13:28 +08:00
|
|
|
|
2015-08-05 07:23:05 +08:00
|
|
|
def test_model_set_on_base_field(self):
|
|
|
|
instance = IntegerArrayModel()
|
|
|
|
field = instance._meta.get_field('field')
|
|
|
|
self.assertEqual(field.model, IntegerArrayModel)
|
|
|
|
self.assertEqual(field.base_field.model, IntegerArrayModel)
|
|
|
|
|
2014-03-27 00:44:21 +08:00
|
|
|
|
2015-05-31 05:18:01 +08:00
|
|
|
class TestQuerying(PostgreSQLTestCase):
|
2014-03-27 00:44:21 +08:00
|
|
|
|
|
|
|
def setUp(self):
|
|
|
|
self.objs = [
|
|
|
|
NullableIntegerArrayModel.objects.create(field=[1]),
|
|
|
|
NullableIntegerArrayModel.objects.create(field=[2]),
|
|
|
|
NullableIntegerArrayModel.objects.create(field=[2, 3]),
|
|
|
|
NullableIntegerArrayModel.objects.create(field=[20, 30, 40]),
|
|
|
|
NullableIntegerArrayModel.objects.create(field=None),
|
|
|
|
]
|
|
|
|
|
|
|
|
def test_exact(self):
|
|
|
|
self.assertSequenceEqual(
|
|
|
|
NullableIntegerArrayModel.objects.filter(field__exact=[1]),
|
|
|
|
self.objs[:1]
|
|
|
|
)
|
|
|
|
|
2015-11-07 20:52:40 +08:00
|
|
|
def test_exact_charfield(self):
|
|
|
|
instance = CharArrayModel.objects.create(field=['text'])
|
|
|
|
self.assertSequenceEqual(
|
|
|
|
CharArrayModel.objects.filter(field=['text']),
|
|
|
|
[instance]
|
|
|
|
)
|
|
|
|
|
|
|
|
def test_exact_nested(self):
|
|
|
|
instance = NestedIntegerArrayModel.objects.create(field=[[1, 2], [3, 4]])
|
|
|
|
self.assertSequenceEqual(
|
|
|
|
NestedIntegerArrayModel.objects.filter(field=[[1, 2], [3, 4]]),
|
|
|
|
[instance]
|
|
|
|
)
|
|
|
|
|
2014-03-27 00:44:21 +08:00
|
|
|
def test_isnull(self):
|
|
|
|
self.assertSequenceEqual(
|
|
|
|
NullableIntegerArrayModel.objects.filter(field__isnull=True),
|
|
|
|
self.objs[-1:]
|
|
|
|
)
|
|
|
|
|
|
|
|
def test_gt(self):
|
|
|
|
self.assertSequenceEqual(
|
|
|
|
NullableIntegerArrayModel.objects.filter(field__gt=[0]),
|
|
|
|
self.objs[:4]
|
|
|
|
)
|
|
|
|
|
|
|
|
def test_lt(self):
|
|
|
|
self.assertSequenceEqual(
|
|
|
|
NullableIntegerArrayModel.objects.filter(field__lt=[2]),
|
|
|
|
self.objs[:1]
|
|
|
|
)
|
|
|
|
|
|
|
|
def test_in(self):
|
|
|
|
self.assertSequenceEqual(
|
|
|
|
NullableIntegerArrayModel.objects.filter(field__in=[[1], [2]]),
|
|
|
|
self.objs[:2]
|
|
|
|
)
|
|
|
|
|
|
|
|
def test_contained_by(self):
|
|
|
|
self.assertSequenceEqual(
|
|
|
|
NullableIntegerArrayModel.objects.filter(field__contained_by=[1, 2]),
|
|
|
|
self.objs[:2]
|
|
|
|
)
|
|
|
|
|
|
|
|
def test_contains(self):
|
|
|
|
self.assertSequenceEqual(
|
|
|
|
NullableIntegerArrayModel.objects.filter(field__contains=[2]),
|
|
|
|
self.objs[1:3]
|
|
|
|
)
|
|
|
|
|
2014-07-15 19:13:23 +08:00
|
|
|
def test_contains_charfield(self):
|
|
|
|
# Regression for #22907
|
|
|
|
self.assertSequenceEqual(
|
|
|
|
CharArrayModel.objects.filter(field__contains=['text']),
|
|
|
|
[]
|
|
|
|
)
|
|
|
|
|
2015-01-20 17:52:23 +08:00
|
|
|
def test_contained_by_charfield(self):
|
|
|
|
self.assertSequenceEqual(
|
|
|
|
CharArrayModel.objects.filter(field__contained_by=['text']),
|
|
|
|
[]
|
|
|
|
)
|
|
|
|
|
|
|
|
def test_overlap_charfield(self):
|
|
|
|
self.assertSequenceEqual(
|
|
|
|
CharArrayModel.objects.filter(field__overlap=['text']),
|
|
|
|
[]
|
|
|
|
)
|
|
|
|
|
2014-03-27 00:44:21 +08:00
|
|
|
def test_index(self):
|
|
|
|
self.assertSequenceEqual(
|
|
|
|
NullableIntegerArrayModel.objects.filter(field__0=2),
|
|
|
|
self.objs[1:3]
|
|
|
|
)
|
|
|
|
|
|
|
|
def test_index_chained(self):
|
|
|
|
self.assertSequenceEqual(
|
|
|
|
NullableIntegerArrayModel.objects.filter(field__0__lt=3),
|
|
|
|
self.objs[0:3]
|
|
|
|
)
|
|
|
|
|
|
|
|
def test_index_nested(self):
|
|
|
|
instance = NestedIntegerArrayModel.objects.create(field=[[1, 2], [3, 4]])
|
|
|
|
self.assertSequenceEqual(
|
|
|
|
NestedIntegerArrayModel.objects.filter(field__0__0=1),
|
|
|
|
[instance]
|
|
|
|
)
|
|
|
|
|
|
|
|
@unittest.expectedFailure
|
|
|
|
def test_index_used_on_nested_data(self):
|
|
|
|
instance = NestedIntegerArrayModel.objects.create(field=[[1, 2], [3, 4]])
|
|
|
|
self.assertSequenceEqual(
|
|
|
|
NestedIntegerArrayModel.objects.filter(field__0=[1, 2]),
|
|
|
|
[instance]
|
|
|
|
)
|
|
|
|
|
|
|
|
def test_overlap(self):
|
|
|
|
self.assertSequenceEqual(
|
|
|
|
NullableIntegerArrayModel.objects.filter(field__overlap=[1, 2]),
|
|
|
|
self.objs[0:3]
|
|
|
|
)
|
|
|
|
|
|
|
|
def test_len(self):
|
|
|
|
self.assertSequenceEqual(
|
|
|
|
NullableIntegerArrayModel.objects.filter(field__len__lte=2),
|
|
|
|
self.objs[0:3]
|
|
|
|
)
|
|
|
|
|
2015-11-22 08:59:37 +08:00
|
|
|
def test_len_empty_array(self):
|
|
|
|
obj = NullableIntegerArrayModel.objects.create(field=[])
|
|
|
|
self.assertSequenceEqual(
|
|
|
|
NullableIntegerArrayModel.objects.filter(field__len=0),
|
|
|
|
[obj]
|
|
|
|
)
|
|
|
|
|
2014-03-27 00:44:21 +08:00
|
|
|
def test_slice(self):
|
|
|
|
self.assertSequenceEqual(
|
|
|
|
NullableIntegerArrayModel.objects.filter(field__0_1=[2]),
|
|
|
|
self.objs[1:3]
|
|
|
|
)
|
|
|
|
|
|
|
|
self.assertSequenceEqual(
|
|
|
|
NullableIntegerArrayModel.objects.filter(field__0_2=[2, 3]),
|
|
|
|
self.objs[2:3]
|
|
|
|
)
|
|
|
|
|
|
|
|
@unittest.expectedFailure
|
|
|
|
def test_slice_nested(self):
|
|
|
|
instance = NestedIntegerArrayModel.objects.create(field=[[1, 2], [3, 4]])
|
|
|
|
self.assertSequenceEqual(
|
|
|
|
NestedIntegerArrayModel.objects.filter(field__0__0_1=[1]),
|
|
|
|
[instance]
|
|
|
|
)
|
|
|
|
|
2015-06-04 23:15:25 +08:00
|
|
|
def test_usage_in_subquery(self):
|
|
|
|
self.assertSequenceEqual(
|
|
|
|
NullableIntegerArrayModel.objects.filter(
|
|
|
|
id__in=NullableIntegerArrayModel.objects.filter(field__len=3)
|
|
|
|
),
|
|
|
|
[self.objs[3]]
|
|
|
|
)
|
|
|
|
|
2014-03-27 00:44:21 +08:00
|
|
|
|
2015-11-07 20:52:40 +08:00
|
|
|
class TestDateTimeExactQuerying(PostgreSQLTestCase):
|
|
|
|
|
|
|
|
def setUp(self):
|
|
|
|
now = timezone.now()
|
|
|
|
self.datetimes = [now]
|
|
|
|
self.dates = [now.date()]
|
|
|
|
self.times = [now.time()]
|
|
|
|
self.objs = [
|
|
|
|
DateTimeArrayModel.objects.create(
|
|
|
|
datetimes=self.datetimes,
|
|
|
|
dates=self.dates,
|
|
|
|
times=self.times,
|
|
|
|
)
|
|
|
|
]
|
|
|
|
|
|
|
|
def test_exact_datetimes(self):
|
|
|
|
self.assertSequenceEqual(
|
|
|
|
DateTimeArrayModel.objects.filter(datetimes=self.datetimes),
|
|
|
|
self.objs
|
|
|
|
)
|
|
|
|
|
|
|
|
def test_exact_dates(self):
|
|
|
|
self.assertSequenceEqual(
|
|
|
|
DateTimeArrayModel.objects.filter(dates=self.dates),
|
|
|
|
self.objs
|
|
|
|
)
|
|
|
|
|
|
|
|
def test_exact_times(self):
|
|
|
|
self.assertSequenceEqual(
|
|
|
|
DateTimeArrayModel.objects.filter(times=self.times),
|
|
|
|
self.objs
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
class TestOtherTypesExactQuerying(PostgreSQLTestCase):
|
|
|
|
|
|
|
|
def setUp(self):
|
|
|
|
self.ips = ['192.168.0.1', '::1']
|
|
|
|
self.uuids = [uuid.uuid4()]
|
|
|
|
self.decimals = [decimal.Decimal(1.25), 1.75]
|
2015-07-23 01:51:05 +08:00
|
|
|
self.tags = [Tag(1), Tag(2), Tag(3)]
|
2015-11-07 20:52:40 +08:00
|
|
|
self.objs = [
|
|
|
|
OtherTypesArrayModel.objects.create(
|
|
|
|
ips=self.ips,
|
|
|
|
uuids=self.uuids,
|
|
|
|
decimals=self.decimals,
|
2015-07-23 01:51:05 +08:00
|
|
|
tags=self.tags,
|
2015-11-07 20:52:40 +08:00
|
|
|
)
|
|
|
|
]
|
|
|
|
|
|
|
|
def test_exact_ip_addresses(self):
|
|
|
|
self.assertSequenceEqual(
|
|
|
|
OtherTypesArrayModel.objects.filter(ips=self.ips),
|
|
|
|
self.objs
|
|
|
|
)
|
|
|
|
|
|
|
|
def test_exact_uuids(self):
|
|
|
|
self.assertSequenceEqual(
|
|
|
|
OtherTypesArrayModel.objects.filter(uuids=self.uuids),
|
|
|
|
self.objs
|
|
|
|
)
|
|
|
|
|
|
|
|
def test_exact_decimals(self):
|
|
|
|
self.assertSequenceEqual(
|
|
|
|
OtherTypesArrayModel.objects.filter(decimals=self.decimals),
|
|
|
|
self.objs
|
|
|
|
)
|
|
|
|
|
2015-07-23 01:51:05 +08:00
|
|
|
def test_exact_tags(self):
|
|
|
|
self.assertSequenceEqual(
|
|
|
|
OtherTypesArrayModel.objects.filter(tags=self.tags),
|
|
|
|
self.objs
|
|
|
|
)
|
|
|
|
|
2015-11-07 20:52:40 +08:00
|
|
|
|
2015-11-17 13:39:28 +08:00
|
|
|
@isolate_apps('postgres_tests')
|
2015-05-31 05:18:01 +08:00
|
|
|
class TestChecks(PostgreSQLTestCase):
|
2014-03-27 00:44:21 +08:00
|
|
|
|
|
|
|
def test_field_checks(self):
|
2015-08-01 22:46:25 +08:00
|
|
|
class MyModel(PostgreSQLModel):
|
|
|
|
field = ArrayField(models.CharField())
|
|
|
|
|
|
|
|
model = MyModel()
|
|
|
|
errors = model.check()
|
2014-03-27 00:44:21 +08:00
|
|
|
self.assertEqual(len(errors), 1)
|
2015-12-05 02:14:12 +08:00
|
|
|
# The inner CharField is missing a max_length.
|
2014-03-27 00:44:21 +08:00
|
|
|
self.assertEqual(errors[0].id, 'postgres.E001')
|
2015-12-05 02:14:12 +08:00
|
|
|
self.assertIn('max_length', errors[0].msg)
|
2014-03-27 00:44:21 +08:00
|
|
|
|
|
|
|
def test_invalid_base_fields(self):
|
2015-08-01 22:46:25 +08:00
|
|
|
class MyModel(PostgreSQLModel):
|
|
|
|
field = ArrayField(models.ManyToManyField('postgres_tests.IntegerArrayModel'))
|
|
|
|
|
|
|
|
model = MyModel()
|
|
|
|
errors = model.check()
|
2014-03-27 00:44:21 +08:00
|
|
|
self.assertEqual(len(errors), 1)
|
|
|
|
self.assertEqual(errors[0].id, 'postgres.E002')
|
|
|
|
|
2015-12-05 02:14:12 +08:00
|
|
|
def test_nested_field_checks(self):
|
|
|
|
"""
|
|
|
|
Nested ArrayFields are permitted.
|
|
|
|
"""
|
|
|
|
class MyModel(PostgreSQLModel):
|
|
|
|
field = ArrayField(ArrayField(models.CharField()))
|
|
|
|
|
|
|
|
model = MyModel()
|
|
|
|
errors = model.check()
|
|
|
|
self.assertEqual(len(errors), 1)
|
|
|
|
# The inner CharField is missing a max_length.
|
|
|
|
self.assertEqual(errors[0].id, 'postgres.E001')
|
|
|
|
self.assertIn('max_length', errors[0].msg)
|
|
|
|
|
2014-03-27 00:44:21 +08:00
|
|
|
|
2015-04-05 00:10:26 +08:00
|
|
|
@unittest.skipUnless(connection.vendor == 'postgresql', "PostgreSQL specific tests")
|
2015-01-22 11:26:37 +08:00
|
|
|
class TestMigrations(TransactionTestCase):
|
|
|
|
|
|
|
|
available_apps = ['postgres_tests']
|
2014-03-27 00:44:21 +08:00
|
|
|
|
|
|
|
def test_deconstruct(self):
|
|
|
|
field = ArrayField(models.IntegerField())
|
|
|
|
name, path, args, kwargs = field.deconstruct()
|
|
|
|
new = ArrayField(*args, **kwargs)
|
|
|
|
self.assertEqual(type(new.base_field), type(field.base_field))
|
|
|
|
|
|
|
|
def test_deconstruct_with_size(self):
|
|
|
|
field = ArrayField(models.IntegerField(), size=3)
|
|
|
|
name, path, args, kwargs = field.deconstruct()
|
|
|
|
new = ArrayField(*args, **kwargs)
|
|
|
|
self.assertEqual(new.size, field.size)
|
|
|
|
|
|
|
|
def test_deconstruct_args(self):
|
|
|
|
field = ArrayField(models.CharField(max_length=20))
|
|
|
|
name, path, args, kwargs = field.deconstruct()
|
|
|
|
new = ArrayField(*args, **kwargs)
|
|
|
|
self.assertEqual(new.base_field.max_length, field.base_field.max_length)
|
|
|
|
|
2014-12-21 18:21:37 +08:00
|
|
|
def test_subclass_deconstruct(self):
|
|
|
|
field = ArrayField(models.IntegerField())
|
|
|
|
name, path, args, kwargs = field.deconstruct()
|
|
|
|
self.assertEqual(path, 'django.contrib.postgres.fields.ArrayField')
|
|
|
|
|
|
|
|
field = ArrayFieldSubclass()
|
|
|
|
name, path, args, kwargs = field.deconstruct()
|
|
|
|
self.assertEqual(path, 'postgres_tests.models.ArrayFieldSubclass')
|
|
|
|
|
2014-07-15 19:30:34 +08:00
|
|
|
@override_settings(MIGRATION_MODULES={
|
|
|
|
"postgres_tests": "postgres_tests.array_default_migrations",
|
|
|
|
})
|
|
|
|
def test_adding_field_with_default(self):
|
|
|
|
# See #22962
|
2015-01-22 11:26:37 +08:00
|
|
|
table_name = 'postgres_tests_integerarraydefaultmodel'
|
|
|
|
with connection.cursor() as cursor:
|
|
|
|
self.assertNotIn(table_name, connection.introspection.table_names(cursor))
|
2014-07-15 19:30:34 +08:00
|
|
|
call_command('migrate', 'postgres_tests', verbosity=0)
|
2015-01-22 11:26:37 +08:00
|
|
|
with connection.cursor() as cursor:
|
|
|
|
self.assertIn(table_name, connection.introspection.table_names(cursor))
|
|
|
|
call_command('migrate', 'postgres_tests', 'zero', verbosity=0)
|
|
|
|
with connection.cursor() as cursor:
|
|
|
|
self.assertNotIn(table_name, connection.introspection.table_names(cursor))
|
2014-07-15 19:30:34 +08:00
|
|
|
|
2015-08-08 20:44:27 +08:00
|
|
|
@override_settings(MIGRATION_MODULES={
|
|
|
|
"postgres_tests": "postgres_tests.array_index_migrations",
|
|
|
|
})
|
|
|
|
def test_adding_arrayfield_with_index(self):
|
|
|
|
"""
|
|
|
|
ArrayField shouldn't have varchar_patterns_ops or text_patterns_ops indexes.
|
|
|
|
"""
|
|
|
|
table_name = 'postgres_tests_chartextarrayindexmodel'
|
|
|
|
call_command('migrate', 'postgres_tests', verbosity=0)
|
|
|
|
with connection.cursor() as cursor:
|
2016-07-08 22:11:19 +08:00
|
|
|
like_constraint_columns_list = [
|
|
|
|
v['columns']
|
|
|
|
for k, v in list(connection.introspection.get_constraints(cursor, table_name).items())
|
|
|
|
if k.endswith('_like')
|
2015-08-08 20:44:27 +08:00
|
|
|
]
|
|
|
|
# Only the CharField should have a LIKE index.
|
2016-07-08 22:11:19 +08:00
|
|
|
self.assertEqual(like_constraint_columns_list, [['char2']])
|
2015-08-08 20:44:27 +08:00
|
|
|
with connection.cursor() as cursor:
|
|
|
|
indexes = connection.introspection.get_indexes(cursor, table_name)
|
|
|
|
# All fields should have regular indexes.
|
|
|
|
self.assertIn('char', indexes)
|
|
|
|
self.assertIn('char2', indexes)
|
|
|
|
self.assertIn('text', indexes)
|
|
|
|
call_command('migrate', 'postgres_tests', 'zero', verbosity=0)
|
|
|
|
with connection.cursor() as cursor:
|
|
|
|
self.assertNotIn(table_name, connection.introspection.table_names(cursor))
|
|
|
|
|
2014-03-27 00:44:21 +08:00
|
|
|
|
2015-05-31 05:18:01 +08:00
|
|
|
class TestSerialization(PostgreSQLTestCase):
|
2015-09-12 07:33:12 +08:00
|
|
|
test_data = (
|
2016-02-16 02:28:49 +08:00
|
|
|
'[{"fields": {"field": "[\\"1\\", \\"2\\", null]"}, "model": "postgres_tests.integerarraymodel", "pk": null}]'
|
2015-09-12 07:33:12 +08:00
|
|
|
)
|
2014-03-27 00:44:21 +08:00
|
|
|
|
|
|
|
def test_dumping(self):
|
2016-02-16 02:28:49 +08:00
|
|
|
instance = IntegerArrayModel(field=[1, 2, None])
|
2014-03-27 00:44:21 +08:00
|
|
|
data = serializers.serialize('json', [instance])
|
2014-05-22 22:05:19 +08:00
|
|
|
self.assertEqual(json.loads(data), json.loads(self.test_data))
|
2014-03-27 00:44:21 +08:00
|
|
|
|
|
|
|
def test_loading(self):
|
|
|
|
instance = list(serializers.deserialize('json', self.test_data))[0].object
|
2016-02-16 02:28:49 +08:00
|
|
|
self.assertEqual(instance.field, [1, 2, None])
|
2014-03-27 00:44:21 +08:00
|
|
|
|
|
|
|
|
2015-05-31 05:18:01 +08:00
|
|
|
class TestValidation(PostgreSQLTestCase):
|
2014-03-27 00:44:21 +08:00
|
|
|
|
|
|
|
def test_unbounded(self):
|
|
|
|
field = ArrayField(models.IntegerField())
|
|
|
|
with self.assertRaises(exceptions.ValidationError) as cm:
|
|
|
|
field.clean([1, None], None)
|
|
|
|
self.assertEqual(cm.exception.code, 'item_invalid')
|
2015-09-12 07:33:12 +08:00
|
|
|
self.assertEqual(
|
|
|
|
cm.exception.message % cm.exception.params,
|
|
|
|
'Item 1 in the array did not validate: This field cannot be null.'
|
|
|
|
)
|
2014-03-27 00:44:21 +08:00
|
|
|
|
|
|
|
def test_blank_true(self):
|
|
|
|
field = ArrayField(models.IntegerField(blank=True, null=True))
|
|
|
|
# This should not raise a validation error
|
|
|
|
field.clean([1, None], None)
|
|
|
|
|
|
|
|
def test_with_size(self):
|
|
|
|
field = ArrayField(models.IntegerField(), size=3)
|
|
|
|
field.clean([1, 2, 3], None)
|
|
|
|
with self.assertRaises(exceptions.ValidationError) as cm:
|
|
|
|
field.clean([1, 2, 3, 4], None)
|
|
|
|
self.assertEqual(cm.exception.messages[0], 'List contains 4 items, it should contain no more than 3.')
|
|
|
|
|
|
|
|
def test_nested_array_mismatch(self):
|
|
|
|
field = ArrayField(ArrayField(models.IntegerField()))
|
|
|
|
field.clean([[1, 2], [3, 4]], None)
|
|
|
|
with self.assertRaises(exceptions.ValidationError) as cm:
|
|
|
|
field.clean([[1, 2], [3, 4, 5]], None)
|
|
|
|
self.assertEqual(cm.exception.code, 'nested_array_mismatch')
|
|
|
|
self.assertEqual(cm.exception.messages[0], 'Nested arrays must have the same length.')
|
|
|
|
|
2015-12-02 06:59:58 +08:00
|
|
|
def test_with_base_field_error_params(self):
|
|
|
|
field = ArrayField(models.CharField(max_length=2))
|
|
|
|
with self.assertRaises(exceptions.ValidationError) as cm:
|
|
|
|
field.clean(['abc'], None)
|
|
|
|
self.assertEqual(len(cm.exception.error_list), 1)
|
|
|
|
exception = cm.exception.error_list[0]
|
|
|
|
self.assertEqual(
|
|
|
|
exception.message,
|
|
|
|
'Item 0 in the array did not validate: Ensure this value has at most 2 characters (it has 3).'
|
|
|
|
)
|
|
|
|
self.assertEqual(exception.code, 'item_invalid')
|
|
|
|
self.assertEqual(exception.params, {'nth': 0, 'value': 'abc', 'limit_value': 2, 'show_value': 3})
|
|
|
|
|
2015-02-20 19:33:02 +08:00
|
|
|
def test_with_validators(self):
|
|
|
|
field = ArrayField(models.IntegerField(validators=[validators.MinValueValidator(1)]))
|
|
|
|
field.clean([1, 2], None)
|
|
|
|
with self.assertRaises(exceptions.ValidationError) as cm:
|
|
|
|
field.clean([0], None)
|
2015-12-02 06:59:58 +08:00
|
|
|
self.assertEqual(len(cm.exception.error_list), 1)
|
|
|
|
exception = cm.exception.error_list[0]
|
2015-09-12 07:33:12 +08:00
|
|
|
self.assertEqual(
|
2015-12-02 06:59:58 +08:00
|
|
|
exception.message,
|
2015-09-12 07:33:12 +08:00
|
|
|
'Item 0 in the array did not validate: Ensure this value is greater than or equal to 1.'
|
|
|
|
)
|
2015-12-02 06:59:58 +08:00
|
|
|
self.assertEqual(exception.code, 'item_invalid')
|
|
|
|
self.assertEqual(exception.params, {'nth': 0, 'value': 0, 'limit_value': 1, 'show_value': 0})
|
2015-02-20 19:33:02 +08:00
|
|
|
|
2014-03-27 00:44:21 +08:00
|
|
|
|
2015-05-31 05:18:01 +08:00
|
|
|
class TestSimpleFormField(PostgreSQLTestCase):
|
2014-03-27 00:44:21 +08:00
|
|
|
|
|
|
|
def test_valid(self):
|
|
|
|
field = SimpleArrayField(forms.CharField())
|
|
|
|
value = field.clean('a,b,c')
|
|
|
|
self.assertEqual(value, ['a', 'b', 'c'])
|
|
|
|
|
|
|
|
def test_to_python_fail(self):
|
|
|
|
field = SimpleArrayField(forms.IntegerField())
|
|
|
|
with self.assertRaises(exceptions.ValidationError) as cm:
|
|
|
|
field.clean('a,b,9')
|
|
|
|
self.assertEqual(cm.exception.messages[0], 'Item 0 in the array did not validate: Enter a whole number.')
|
|
|
|
|
|
|
|
def test_validate_fail(self):
|
|
|
|
field = SimpleArrayField(forms.CharField(required=True))
|
|
|
|
with self.assertRaises(exceptions.ValidationError) as cm:
|
|
|
|
field.clean('a,b,')
|
|
|
|
self.assertEqual(cm.exception.messages[0], 'Item 2 in the array did not validate: This field is required.')
|
|
|
|
|
2015-12-02 06:59:58 +08:00
|
|
|
def test_validate_fail_base_field_error_params(self):
|
|
|
|
field = SimpleArrayField(forms.CharField(max_length=2))
|
|
|
|
with self.assertRaises(exceptions.ValidationError) as cm:
|
|
|
|
field.clean('abc,c,defg')
|
|
|
|
errors = cm.exception.error_list
|
|
|
|
self.assertEqual(len(errors), 2)
|
|
|
|
first_error = errors[0]
|
|
|
|
self.assertEqual(
|
|
|
|
first_error.message,
|
|
|
|
'Item 0 in the array did not validate: Ensure this value has at most 2 characters (it has 3).'
|
|
|
|
)
|
|
|
|
self.assertEqual(first_error.code, 'item_invalid')
|
|
|
|
self.assertEqual(first_error.params, {'nth': 0, 'value': 'abc', 'limit_value': 2, 'show_value': 3})
|
|
|
|
second_error = errors[1]
|
|
|
|
self.assertEqual(
|
|
|
|
second_error.message,
|
|
|
|
'Item 2 in the array did not validate: Ensure this value has at most 2 characters (it has 4).'
|
|
|
|
)
|
|
|
|
self.assertEqual(second_error.code, 'item_invalid')
|
|
|
|
self.assertEqual(second_error.params, {'nth': 2, 'value': 'defg', 'limit_value': 2, 'show_value': 4})
|
|
|
|
|
2014-03-27 00:44:21 +08:00
|
|
|
def test_validators_fail(self):
|
|
|
|
field = SimpleArrayField(forms.RegexField('[a-e]{2}'))
|
|
|
|
with self.assertRaises(exceptions.ValidationError) as cm:
|
|
|
|
field.clean('a,bc,de')
|
|
|
|
self.assertEqual(cm.exception.messages[0], 'Item 0 in the array did not validate: Enter a valid value.')
|
|
|
|
|
|
|
|
def test_delimiter(self):
|
|
|
|
field = SimpleArrayField(forms.CharField(), delimiter='|')
|
|
|
|
value = field.clean('a|b|c')
|
|
|
|
self.assertEqual(value, ['a', 'b', 'c'])
|
|
|
|
|
|
|
|
def test_delimiter_with_nesting(self):
|
|
|
|
field = SimpleArrayField(SimpleArrayField(forms.CharField()), delimiter='|')
|
|
|
|
value = field.clean('a,b|c,d')
|
|
|
|
self.assertEqual(value, [['a', 'b'], ['c', 'd']])
|
|
|
|
|
|
|
|
def test_prepare_value(self):
|
|
|
|
field = SimpleArrayField(forms.CharField())
|
|
|
|
value = field.prepare_value(['a', 'b', 'c'])
|
|
|
|
self.assertEqual(value, 'a,b,c')
|
|
|
|
|
|
|
|
def test_max_length(self):
|
|
|
|
field = SimpleArrayField(forms.CharField(), max_length=2)
|
|
|
|
with self.assertRaises(exceptions.ValidationError) as cm:
|
|
|
|
field.clean('a,b,c')
|
|
|
|
self.assertEqual(cm.exception.messages[0], 'List contains 3 items, it should contain no more than 2.')
|
|
|
|
|
|
|
|
def test_min_length(self):
|
|
|
|
field = SimpleArrayField(forms.CharField(), min_length=4)
|
|
|
|
with self.assertRaises(exceptions.ValidationError) as cm:
|
|
|
|
field.clean('a,b,c')
|
|
|
|
self.assertEqual(cm.exception.messages[0], 'List contains 3 items, it should contain no fewer than 4.')
|
|
|
|
|
|
|
|
def test_required(self):
|
|
|
|
field = SimpleArrayField(forms.CharField(), required=True)
|
|
|
|
with self.assertRaises(exceptions.ValidationError) as cm:
|
|
|
|
field.clean('')
|
|
|
|
self.assertEqual(cm.exception.messages[0], 'This field is required.')
|
|
|
|
|
|
|
|
def test_model_field_formfield(self):
|
|
|
|
model_field = ArrayField(models.CharField(max_length=27))
|
|
|
|
form_field = model_field.formfield()
|
|
|
|
self.assertIsInstance(form_field, SimpleArrayField)
|
|
|
|
self.assertIsInstance(form_field.base_field, forms.CharField)
|
|
|
|
self.assertEqual(form_field.base_field.max_length, 27)
|
|
|
|
|
|
|
|
def test_model_field_formfield_size(self):
|
|
|
|
model_field = ArrayField(models.CharField(max_length=27), size=4)
|
|
|
|
form_field = model_field.formfield()
|
|
|
|
self.assertIsInstance(form_field, SimpleArrayField)
|
|
|
|
self.assertEqual(form_field.max_length, 4)
|
|
|
|
|
|
|
|
|
2015-05-31 05:18:01 +08:00
|
|
|
class TestSplitFormField(PostgreSQLTestCase):
|
2014-03-27 00:44:21 +08:00
|
|
|
|
|
|
|
def test_valid(self):
|
|
|
|
class SplitForm(forms.Form):
|
|
|
|
array = SplitArrayField(forms.CharField(), size=3)
|
|
|
|
|
|
|
|
data = {'array_0': 'a', 'array_1': 'b', 'array_2': 'c'}
|
|
|
|
form = SplitForm(data)
|
|
|
|
self.assertTrue(form.is_valid())
|
|
|
|
self.assertEqual(form.cleaned_data, {'array': ['a', 'b', 'c']})
|
|
|
|
|
|
|
|
def test_required(self):
|
|
|
|
class SplitForm(forms.Form):
|
|
|
|
array = SplitArrayField(forms.CharField(), required=True, size=3)
|
|
|
|
|
|
|
|
data = {'array_0': '', 'array_1': '', 'array_2': ''}
|
|
|
|
form = SplitForm(data)
|
|
|
|
self.assertFalse(form.is_valid())
|
|
|
|
self.assertEqual(form.errors, {'array': ['This field is required.']})
|
|
|
|
|
|
|
|
def test_remove_trailing_nulls(self):
|
|
|
|
class SplitForm(forms.Form):
|
|
|
|
array = SplitArrayField(forms.CharField(required=False), size=5, remove_trailing_nulls=True)
|
|
|
|
|
|
|
|
data = {'array_0': 'a', 'array_1': '', 'array_2': 'b', 'array_3': '', 'array_4': ''}
|
|
|
|
form = SplitForm(data)
|
|
|
|
self.assertTrue(form.is_valid(), form.errors)
|
|
|
|
self.assertEqual(form.cleaned_data, {'array': ['a', '', 'b']})
|
|
|
|
|
2016-03-13 05:44:39 +08:00
|
|
|
def test_remove_trailing_nulls_not_required(self):
|
|
|
|
class SplitForm(forms.Form):
|
|
|
|
array = SplitArrayField(
|
|
|
|
forms.CharField(required=False),
|
|
|
|
size=2,
|
|
|
|
remove_trailing_nulls=True,
|
|
|
|
required=False,
|
|
|
|
)
|
|
|
|
|
|
|
|
data = {'array_0': '', 'array_1': ''}
|
|
|
|
form = SplitForm(data)
|
|
|
|
self.assertTrue(form.is_valid())
|
|
|
|
self.assertEqual(form.cleaned_data, {'array': []})
|
|
|
|
|
2014-03-27 00:44:21 +08:00
|
|
|
def test_required_field(self):
|
|
|
|
class SplitForm(forms.Form):
|
|
|
|
array = SplitArrayField(forms.CharField(), size=3)
|
|
|
|
|
|
|
|
data = {'array_0': 'a', 'array_1': 'b', 'array_2': ''}
|
|
|
|
form = SplitForm(data)
|
|
|
|
self.assertFalse(form.is_valid())
|
|
|
|
self.assertEqual(form.errors, {'array': ['Item 2 in the array did not validate: This field is required.']})
|
|
|
|
|
2015-10-27 07:37:47 +08:00
|
|
|
def test_invalid_integer(self):
|
|
|
|
msg = 'Item 1 in the array did not validate: Ensure this value is less than or equal to 100.'
|
|
|
|
with self.assertRaisesMessage(exceptions.ValidationError, msg):
|
|
|
|
SplitArrayField(forms.IntegerField(max_value=100), size=2).clean([0, 101])
|
|
|
|
|
2014-03-27 00:44:21 +08:00
|
|
|
def test_rendering(self):
|
|
|
|
class SplitForm(forms.Form):
|
|
|
|
array = SplitArrayField(forms.CharField(), size=3)
|
|
|
|
|
|
|
|
self.assertHTMLEqual(str(SplitForm()), '''
|
|
|
|
<tr>
|
|
|
|
<th><label for="id_array_0">Array:</label></th>
|
|
|
|
<td>
|
2016-03-29 02:02:04 +08:00
|
|
|
<input id="id_array_0" name="array_0" type="text" required />
|
|
|
|
<input id="id_array_1" name="array_1" type="text" required />
|
|
|
|
<input id="id_array_2" name="array_2" type="text" required />
|
2014-03-27 00:44:21 +08:00
|
|
|
</td>
|
|
|
|
</tr>
|
|
|
|
''')
|
2015-12-02 06:59:58 +08:00
|
|
|
|
|
|
|
def test_invalid_char_length(self):
|
|
|
|
field = SplitArrayField(forms.CharField(max_length=2), size=3)
|
|
|
|
with self.assertRaises(exceptions.ValidationError) as cm:
|
|
|
|
field.clean(['abc', 'c', 'defg'])
|
|
|
|
self.assertEqual(cm.exception.messages, [
|
|
|
|
'Item 0 in the array did not validate: Ensure this value has at most 2 characters (it has 3).',
|
|
|
|
'Item 2 in the array did not validate: Ensure this value has at most 2 characters (it has 4).',
|
|
|
|
])
|