aboutsummaryrefslogtreecommitdiffstats
path: root/djangorestframework/tests/files.py
diff options
context:
space:
mode:
authorTom Christie2011-06-02 12:58:10 +0100
committerTom Christie2011-06-02 12:58:10 +0100
commitb50492853f537a2473bb0a9eea86c8b0ed6b8824 (patch)
treed289d39aacf187a8a0696a4c1c863aabe1472c3a /djangorestframework/tests/files.py
parent7ee9adbe5c03c29cd4a894dd476548f7fe73b5e4 (diff)
parentfc1640de75511006e89f033c9270ec91a9f1e4d4 (diff)
downloaddjango-rest-framework-b50492853f537a2473bb0a9eea86c8b0ed6b8824.tar.bz2
pull in -dev as 0.2.0
Diffstat (limited to 'djangorestframework/tests/files.py')
-rw-r--r--djangorestframework/tests/files.py19
1 files changed, 11 insertions, 8 deletions
diff --git a/djangorestframework/tests/files.py b/djangorestframework/tests/files.py
index e155f181..25aad9b4 100644
--- a/djangorestframework/tests/files.py
+++ b/djangorestframework/tests/files.py
@@ -1,7 +1,8 @@
from django.test import TestCase
from django import forms
from djangorestframework.compat import RequestFactory
-from djangorestframework.resource import Resource
+from djangorestframework.views import View
+from djangorestframework.resources import FormResource
import StringIO
class UploadFilesTests(TestCase):
@@ -15,19 +16,21 @@ class UploadFilesTests(TestCase):
class FileForm(forms.Form):
file = forms.FileField
- class MockResource(Resource):
- allowed_methods = anon_allowed_methods = ('POST',)
+ class MockResource(FormResource):
form = FileForm
- def post(self, request, auth, content, *args, **kwargs):
- #self.uploaded = content.file
- return {'FILE_NAME': content['file'].name,
- 'FILE_CONTENT': content['file'].read()}
+ class MockView(View):
+ 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 = MockResource.as_view()
+ view = MockView.as_view()
response = view(request)
self.assertEquals(response.content, '{"FILE_CONTENT": "stuff", "FILE_NAME": "stuff.txt"}')