diff options
| -rw-r--r-- | tests/model_fields/test_jsonfield.py | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/tests/model_fields/test_jsonfield.py b/tests/model_fields/test_jsonfield.py index d0561e14d2..76cdc0e421 100644 --- a/tests/model_fields/test_jsonfield.py +++ b/tests/model_fields/test_jsonfield.py @@ -364,14 +364,14 @@ class TestQuerying(TestCase): "bax": {"foo": "bar"}, }, ] - cls.objs = [NullableJSONModel.objects.create(value=value) for value in values] + objs = [NullableJSONModel(value=value) for value in values] if connection.features.supports_primitives_in_json_field: - cls.objs.extend( - [ - NullableJSONModel.objects.create(value=value) - for value in cls.primitives - ] - ) + objs.extend([NullableJSONModel(value=value) for value in cls.primitives]) + objs = NullableJSONModel.objects.bulk_create(objs) + # Some backends don't return primary keys after bulk_create. + if any(obj.pk is None for obj in objs): + objs = list(NullableJSONModel.objects.order_by("id")) + cls.objs = objs cls.raw_sql = "%s::jsonb" if connection.vendor == "postgresql" else "%s" def test_exact(self): |