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 | |
| 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')
| -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'))  | 
