Initial import from Docker volume

tests/test_wsgi.py

@@ -0,0 +1,111 @@
+# This file is part of Tryton.  The COPYRIGHT file at the top level of
+# this repository contains the full copyright notices and license terms.
+
+import unittest
+from unittest.mock import Mock, sentinel
+
+from trytond.exceptions import TrytonException
+from trytond.protocols.wrappers import Response
+from trytond.tests.test_tryton import Client
+from trytond.wsgi import TrytondWSGI
+
+
+class WSGIAppTestCase(unittest.TestCase):
+    'Test WSGI Application'
+
+    class TestException(TrytonException):
+        pass
+
+    def test_one_error_handler_called(self):
+        """Test an error handler is called
+        with the app, the request and the exception."""
+        app = TrytondWSGI()
+        spy = Mock()
+
+        @app.error_handler
+        def _handler(*args, **kwargs):
+            spy(*args, **kwargs)
+
+        exception = self.TestException('foo')
+
+        @app.route('/willfail')
+        def _route(request):
+            sentinel.request = request
+            raise exception
+
+        client = Client(app, Response)
+        _ = client.get('/willfail')
+
+        spy.assert_called_once_with(app, sentinel.request, exception)
+
+    def test_many_error_handlers_called(self):
+        "Test many error handlers are called"
+        app = TrytondWSGI()
+        spy1 = Mock()
+        spy2 = Mock()
+
+        @app.error_handler
+        def _handler1(*args, **kwargs):
+            spy1(*args, **kwargs)
+
+        @app.error_handler
+        def _handler2(*args, **kwargs):
+            spy2(*args, **kwargs)
+
+        exception = self.TestException('foo')
+
+        @app.route('/willfail')
+        def _route(request):
+            sentinel.request = request
+            raise exception
+
+        client = Client(app, Response)
+        _ = client.get('/willfail')
+
+        spy1.assert_called_once_with(app, sentinel.request, exception)
+        spy2.assert_called_once_with(app, sentinel.request, exception)
+
+    def test_class_error_handler_called(self):
+        "Test class error handlers can be used"
+        app = TrytondWSGI()
+        spy = Mock()
+
+        class ErrorHandler():
+            def __call__(self, *args, **kwargs):
+                spy(*args, **kwargs)
+
+        app.error_handler(ErrorHandler())
+
+        exception = self.TestException('foo')
+
+        @app.route('/willfail')
+        def _route(request):
+            sentinel.request = request
+            raise exception
+
+        client = Client(app, Response)
+        _ = client.get('/willfail')
+
+        spy.assert_called_once_with(app, sentinel.request, exception)
+
+    def test_error_handlers_last_response(self):
+        "Test last handler response is used"
+        app = TrytondWSGI()
+
+        @app.error_handler
+        def _handler1(*args, **kwargs):
+            return Response(b'bar')
+
+        @app.error_handler
+        def _handler2(*args, **kwargs):
+            return Response(b'baz', status=418)
+
+        @app.route('/willfail')
+        def _route(request):
+            raise self.TestException('foo')
+
+        client = Client(app, Response)
+        response = client.get('/willfail')
+
+        self.assertEqual(next(response.response), b'baz')
+        self.assertEqual(response.status, "418 I'M A TEAPOT")