trac#5397: Allow decode_request to parse content-type headers with extra options.

It previously parsed "Content-Type: application/x-www-form-urlencoded", but not "Content-Type: application/x-www-form-urlencoded; charset=utf-8".

1 files changed, 61 insertions, 0 deletions

diff --git a/mediagoblin/tests/test_tools.py b/mediagoblin/tests/test_tools.py
new file mode 100644
index 00000000..6d3dd475
--- /dev/null
+++ b/mediagoblin/tests/test_tools.py
@@ -0,0 +1,61 @@
+# GNU MediaGoblin -- federated, autonomous media hosting
+# Copyright (C) 2011, 2012 MediaGoblin contributors.  See AUTHORS.
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+from __future__ import absolute_import, unicode_literals
+
+from werkzeug.wrappers import Request
+from werkzeug.test import EnvironBuilder
+
+from mediagoblin.tools.request import decode_request
+
+class TestDecodeRequest(object):
+    """Test the decode_request function."""
+
+    def test_form_type(self):
+        """Try a normal form-urlencoded request."""
+        builder = EnvironBuilder(method='POST', data={'foo': 'bar'})
+        request = Request(builder.get_environ())
+        data = decode_request(request)
+        assert data['foo'] == 'bar'
+
+    def test_json_type(self):
+        """Try a normal JSON request."""
+        builder = EnvironBuilder(
+            method='POST', content_type='application/json',
+            data='{"foo": "bar"}')
+        request = Request(builder.get_environ())
+        data = decode_request(request)
+        assert data['foo'] == 'bar'
+
+    def test_content_type_with_options(self):
+        """Content-Type can also have options."""
+        builder = EnvironBuilder(
+            method='POST',
+            content_type='application/x-www-form-urlencoded; charset=utf-8')
+        request = Request(builder.get_environ())
+        # Must populate form field manually with non-default content-type.
+        request.form = {'foo': 'bar'}
+        data = decode_request(request)
+        assert data['foo'] == 'bar'
+
+    def test_form_type_is_default(self):
+        """Assume form-urlencoded if blank in the request."""
+        builder = EnvironBuilder(method='POST', content_type='')
+        request = Request(builder.get_environ())
+        # Must populate form field manually with non-default content-type.
+        request.form = {'foo': 'bar'}
+        data = decode_request(request)
+        assert data['foo'] == 'bar'