CODEWITHSUNDEEP
djangounit-testing
Kit Sunde
Kit Sunde

Reputation: 37075

How can I write tests that populates raw_post_data and request.FILES['myfile']

I have something like this:

def upload_something(request):
    data = {}
    if request.FILES:
        raw_file = request.FILES['myfile'].read()
    else:
        raw_file = request.raw_post_data

I can't seem to be able to write a unit-test that populates raw_post_data, how would I go about doing that? I basically just want to send an image file. I'm trying to create a test case for when I read raw_post_data and it errors with:

You cannot access raw_post_data after reading from request's data stream

Upvotes: 4

Views: 1672

Answers (4)

Danger
Danger

Reputation: 153

I'm assuming you have figured this out by now, but as the answers are almost out of date with the deprecation of raw_post_data I thought i'd post.

def test_xml_payload(self):
    data = '<?xml version="1.0" encoding="UTF-8"?><blah></blah>'
    response = self.client.post(reverse('my_url'),
                                data=data,
                                content_type='application/xml')


def my_view(request):
    xml = request.body

Upvotes: 4

Ilya
Ilya

Reputation: 1399

I took this listing here

c = Client()
f = open('wishlist.doc')
c.post('/customers/wishes/', {'name': 'fred', 'attachment': f})
f.close()

Client is a special class for testing your views. This is the example of posting files to your view. It's part of Django testing framework.

Upvotes: 0

jeffknupp
jeffknupp

Reputation: 6274

The error message the interpreter is giving is correct. After you access the POST data via if request.FILES, you can no longer access the raw_post_data. If in your actual code (not the tests) you hit that line, it would error with the same message. Basically, you need two separate views for form-based POSTS and direct file POSTS.

Upvotes: 1

Alexey Savanovich
Alexey Savanovich

Reputation: 1903

You can use mocking. Some examples available here and in docs here

Updated

Kit, I think it's very depends on your test case. But in general you shouldn't use raw_post_data directly. Instead it's have to be patched like in example below:

from mock import Mock, MagicMock

class SomeTestCase(TestCase):

    def testRawPostData(self):
        ...

        request = Mock(spec=request)
        request.raw_post_data = 'myrawdata'
        print request.raw_post_data  # prints 'myrawdata'

        file_mock = MagicMock(spec=file)
        file_mock.read.return_value = 'myfiledata'
        request.FILES = {'myfile': file_mock}    
        print request.FILES['myfile'].read()  # prints 'myfiledata'

Upvotes: 1

Related Questions