Now the BinaryField value is wrapepd in bin_type; added a test to avoid regression

Author: mapio

flask_mongoengine/wtf/fields.py

@@ -9,7 +9,7 @@
 from wtforms.validators import ValidationError
 
 from mongoengine.queryset import DoesNotExist
-from mongoengine.python_support import txt_type
+from mongoengine.python_support import txt_type, bin_type
 
 __all__ = (
     'ModelSelectField', 'QuerySetSelectField',
@@ -163,11 +163,9 @@ def process_formdata(self, valuelist):
 
 class BinaryField(TextAreaField):
     """
-    Custom TextAreaField that returns str (instead of unicode) data.
+    Custom TextAreaField that converts its value with bin_type.
     """
 
     def process_formdata(self, valuelist):
         if valuelist:
-            self.data = str( valuelist[0] )
-        if self.data == "":
-            self.data = None
+            self.data = bin_type( valuelist[0] )

tests/test_forms.py

@@ -32,6 +32,19 @@ def setUp(self):
     def tearDown(self):
         self.db.connection.drop_database(self.db_name)
 
+    def test_binaryfield(self):
+
+        with self.app.test_request_context('/'):
+            db = self.db
+
+            class Binary(db.Document):
+                binary = db.BinaryField()
+
+            BinaryForm = model_form(Binary)
+            form = BinaryForm(MultiDict({'binary': '1'}))
+            self.assertTrue(form.validate())
+            form.save()
+
     def test_choices_coerce(self):
 
         with self.app.test_request_context('/'):