Check sizes in form

Author: hirunatan

validatedfile/models.py

@@ -1,5 +1,6 @@
 from django.db import models
 from django import forms
+from django.template.defaultfilters import filesizeformat
 from django.utils.translation import ugettext as _
 
 class ValidatedFileField(models.FileField):

validatedfile/tests/__init__.py

@@ -71,3 +71,15 @@ def test_form_invalid_filetype(self):
         self.assertEqual(len(form.errors), 1)
         self.assertEqual(len(form.errors['the_file']), 1)
 
+
+    def test_form_invalid_size(self):
+        uploaded_file = SimpleUploadedFile(
+                name = 'the_file.pdf',
+                content = self._get_sample_file('image15k.png').read(),
+                content_type = 'image/png',
+            )
+        form = TestModelForm(data = {}, files = {'the_file': uploaded_file})
+        self.assertFalse(form.is_valid())
+        self.assertEqual(len(form.errors), 1)
+        self.assertEqual(len(form.errors['the_file']), 1)
+