Source

django-jsonfield / jsonfield / tests / test_forms.py

from django.test import TestCase as DjangoTestCase
from django.utils import unittest
from django.forms import ValidationError

from jsonfield.forms import JSONFormField
from jsonfield.tests.jsonfield_test_app.forms import JSONTestForm, JSONTestModelForm

class JSONFormFieldTest(DjangoTestCase):
    def test_form_field_clean_empty_object(self):
        field = JSONFormField(required=False)
        self.assertEquals({}, field.clean('{}'))
        
    def test_form_field_clean_object(self):
        field = JSONFormField(required=False)
        self.assertEquals(
            {'foo':'bar', 'baz':2},
            field.clean('{"foo":"bar","baz":2}')
        )
    
    def test_form_field_clean_empty_array(self):
        field = JSONFormField(required=False)
        self.assertEquals([],field.clean('[]'))
    
    def test_required_form_field_array(self):
        field = JSONFormField(required=True)
        self.assertEquals([], field.clean('[]'))
        
    def test_required_form_field_object(self):
        field = JSONFormField(required=True)
        self.assertEquals({}, field.clean('{}'))
    
    def test_required_form_field_empty(self):
        field = JSONFormField(required=True)
        with self.assertRaises(ValidationError):
            field.clean('')
    
    def test_invalid_json(self):
        field = JSONFormField(required=True)
        
        with self.assertRaises(ValidationError):
            field.clean('{"foo"}')

class JSONFormTest(DjangoTestCase):
    def test_form_clean(self):
        form = JSONTestForm({})
        self.assertFalse(form.is_valid())
    

class JSONFormMultipleSelectFieldTest(DjangoTestCase):
    def test_multiple_select_data(self):
        form = JSONTestForm({'json_data': ['SA', 'WA']})
        assert form.is_valid()
        
        self.assertEquals(['SA', 'WA'], form.cleaned_data['json_data'])