Add DictField tests

Author: insspb

tests/test_forms_v2.py

@@ -3,6 +3,7 @@
 
 import pytest
 from bson.json_util import RELAXED_JSON_OPTIONS
+from markupsafe import Markup
 from werkzeug.datastructures import MultiDict
 
 wtforms = pytest.importorskip("wtforms")
@@ -212,6 +213,105 @@ def test__process_formdata__does_call_parent_method_if_value_is_not_empty(
         assert obj.data is False
 
 
+class TestMongoDictField:
+    def test_mongo_dict_field_mro_not_changed(self):
+        field_mro = list(mongo_fields.MongoDictField.__mro__[:5])
+        assert field_mro == [
+            mongo_fields.MongoDictField,
+            mongo_fields.MongoTextAreaField,
+            mongo_fields.EmptyStringIsNoneMixin,
+            wtf_fields.TextAreaField,
+            wtf_fields.StringField,
+        ]
+
+    def test_mongo_dict_field_default_dict_in_form_object(self, db):
+        class DefaultDictModel(db.Document):
+            """Should populate form with {}."""
+
+            dict_field = db.DictField()
+
+        DefaultDictForm = DefaultDictModel.to_wtf_form()
+
+        form = DefaultDictForm()
+
+        assert str(form.dict_field) == Markup(
+            '<textarea id="dict_field" name="dict_field">\r\n{}</textarea>'
+        )
+        assert form.dict_field.data == {}  # This is mongoengine default
+        assert form.dict_field.null is False
+
+        assert form.validate()
+        form.save()
+
+        obj = DefaultDictModel.objects.get(id=form.instance.pk)
+        object_dict = json.loads(obj.to_json(json_options=RELAXED_JSON_OPTIONS))
+        object_dict.pop("_id")
+
+        assert object_dict == {"dict_field": {}}
+
+    @pytest.mark.parametrize(
+        ["null", "expected_obj"],
+        [
+            (True, {"placeholder_string": "1", "null_dict_field": None}),
+            (False, {"placeholder_string": "1"}),
+        ],
+    )
+    def test_mongo_dict_field_default_null_dict_in_form_object(
+        self, db, null, expected_obj
+    ):
+        class DefaultDictModel(db.Document):
+            """Should populate form with empty form."""
+
+            placeholder_string = db.StringField(default="1")
+            null_dict_field = db.DictField(default=None, null=null)
+
+        DefaultDictForm = DefaultDictModel.to_wtf_form()
+
+        form = DefaultDictForm()
+
+        assert str(form.null_dict_field) == Markup(
+            '<textarea id="null_dict_field" name="null_dict_field">\r\n</textarea>'
+        )
+        assert form.null_dict_field.data is None
+
+        assert form.validate()
+        form.save()
+
+        obj = DefaultDictModel.objects.get(id=form.instance.pk)
+        object_dict = json.loads(obj.to_json(json_options=RELAXED_JSON_OPTIONS))
+        object_dict.pop("_id")
+
+        assert object_dict == expected_obj
+
+    def test__parse_json_data__raise_error_when_input_is_incorrect_json(self, db):
+        class DictModel(db.Document):
+            """Should populate form with empty form."""
+
+            dict_field = db.DictField()
+
+        FormClass = DictModel.to_wtf_form()
+        form = FormClass(MultiDict({"dict_field": "foobar"}))
+        assert not form.validate()
+        assert form.errors == {
+            "dict_field": [
+                "Cannot load data: Expecting value: line 1 column 1 (char 0)"
+            ]
+        }
+
+    def test__ensure_data_is_dict__raise_error_when_input_is_a_list(self, db):
+        class DictModel(db.Document):
+            """Should populate form with empty form."""
+
+            dict_field = db.DictField()
+
+        FormClass = DictModel.to_wtf_form()
+        form = FormClass(MultiDict({"dict_field": "[]"}))
+        assert not form.validate()
+        assert form.errors == {
+            "dict_field": ["Not a valid dictionary (list input detected)."]
+        }
+
+
 class TestMongoEmailField:
     def test_email_field_mro_not_changed(self):
         field_mro = list(mongo_fields.MongoEmailField.__mro__[:4])
@@ -223,6 +323,12 @@ def test_email_field_mro_not_changed(self):
         ]
 
 
+class TestMongoFloatField:
+    def test_ensure_widget_not_accidentally_replaced(self):
+        field = mongo_fields.MongoFloatField
+        assert isinstance(field.widget, wtf_widgets.NumberInput)
+
+
 class TestMongoHiddenField:
     def test_hidden_field_mro_not_changed(self):
         field_mro = list(mongo_fields.MongoHiddenField.__mro__[:4])
@@ -317,9 +423,3 @@ def test_url_field_mro_not_changed(self):
             wtf_fields.URLField,
             wtf_fields.StringField,
         ]
-
-
-class TestMongoFloatField:
-    def test_ensure_widget_not_accidentally_replaced(self):
-        field = mongo_fields.MongoFloatField
-        assert isinstance(field.widget, wtf_widgets.NumberInput)