mirror of
https://github.com/django/django.git
synced 2025-10-24 14:16:09 +00:00
Fixed #34944 -- Made GeneratedField.output_field required.
Regression in f333e3513e.
This commit is contained in:
@@ -1,5 +1,12 @@
|
||||
from django.db import IntegrityError, connection
|
||||
from django.db.models import F, FloatField, GeneratedField, IntegerField, Model
|
||||
from django.db.models import (
|
||||
CharField,
|
||||
F,
|
||||
FloatField,
|
||||
GeneratedField,
|
||||
IntegerField,
|
||||
Model,
|
||||
)
|
||||
from django.db.models.functions import Lower
|
||||
from django.test import SimpleTestCase, TestCase, skipUnlessDBFeature
|
||||
from django.test.utils import isolate_apps
|
||||
@@ -8,8 +15,8 @@ from .models import (
|
||||
GeneratedModel,
|
||||
GeneratedModelNull,
|
||||
GeneratedModelNullVirtual,
|
||||
GeneratedModelOutputField,
|
||||
GeneratedModelOutputFieldVirtual,
|
||||
GeneratedModelOutputFieldDbCollation,
|
||||
GeneratedModelOutputFieldDbCollationVirtual,
|
||||
GeneratedModelParams,
|
||||
GeneratedModelParamsVirtual,
|
||||
GeneratedModelVirtual,
|
||||
@@ -19,41 +26,77 @@ from .models import (
|
||||
class BaseGeneratedFieldTests(SimpleTestCase):
|
||||
def test_editable_unsupported(self):
|
||||
with self.assertRaisesMessage(ValueError, "GeneratedField cannot be editable."):
|
||||
GeneratedField(expression=Lower("name"), editable=True, db_persist=False)
|
||||
GeneratedField(
|
||||
expression=Lower("name"),
|
||||
output_field=CharField(max_length=255),
|
||||
editable=True,
|
||||
db_persist=False,
|
||||
)
|
||||
|
||||
def test_blank_unsupported(self):
|
||||
with self.assertRaisesMessage(ValueError, "GeneratedField must be blank."):
|
||||
GeneratedField(expression=Lower("name"), blank=False, db_persist=False)
|
||||
GeneratedField(
|
||||
expression=Lower("name"),
|
||||
output_field=CharField(max_length=255),
|
||||
blank=False,
|
||||
db_persist=False,
|
||||
)
|
||||
|
||||
def test_default_unsupported(self):
|
||||
msg = "GeneratedField cannot have a default."
|
||||
with self.assertRaisesMessage(ValueError, msg):
|
||||
GeneratedField(expression=Lower("name"), default="", db_persist=False)
|
||||
GeneratedField(
|
||||
expression=Lower("name"),
|
||||
output_field=CharField(max_length=255),
|
||||
default="",
|
||||
db_persist=False,
|
||||
)
|
||||
|
||||
def test_database_default_unsupported(self):
|
||||
msg = "GeneratedField cannot have a database default."
|
||||
with self.assertRaisesMessage(ValueError, msg):
|
||||
GeneratedField(expression=Lower("name"), db_default="", db_persist=False)
|
||||
GeneratedField(
|
||||
expression=Lower("name"),
|
||||
output_field=CharField(max_length=255),
|
||||
db_default="",
|
||||
db_persist=False,
|
||||
)
|
||||
|
||||
def test_db_persist_required(self):
|
||||
msg = "GeneratedField.db_persist must be True or False."
|
||||
with self.assertRaisesMessage(ValueError, msg):
|
||||
GeneratedField(expression=Lower("name"))
|
||||
GeneratedField(
|
||||
expression=Lower("name"), output_field=CharField(max_length=255)
|
||||
)
|
||||
with self.assertRaisesMessage(ValueError, msg):
|
||||
GeneratedField(expression=Lower("name"), db_persist=None)
|
||||
GeneratedField(
|
||||
expression=Lower("name"),
|
||||
output_field=CharField(max_length=255),
|
||||
db_persist=None,
|
||||
)
|
||||
|
||||
def test_deconstruct(self):
|
||||
field = GeneratedField(expression=F("a") + F("b"), db_persist=True)
|
||||
field = GeneratedField(
|
||||
expression=F("a") + F("b"), output_field=IntegerField(), db_persist=True
|
||||
)
|
||||
_, path, args, kwargs = field.deconstruct()
|
||||
self.assertEqual(path, "django.db.models.GeneratedField")
|
||||
self.assertEqual(args, [])
|
||||
self.assertEqual(kwargs, {"db_persist": True, "expression": F("a") + F("b")})
|
||||
self.assertEqual(kwargs["db_persist"], True)
|
||||
self.assertEqual(kwargs["expression"], F("a") + F("b"))
|
||||
self.assertEqual(
|
||||
kwargs["output_field"].deconstruct(), IntegerField().deconstruct()
|
||||
)
|
||||
|
||||
@isolate_apps("model_fields")
|
||||
def test_get_col(self):
|
||||
class Square(Model):
|
||||
side = IntegerField()
|
||||
area = GeneratedField(expression=F("side") * F("side"), db_persist=True)
|
||||
area = GeneratedField(
|
||||
expression=F("side") * F("side"),
|
||||
output_field=IntegerField(),
|
||||
db_persist=True,
|
||||
)
|
||||
|
||||
col = Square._meta.get_field("area").get_col("alias")
|
||||
self.assertIsInstance(col.output_field, IntegerField)
|
||||
@@ -74,7 +117,9 @@ class BaseGeneratedFieldTests(SimpleTestCase):
|
||||
class Sum(Model):
|
||||
a = IntegerField()
|
||||
b = IntegerField()
|
||||
total = GeneratedField(expression=F("a") + F("b"), db_persist=True)
|
||||
total = GeneratedField(
|
||||
expression=F("a") + F("b"), output_field=IntegerField(), db_persist=True
|
||||
)
|
||||
|
||||
field = Sum._meta.get_field("total")
|
||||
cached_col = field.cached_col
|
||||
@@ -165,9 +210,9 @@ class GeneratedFieldTestMixin:
|
||||
with self.assertNumQueries(0), self.assertRaises(does_not_exist):
|
||||
self.base_model.objects.get(field__gte=overflow_value)
|
||||
|
||||
def test_output_field(self):
|
||||
def test_output_field_db_collation(self):
|
||||
collation = connection.features.test_collations["virtual"]
|
||||
m = self.output_field_model.objects.create(name="NAME")
|
||||
m = self.output_field_db_collation_model.objects.create(name="NAME")
|
||||
field = m._meta.get_field("lower_name")
|
||||
db_parameters = field.db_parameters(connection)
|
||||
self.assertEqual(db_parameters["collation"], collation)
|
||||
@@ -178,7 +223,7 @@ class GeneratedFieldTestMixin:
|
||||
)
|
||||
|
||||
def test_db_type_parameters(self):
|
||||
db_type_parameters = self.output_field_model._meta.get_field(
|
||||
db_type_parameters = self.output_field_db_collation_model._meta.get_field(
|
||||
"lower_name"
|
||||
).db_type_parameters(connection)
|
||||
self.assertEqual(db_type_parameters["max_length"], 11)
|
||||
@@ -202,7 +247,7 @@ class GeneratedFieldTestMixin:
|
||||
class StoredGeneratedFieldTests(GeneratedFieldTestMixin, TestCase):
|
||||
base_model = GeneratedModel
|
||||
nullable_model = GeneratedModelNull
|
||||
output_field_model = GeneratedModelOutputField
|
||||
output_field_db_collation_model = GeneratedModelOutputFieldDbCollation
|
||||
params_model = GeneratedModelParams
|
||||
|
||||
|
||||
@@ -210,5 +255,5 @@ class StoredGeneratedFieldTests(GeneratedFieldTestMixin, TestCase):
|
||||
class VirtualGeneratedFieldTests(GeneratedFieldTestMixin, TestCase):
|
||||
base_model = GeneratedModelVirtual
|
||||
nullable_model = GeneratedModelNullVirtual
|
||||
output_field_model = GeneratedModelOutputFieldVirtual
|
||||
output_field_db_collation_model = GeneratedModelOutputFieldDbCollationVirtual
|
||||
params_model = GeneratedModelParamsVirtual
|
||||
|
||||
Reference in New Issue
Block a user