diff options
| author | Michael Elovskikh | 2013-05-04 14:58:21 +0600 |
|---|---|---|
| committer | Michael Elovskikh | 2013-05-04 14:58:21 +0600 |
| commit | e36e4f48ad481b4303e68ed524677add07b224f7 (patch) | |
| tree | 6e97a37b8ae8628781750c002fd40935fbb71f9a /rest_framework/tests/parsers.py | |
| parent | 318fdaabe560c99de4983e0a3cdcb79756baaf01 (diff) | |
| download | django-rest-framework-e36e4f48ad481b4303e68ed524677add07b224f7.tar.bz2 | |
Codebase improvements on FileUploadParser
* Added docstrings.
* Added `FileUploadParser.get_filename` to make it easier to override.
* Added url kwargs filename detection step.
* Updated tests corresponding to these changes.
Diffstat (limited to 'rest_framework/tests/parsers.py')
| -rw-r--r-- | rest_framework/tests/parsers.py | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/rest_framework/tests/parsers.py b/rest_framework/tests/parsers.py index b18ecbf2..7699e10c 100644 --- a/rest_framework/tests/parsers.py +++ b/rest_framework/tests/parsers.py @@ -99,11 +99,17 @@ class TestFileUploadParser(TestCase): 'HTTP_CONTENT_DISPOSITION': 'Content-Disposition: inline; filename=file.txt'.encode('utf-8'), 'HTTP_CONTENT_LENGTH': 14, } - self.parser_context = {'request': request} + self.parser_context = {'request': request, 'kwargs': {}} def test_parse(self): """ Make sure the `QueryDict` works OK """ parser = FileUploadParser() - data_and_files = parser.parse(self.stream, parser_context=self.parser_context) + self.stream.seek(0) + data_and_files = parser.parse(self.stream, None, self.parser_context) file_obj = data_and_files.files['file'] self.assertEqual(file_obj._size, 14) + + def test_get_filename(self): + parser = FileUploadParser() + filename = parser.get_filename(self.stream, None, self.parser_context) + self.assertEqual(filename, 'file.txt'.encode('utf-8')) |
