From 75094949930fcba364f1a38db31c10feae378fed Mon Sep 17 00:00:00 2001 From: PandaFeeder Date: Sat, 13 May 2017 22:02:40 +0800 Subject: [PATCH] fix unproper test case --- tests/test_mounted_middleware.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/tests/test_mounted_middleware.py b/tests/test_mounted_middleware.py index 87180982..df7c09a6 100644 --- a/tests/test_mounted_middleware.py +++ b/tests/test_mounted_middleware.py @@ -106,7 +106,7 @@ def test_mounted_response_middleware(): results = [] async def response_middleware(request, response): - response.response_middleware_added = "TEST" + results.append(42) @app.route('/', middleware={"response": response_middleware}) async def handler(request): @@ -114,7 +114,7 @@ def test_mounted_response_middleware(): request, response = app.test_client.get('/') - assert response.response_middleware_added == "TEST" + assert results[0] == 42 def test_early_response_from_mounted_response_middleware(): @@ -128,7 +128,7 @@ def test_early_response_from_mounted_response_middleware(): async def early_response_middleware(request, response): results.append(added[2]) - return response.text(response.text+"middlewareOK") + response.headers.update({"Added": "Response Middleware"}) def not_called_response_middleware(requst, response): results.append(added[3]) @@ -149,7 +149,9 @@ def test_early_response_from_mounted_response_middleware(): for i,v in enumerate(added): if i != 3: assert results[i] == added[i] + assert len(results) == 3 + assert response.headers["Added"]== "Response Middleware" def test_mounted_response_middleware_parameter(): app = Sanic("test_mounted_response_middleware_parameter")