summaryrefslogtreecommitdiff
path: root/tests/postgres_tests
diff options
context:
space:
mode:
Diffstat (limited to 'tests/postgres_tests')
-rw-r--r--tests/postgres_tests/fields.py6
-rw-r--r--tests/postgres_tests/migrations/0002_create_test_models.py2
-rw-r--r--tests/postgres_tests/models.py2
-rw-r--r--tests/postgres_tests/test_json.py37
4 files changed, 45 insertions, 2 deletions
diff --git a/tests/postgres_tests/fields.py b/tests/postgres_tests/fields.py
index d50c6d6a91..c02bc2faf4 100644
--- a/tests/postgres_tests/fields.py
+++ b/tests/postgres_tests/fields.py
@@ -23,6 +23,10 @@ except ImportError:
})
return name, path, args, kwargs
+ class DummyJSONField(models.Field):
+ def __init__(self, encoder=None, **kwargs):
+ super(DummyJSONField, self).__init__(**kwargs)
+
ArrayField = DummyArrayField
BigIntegerRangeField = models.Field
DateRangeField = models.Field
@@ -30,5 +34,5 @@ except ImportError:
FloatRangeField = models.Field
HStoreField = models.Field
IntegerRangeField = models.Field
- JSONField = models.Field
+ JSONField = DummyJSONField
SearchVectorField = models.Field
diff --git a/tests/postgres_tests/migrations/0002_create_test_models.py b/tests/postgres_tests/migrations/0002_create_test_models.py
index 69c5b17f77..974aae1ec8 100644
--- a/tests/postgres_tests/migrations/0002_create_test_models.py
+++ b/tests/postgres_tests/migrations/0002_create_test_models.py
@@ -1,6 +1,7 @@
# -*- coding: utf-8 -*-
from __future__ import unicode_literals
+from django.core.serializers.json import DjangoJSONEncoder
from django.db import migrations, models
from ..fields import (
@@ -223,6 +224,7 @@ class Migration(migrations.Migration):
fields=[
('id', models.AutoField(verbose_name='ID', serialize=False, auto_created=True, primary_key=True)),
('field', JSONField(null=True, blank=True)),
+ ('field_custom', JSONField(null=True, blank=True, encoder=DjangoJSONEncoder)),
],
options={
},
diff --git a/tests/postgres_tests/models.py b/tests/postgres_tests/models.py
index d94eb90d4a..52dbc0335c 100644
--- a/tests/postgres_tests/models.py
+++ b/tests/postgres_tests/models.py
@@ -1,3 +1,4 @@
+from django.core.serializers.json import DjangoJSONEncoder
from django.db import connection, models
from .fields import (
@@ -132,6 +133,7 @@ class RangeLookupsModel(PostgreSQLModel):
if connection.vendor == 'postgresql' and connection.pg_version >= 90400:
class JSONModel(models.Model):
field = JSONField(blank=True, null=True)
+ field_custom = JSONField(blank=True, null=True, encoder=DjangoJSONEncoder)
else:
# create an object with this name so we don't have failing imports
class JSONModel(object):
diff --git a/tests/postgres_tests/test_json.py b/tests/postgres_tests/test_json.py
index b88d103761..fd4db7fae4 100644
--- a/tests/postgres_tests/test_json.py
+++ b/tests/postgres_tests/test_json.py
@@ -1,7 +1,12 @@
+from __future__ import unicode_literals
+
import datetime
import unittest
+import uuid
+from decimal import Decimal
from django.core import exceptions, serializers
+from django.core.serializers.json import DjangoJSONEncoder
from django.db import connection
from django.forms import CharField, Form, widgets
from django.test import TestCase
@@ -79,6 +84,27 @@ class TestSaveLoad(TestCase):
loaded = JSONModel.objects.get()
self.assertEqual(loaded.field, obj)
+ def test_custom_encoding(self):
+ """
+ JSONModel.field_custom has a custom DjangoJSONEncoder.
+ """
+ some_uuid = uuid.uuid4()
+ obj_before = {
+ 'date': datetime.date(2016, 8, 12),
+ 'datetime': datetime.datetime(2016, 8, 12, 13, 44, 47, 575981),
+ 'decimal': Decimal('10.54'),
+ 'uuid': some_uuid,
+ }
+ obj_after = {
+ 'date': '2016-08-12',
+ 'datetime': '2016-08-12T13:44:47.575',
+ 'decimal': '10.54',
+ 'uuid': str(some_uuid),
+ }
+ JSONModel.objects.create(field_custom=obj_before)
+ loaded = JSONModel.objects.get()
+ self.assertEqual(loaded.field_custom, obj_after)
+
@skipUnlessPG94
class TestQuerying(TestCase):
@@ -215,7 +241,10 @@ class TestQuerying(TestCase):
@skipUnlessPG94
class TestSerialization(TestCase):
- test_data = '[{"fields": {"field": {"a": "b", "c": null}}, "model": "postgres_tests.jsonmodel", "pk": null}]'
+ test_data = (
+ '[{"fields": {"field": {"a": "b", "c": null}, "field_custom": null}, '
+ '"model": "postgres_tests.jsonmodel", "pk": null}]'
+ )
def test_dumping(self):
instance = JSONModel(field={'a': 'b', 'c': None})
@@ -236,6 +265,12 @@ class TestValidation(PostgreSQLTestCase):
self.assertEqual(cm.exception.code, 'invalid')
self.assertEqual(cm.exception.message % cm.exception.params, "Value must be valid JSON.")
+ def test_custom_encoder(self):
+ with self.assertRaisesMessage(ValueError, "The encoder parameter must be a callable object."):
+ field = JSONField(encoder=DjangoJSONEncoder())
+ field = JSONField(encoder=DjangoJSONEncoder)
+ self.assertEqual(field.clean(datetime.timedelta(days=1), None), datetime.timedelta(days=1))
+
class TestFormField(PostgreSQLTestCase):