summaryrefslogtreecommitdiff
path: root/tests/model_fields
diff options
context:
space:
mode:
Diffstat (limited to 'tests/model_fields')
-rw-r--r--tests/model_fields/test_jsonfield.py16
1 files changed, 15 insertions, 1 deletions
diff --git a/tests/model_fields/test_jsonfield.py b/tests/model_fields/test_jsonfield.py
index a590ac1221..f45589e075 100644
--- a/tests/model_fields/test_jsonfield.py
+++ b/tests/model_fields/test_jsonfield.py
@@ -10,7 +10,10 @@ from django.db import (
DataError, IntegrityError, NotSupportedError, OperationalError, connection,
models,
)
-from django.db.models import Count, F, OuterRef, Q, Subquery, Transform, Value
+from django.db.models import (
+ Count, ExpressionWrapper, F, IntegerField, OuterRef, Q, Subquery,
+ Transform, Value,
+)
from django.db.models.expressions import RawSQL
from django.db.models.fields.json import (
KeyTextTransform, KeyTransform, KeyTransformFactory,
@@ -405,6 +408,17 @@ class TestQuerying(TestCase):
[self.objs[4]],
)
+ def test_expression_wrapper_key_transform(self):
+ self.assertSequenceEqual(
+ NullableJSONModel.objects.annotate(
+ expr=ExpressionWrapper(
+ KeyTransform('c', 'value'),
+ output_field=IntegerField(),
+ ),
+ ).filter(expr__isnull=False),
+ self.objs[3:5],
+ )
+
def test_has_key(self):
self.assertSequenceEqual(
NullableJSONModel.objects.filter(value__has_key='a'),