From 463bc488e4bf056aa039d5c8dca7dbf664a99bcd Mon Sep 17 00:00:00 2001
From: PandaFeeder <pandafeeder@gmail.com>
Date: Sat, 13 May 2017 18:09:56 +0800
Subject: [PATCH] add test for mounted_middleware that supports add middleware
 to specific handlers

---
 tests/test_mounted_middleware.py | 183 +++++++++++++++++++++++++++++++
 1 file changed, 183 insertions(+)
 create mode 100644 tests/test_mounted_middleware.py

diff --git a/tests/test_mounted_middleware.py b/tests/test_mounted_middleware.py
new file mode 100644
index 00000000..927b4ff1
--- /dev/null
+++ b/tests/test_mounted_middleware.py
@@ -0,0 +1,183 @@
+from json import loads as json_loads, dumps as json_dumps
+from sanic import Sanic
+from sanic.request import Request
+from sanic.response import json, text, HTTPResponse
+from sanic.exceptions import MiddlewareTypeError
+
+
+# ------------------------------------------------------------ #
+#  GET
+# ------------------------------------------------------------ #
+
+def test_mounted_request_middleware():
+    app = Sanic("test_mounted_request_middleware")
+
+    results = []
+    
+    async def request_middleware(request):
+        results.append(request)
+
+    @app.route('/', middleware={"request": request_middleware})
+    async def handler(request):
+        return text("OK")
+
+    request, response = app.test_client.get('/')
+    
+    assert response.text == "OK"
+    assert type(results[0]) is Request
+
+def test_early_response_from_mounted_request_middleware():
+    app = Sanic("test_early_response_from_mounted_request_middleware")
+
+    results = []
+
+    async def early_request_middleware(request):
+        results.append(1)
+        return text("response from early_request_middleware")
+
+    @app.route('/', middleware={"request": early_request_middleware})
+    async def handler(request):
+        results.append(2)
+        return text("response from handler")
+
+    request, response = app.test_client.get('/')
+
+    assert response.text == "reponse from early_request_middleware"
+    assert results[0] == 1
+    assert len(results) == 1
+
+def test_multi_mounted_request_middleware():
+    app = Sanic("test_multi_mounted_request_middleware")
+
+    results = []
+    added = ['ADD1', 'aADD1', 'ADD2', 'aADD2']
+
+    def mounted_request_middleware1(request):
+        results.append(added[0])
+    
+    async def mounted_async_request_middleware1(request):
+        results.append(added[1])
+
+    def mounted_request_middleware2(request):
+        results.append(added[2])
+    
+    async def mounted_async_request_middleware2(request):
+        results.append(added[3])
+    
+    middlewares = (
+        mounted_request_middleware1,
+        mounted_async_request_middleware1,
+        mounted_request_middleware2,
+        mounted_async_request_middleware2
+    )
+
+    @app.route('/', middleware={"request": middlewares})
+    async def handler(request):
+        return text("OK")
+
+    request, response = app.test_client.get('/')
+
+    for i, v in enumerate(added):
+        assert results[i] == added[i]
+
+    assert response.text == "OK"
+
+def test_mounted_request_middleware_parameter():
+    app = Sanic("test_mounted_request_middleware_parameter")
+
+    results = []
+    async def request_middleware(request, id):
+        results.append(id)
+
+    @app.route('/<id>', middleware={"request": request_middleware})
+    def handler(request, id):
+        return text("OK")
+
+    request, response = app.test_client.get('/42')
+
+    assert results[0] == 42 
+    assert response.text == "OK"
+    
+
+
+def test_mounted_response_middleware():
+    app = Sanic("test_mounted_response_middleware")
+
+    results = []
+
+    async def response_middleware(request, response):
+        response["response_middleware_added"] = "TEST"
+
+    @app.route('/', middleware={"response": response_middleware})
+    async def handler(request):
+        return text("OK")
+
+    request, response = app.test_client.get('/')
+
+    assert response["response_middleware_added"] == "TEST"
+
+
+def test_early_response_from_mounted_response_middleware():
+    app = Sanic("test_early_response_from_mounted_response_middleware")
+
+    results = []
+    added = ['hADD1', 'mADD2', 'eADD3', 'nADD4']
+
+    def response_middleware(request, response):
+        results.append(added[1])
+
+    async def early_response_middleware(request, response):
+        results.append(added[2])
+        return response.text(response.text+"middlewareOK")
+    
+    def not_called_response_middleware(requst, response):
+        results.append(added[3])
+
+    middlewares = [
+        response_middleware,
+        early_response_middleware,
+        not_called_response_middleware
+    ]
+
+    @app.route('/', middleware={"response": middlewares})
+    def handler(request):
+        results.append(added[0])        
+        return text('handlerOK')
+
+    request, response = app.test_client.get('/')
+
+    for i,v in enumerate(added):
+        if i != 3:
+            assert results[i] == added[i]
+        else:
+            assert results[i] != added[i]
+
+def test_mounted_response_middleware_parameter():
+    app = Sanic("test_mounted_response_middleware_parameter")
+
+    results = []
+
+    async def response_middleware(request, response, id):
+        results.append(id)
+
+    @app.route('/<id>', middleware={"response": response_middleware})
+    async def handler(request, id):
+        return text("OK")
+
+    request, response = app.test_client.get('/42')
+
+    assert results[0] == 42
+    assert response.text == "OK"
+    
+
+def test_exception():
+    app = Sanic("test_exception")
+
+    wrong_request = 123
+
+    try:
+        @app.route('/', middleware={"request": wrong_request})
+        async def handler1(request):
+            return text("BAD")
+    except MiddlewareTypeError as ex:
+        assert str(ex) == "Middleware mounted on a handler should be a function or a sequence of functions"