Add stream for CompositionView

tests/test_request_stream.py

@@ -1,11 +1,19 @@
 from sanic import Sanic
 from sanic.blueprints import Blueprint
+from sanic.views import CompositionView
+from sanic.views import HTTPMethodView
 from sanic.response import stream, text
 
 bp = Blueprint('test_blueprint_request_stream')
 app = Sanic('test_request_stream')
 
 
+class SimpleView(HTTPMethodView):
+
+    def get(self, request):
+        return text('OK')
+
+
 @app.stream('/stream')
 async def handler(request):
     async def streaming(response):
@@ -32,12 +40,44 @@ async def bp_handler(request):
         result += body.decode('utf-8')
     return text(result)
 
+
+def get_handler(request):
+    return text('OK')
+
+
+async def post_handler(request):
+    result = ''
+    while True:
+        body = await request.stream.get()
+        if body is None:
+            break
+        result += body.decode('utf-8')
+    return text(result)
+
+
+app.add_route(SimpleView.as_view(), '/method_view')
+
+view = CompositionView()
+view.add(['GET'], get_handler)
+view.add(['POST'], post_handler, stream=True)
+
 app.blueprint(bp)
 
+app.add_route(view, '/composition_view')
+
 
 def test_request_stream():
     data = "abc" * 100000
-    request, response = app.test_client.post('/stream', data=data)
+
+    request, response = app.test_client.get('/method_view')
+    assert response.status == 200
+    assert response.text == 'OK'
+
+    request, response = app.test_client.get('/composition_view')
+    assert response.status == 200
+    assert response.text == 'OK'
+
+    request, response = app.test_client.post('/composition_view', data=data)
     assert response.status == 200
     assert response.text == data
 
@@ -45,6 +85,10 @@ def test_request_stream():
     assert response.status == 200
     assert response.text == 'OK'
 
+    request, response = app.test_client.post('/stream', data=data)
+    assert response.status == 200
+    assert response.text == data
+
     request, response = app.test_client.post('/bp_stream', data=data)
     assert response.status == 200
     assert response.text == data