2
如果上傳的文件包含不在硬編碼列表中的擴展名,我有一個基本模型字段驗證程序以引發ValidationError。測試不會引發Django模型字段上的ValidationError
只能從管理角度使用模型表單。但在我的測試中,儘管提供了無效的文件擴展名,但仍無法獲得例外。我究竟做錯了什麼?
驗證:
import os
from django.core.exceptions import ValidationError
def validate_file_type(value):
accepted_extensions = ['.png', '.jpg', '.jpeg', '.pdf']
extension = os.path.splitext(value.name)[1]
if extension not in accepted_extensions:
raise ValidationError(u'{} is not an accepted file type'.format(value))
型號:
from agency.utils.validators import validate_file_type
from django.db import models
from sorl.thumbnail import ImageField
class Client(models.Model):
"""
A past or current client of the agency.
"""
logo = ImageField(
help_text='Please use jpg (jpeg) or png files only. Will be resized \
for public display.',
upload_to='clients/logos',
default='',
validators=[validate_file_type]
測試:
from django.test import TestCase
import tempfile
import os
from settings import base
from clients.models import Client
class ClientTest(TestCase):
def setUp(self):
tempfile.tempdir = os.path.join(base.MEDIA_ROOT, 'clients/logos')
tf = tempfile.NamedTemporaryFile(delete=False, suffix='.png')
tf.close()
self.logo = tf.name
Client.objects.create(
name='Coca-Cola',
logo=self.logo,
website='http://us.coca-cola.com/home/'
)
def test_can_create_client(self):
client = Client.objects.get(name='Coca-Cola')
expected = 'Coca-Cola'
self.assertEqual(client.name, expected)
def tearDown(self):
os.remove(self.logo)
clients = Client.objects.all()
clients.delete()
你在哪裏希望錯誤得到提升? – 2014-11-03 19:56:44
好問題。我想在ClientTest的setUp函數中。 – 2014-11-03 19:57:28
此外,這個問題暗示我使用類似.txt而不是.png顯然。 – 2014-11-03 20:11:17