mirror of
https://github.com/kennethreitz/httpbin.git
synced 2026-06-05 23:00:18 +00:00
Radomir Stevanovic
9a56bb8dd0
Encoding of the received files and/or body data was failing in case of raw binary data being POSTed/PUT/etc. (The exact trigger was actually a non-UTF-8 data stream, since Flask's `jsonify`, i.e. `simplejson.dumps` assumes UTF-8.) Binary data in a JSON response are now encoded as "data" URLs, according to RFC 2397. "Data" URL scheme was chosen for its simplicity and clarity. MIME type is included. Plain text is passed thru unmodified, as before. Also, tests demonstrating the bug/fix added to `test_httpbin.py`.
47 lines
1.4 KiB
Python
Executable File
47 lines
1.4 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import httpbin
|
|
import unittest
|
|
import base64
|
|
|
|
|
|
def _string_to_base64(string):
|
|
"""Encodes string to utf-8 and then base64"""
|
|
utf8_encoded = string.encode('utf-8')
|
|
return base64.urlsafe_b64encode(utf8_encoded)
|
|
|
|
|
|
class HttpbinTestCase(unittest.TestCase):
|
|
"""Httpbin tests"""
|
|
|
|
def setUp(self):
|
|
self.app = httpbin.app.test_client()
|
|
|
|
def test_base64(self):
|
|
greeting = u'Здравствуй, мир!'
|
|
b64_encoded = _string_to_base64(greeting)
|
|
response = self.app.get('/base64/{0}'.format(b64_encoded))
|
|
content = response.data.decode('utf-8')
|
|
self.assertEquals(greeting, content)
|
|
|
|
def test_post_binary(self):
|
|
response = self.app.post('/post',
|
|
data='\x01\x02\x03\x81\x82\x83',
|
|
content_type='application/octet-stream')
|
|
self.assertEquals(response.status_code, 200)
|
|
|
|
def test_post_file_text(self):
|
|
with open('httpbin/core.py') as f:
|
|
response = self.app.post('/post', data={"file": f})
|
|
self.assertEquals(response.status_code, 200)
|
|
|
|
def test_post_file_binary(self):
|
|
with open('httpbin/core.pyc') as f:
|
|
response = self.app.post('/post', data={"file": f})
|
|
self.assertEquals(response.status_code, 200)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|