From ac44900fc40f3296ba858eaf371a1213339f2db2 Mon Sep 17 00:00:00 2001
From: 38elements <mh19820223@gmail.com>
Date: Mon, 26 Dec 2016 23:41:10 +0900
Subject: [PATCH] Add test and example for custom protocol

---
 examples/custom_protocol.py   | 24 ++++++++++++++++++++++++
 sanic/utils.py                |  6 +++---
 tests/test_custom_protocol.py | 32 ++++++++++++++++++++++++++++++++
 3 files changed, 59 insertions(+), 3 deletions(-)
 create mode 100644 examples/custom_protocol.py
 create mode 100644 tests/test_custom_protocol.py

diff --git a/examples/custom_protocol.py b/examples/custom_protocol.py
new file mode 100644
index 00000000..b5e20ee6
--- /dev/null
+++ b/examples/custom_protocol.py
@@ -0,0 +1,24 @@
+from sanic import Sanic
+from sanic.server import HttpProtocol
+from sanic.response import text
+
+app = Sanic(__name__)
+
+
+class CustomHttpProtocol(HttpProtocol):
+
+    def write_response(self, response):
+        if isinstance(response, str):
+            response = text(response)
+        self.transport.write(
+            response.output(self.request.version)
+        )
+        self.transport.close()
+
+
+@app.route("/")
+async def test(request):
+    return 'Hello, world!'
+
+
+app.run(host="0.0.0.0", port=8000, protocol=CustomHttpProtocol)
diff --git a/sanic/utils.py b/sanic/utils.py
index 88444b3c..9f4a97d8 100644
--- a/sanic/utils.py
+++ b/sanic/utils.py
@@ -16,8 +16,8 @@ async def local_request(method, uri, cookies=None, *args, **kwargs):
 
 
 def sanic_endpoint_test(app, method='get', uri='/', gather_request=True,
-                        loop=None, debug=False, *request_args,
-                        **request_kwargs):
+                        loop=None, debug=False, server_kwargs={},
+                        *request_args, **request_kwargs):
     results = []
     exceptions = []
 
@@ -36,7 +36,7 @@ def sanic_endpoint_test(app, method='get', uri='/', gather_request=True,
         app.stop()
 
     app.run(host=HOST, debug=debug, port=42101,
-            after_start=_collect_response, loop=loop)
+            after_start=_collect_response, loop=loop, **server_kwargs)
 
     if exceptions:
         raise ValueError("Exception during request: {}".format(exceptions))
diff --git a/tests/test_custom_protocol.py b/tests/test_custom_protocol.py
new file mode 100644
index 00000000..88202428
--- /dev/null
+++ b/tests/test_custom_protocol.py
@@ -0,0 +1,32 @@
+from sanic import Sanic
+from sanic.server import HttpProtocol
+from sanic.response import text
+from sanic.utils import sanic_endpoint_test
+
+app = Sanic('test_custom_porotocol')
+
+
+class CustomHttpProtocol(HttpProtocol):
+
+    def write_response(self, response):
+        if isinstance(response, str):
+            response = text(response)
+        self.transport.write(
+            response.output(self.request.version)
+        )
+        self.transport.close()
+
+
+@app.route('/1')
+async def handler_1(request):
+    return 'OK'
+
+
+def test_use_custom_protocol():
+    server_kwargs = {
+        'protocol': CustomHttpProtocol
+    }
+    request, response = sanic_endpoint_test(app, uri='/1',
+                                            server_kwargs=server_kwargs)
+    assert response.status == 200
+    assert response.text == 'OK'