Ticket #11159: test_client_test.diff

tests/regressiontests/test_client_regress/models.py

@@ -11 +11 @@
 from django.core.exceptions import SuspiciousOperation
 from django.template import TemplateDoesNotExist, TemplateSyntaxError, Context
 from django.template import loader
+from django.test.client import encode_file
 
 class AssertContainsTests(TestCase):
     def setUp(self):
@@ -821 +822 @@
         response = self.client.post("/test_client_regress/parse_unicode_json/", json,
                                     content_type="application/json; charset=koi8-r")
         self.assertEqual(response.content, json.encode('koi8-r'))
+
+class TestClientTest(TestCase):
+    class DummyFile(object):
+        def __init__(self, filename):
+            self.name = filename
+        def read(self):
+            return 'TEST_FILE_CONTENT'
+            
+    def setUp(self):
+        self.encoded_file = encode_file('TEST_BOUNDARY', 'TEST_KEY', self.DummyFile('TEST_NAME.BIN'))
+    
+    def test_first_line_starts_with_dashes_and_boundary(self):
+        self.assertEqual('--TEST_BOUNDARY', self.encoded_file[0])
+        
+    def test_adds_content_disposition_form_data_filename(self):
+        self.assertEqual('Content-Disposition: form-data; name="TEST_KEY"; filename="TEST_NAME.BIN"', self.encoded_file[1])
+    
+    def test_guesses_content_type_on_file_encoding(self):
+        def assert_content_type_for_file_extension(expected, file_extension):
+            self.assertEqual('Content-Type: %s' % expected, 
+                             encode_file('IGNORE', 'IGNORE', self.DummyFile("FILE.%s" % file_extension))[2])
+            
+        assert_content_type_for_file_extension('application/octet-stream', 'BIN')
+        assert_content_type_for_file_extension('text/plain', 'TXT')
+        assert_content_type_for_file_extension('application/zip', 'ZIP')
+        assert_content_type_for_file_extension('application/octet-stream', 'UNKNOWN')
+        
+    def tests_appends_file_content(self):
+        self.assertEqual('TEST_FILE_CONTENT', self.encoded_file[-1])
+ No newline at end of file