def test_accepted_types_field(self):
accepted_types = ['image/bmp','image/jpg', 'image/png']
field = RestrictedTypeFileField(accepted_types=accepted_types)
f = File(open(os.path.join(os.path.abspath(os.path.dirname(__file__)), '__init__.py')))
f.content_type = 'image/bmp'
data = type("",(object,),dict(file=f, accepted_types=accepted_types))
self.assertEquals(field.clean(data, {}).file, f)
def test_no_accepted_types_field(self):
field = RestrictedTypeFileField()
f = File(open(os.path.join(os.path.abspath(os.path.dirname(__file__)), '__init__.py')))
f.content_type = 'image/bmp'
data = type("",(object,),dict(file=f))
with self.assertRaises(forms.ValidationError):
field.clean(data, None)