django-rest-framework/djangorestframework/tests/files.py

from django.test import TestCase
from django import forms
from djangorestframework.compat import RequestFactory
from djangorestframework.views import BaseView
from djangorestframework.resources import FormResource
import StringIO

class UploadFilesTests(TestCase):
    """Check uploading of files"""
    def setUp(self):
        self.factory = RequestFactory()

    def test_upload_file(self):


        class FileForm(forms.Form):
            file = forms.FileField

        class MockResource(FormResource):
            form = FileForm

        class MockView(BaseView):
            permissions = ()
            resource = MockResource

            def post(self, request, *args, **kwargs):
                return {'FILE_NAME': self.CONTENT['file'][0].name,
                        'FILE_CONTENT': self.CONTENT['file'][0].read()}
                
        file = StringIO.StringIO('stuff')
        file.name = 'stuff.txt'
        request = self.factory.post('/', {'file': file})
        view = MockView.as_view()
        response = view(request)
        self.assertEquals(response.content, '{"FILE_CONTENT": "stuff", "FILE_NAME": "stuff.txt"}')