cubicweb/pyramid/test/test_bw_request.py
changeset 11631 faf279e33298
parent 11630 1400aee10df4
child 11811 f09efeead7f9
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/cubicweb/pyramid/test/test_bw_request.py	Mon Sep 26 14:52:12 2016 +0200
@@ -0,0 +1,103 @@
+# -*- coding: utf-8 -*-
+from io import BytesIO
+
+import webtest
+
+import pyramid.request
+
+from cubicweb.pyramid.core import CubicWebPyramidRequest
+from cubicweb.pyramid.test import PyramidCWTest
+
+
+class WSGIAppTest(PyramidCWTest):
+    def make_request(self, path, environ=None, **kw):
+        r = webtest.app.TestRequest.blank(path, environ, **kw)
+
+        request = pyramid.request.Request(r.environ)
+        request.registry = self.pyr_registry
+
+        return request
+
+    def test_content_type(self):
+        req = CubicWebPyramidRequest(
+            self.make_request('/', {'CONTENT_TYPE': 'text/plain'}))
+
+        self.assertEqual('text/plain', req.get_header('Content-Type'))
+
+    def test_content_body(self):
+        req = CubicWebPyramidRequest(
+            self.make_request('/', {
+                'CONTENT_LENGTH': 12,
+                'CONTENT_TYPE': 'text/plain',
+                'wsgi.input': BytesIO(b'some content')}))
+
+        self.assertEqual(b'some content', req.content.read())
+
+    def test_http_scheme(self):
+        req = CubicWebPyramidRequest(
+            self.make_request('/', {
+                'wsgi.url_scheme': 'http'}))
+
+        self.assertFalse(req.https)
+
+    def test_https_scheme(self):
+        req = CubicWebPyramidRequest(
+            self.make_request('/', {
+                'wsgi.url_scheme': 'https'}))
+
+        self.assertTrue(req.https)
+
+    def test_https_prefix(self):
+        r = self.webapp.get('/https/')
+        self.assertIn('https://', r.text)
+
+    def test_big_content(self):
+        content = b'x'*100001
+
+        req = CubicWebPyramidRequest(
+            self.make_request('/', {
+                'CONTENT_LENGTH': len(content),
+                'CONTENT_TYPE': 'text/plain',
+                'wsgi.input': BytesIO(content)}))
+
+        self.assertEqual(content, req.content.read())
+
+    def test_post(self):
+        self.webapp.post(
+            '/',
+            params={'__login': self.admlogin, '__password': self.admpassword})
+
+    def test_get_multiple_variables(self):
+        req = CubicWebPyramidRequest(
+            self.make_request('/?arg=1&arg=2'))
+
+        self.assertEqual([u'1', u'2'], req.form['arg'])
+
+    def test_post_multiple_variables(self):
+        req = CubicWebPyramidRequest(
+            self.make_request('/', POST='arg=1&arg=2'))
+
+        self.assertEqual([u'1', u'2'], req.form['arg'])
+
+    def test_post_files(self):
+        content_type, params = self.webapp.encode_multipart(
+            (), (('filefield', 'aname', b'acontent'),))
+        req = CubicWebPyramidRequest(
+            self.make_request('/', POST=params, content_type=content_type))
+        self.assertIn('filefield', req.form)
+        fieldvalue = req.form['filefield']
+        self.assertEqual(u'aname', fieldvalue[0])
+        self.assertEqual(b'acontent', fieldvalue[1].read())
+
+    def test_post_unicode_urlencoded(self):
+        params = 'arg=%C3%A9'
+        req = CubicWebPyramidRequest(
+            self.make_request(
+                '/', POST=params,
+                content_type='application/x-www-form-urlencoded'))
+        self.assertEqual(u"é", req.form['arg'])
+
+
+if __name__ == '__main__':
+    from unittest import main
+    main()