In [10]:

    

from django.test.client import RequestFactory
from cast import upload_handler


    

In [2]:

    

factory = RequestFactory()


    

In [22]:

    

request = factory.post('')
request.FILES["original"] = "blub.jpg"
form_class, context = upload_handler(request)


    

In [23]:

    

form_class._meta.model == Image


    

    
Out[23]:





True

In [24]:

    

request = factory.post('')
request.FILES["original"] = "blub.xls"
form_class, context = upload_handler(request)


    

    




---------------------------------------------------------------------------
KeyError                                  Traceback (most recent call last)
<ipython-input-24-4ac540bf613b> in <module>
      1 request = factory.post('')
      2 request.FILES["original"] = "blub.xls"
----> 3 form_class, context = upload_handler(request)

~/projects/django-cast/cast/__init__.py in upload_handler(request)
     41     file_name = str(request.FILES["original"])
     42     ending = Path(file_name).suffix.split(".")[-1].lower()
---> 43     local_model, upload_field_name, user_field = lookup[ending]
     44     form_class, context = get_model_form(
     45         local_model, upload_field_name=upload_field_name, user_field=user_field

KeyError: 'xls'

In [25]:

    

request = factory.post('')
request.FILES["original"] = "blub.ogg"
form_class, context = upload_handler(request)


    

In [26]:

    

context


    

    
Out[26]:





{'user_field': 'user', 'upload_field_name': 'oga'}

In [ ]: