Update asyncio-server example to let it could be test with h2spec

chhsiao90 · chhsiao90 · commit cd7d9e8aead4 · 2017-02-09T16:48:44.000+08:00
diff --git a/examples/asyncio/asyncio-server.py b/examples/asyncio/asyncio-server.py
@@ -21,8 +21,11 @@
 from typing import List, Tuple
 
 from h2.connection import H2Connection
-from h2.events import DataReceived, RequestReceived, StreamEnded
+from h2.events import (
+    ConnectionTerminated, DataReceived, RequestReceived, StreamEnded
+)
 from h2.errors import ErrorCodes
+from h2.exceptions import ProtocolError
 
 
 RequestData = collections.namedtuple('RequestData', ['headers', 'data'])
@@ -40,18 +43,24 @@ def connection_made(self, transport: asyncio.Transport):
         self.transport.write(self.conn.data_to_send())
 
     def data_received(self, data: bytes):
-        events = self.conn.receive_data(data)
-        self.transport.write(self.conn.data_to_send())
-
-        for event in events:
-            if isinstance(event, RequestReceived):
-                self.request_received(event.headers, event.stream_id)
-            elif isinstance(event, DataReceived):
-                self.receive_data(event.data, event.stream_id)
-            elif isinstance(event, StreamEnded):
-                self.stream_complete(event.stream_id)
-
+        try:
+            events = self.conn.receive_data(data)
+        except ProtocolError as e:
+            self.transport.write(self.conn.data_to_send())
+            self.transport.close()
+        else:
             self.transport.write(self.conn.data_to_send())
+            for event in events:
+                if isinstance(event, RequestReceived):
+                    self.request_received(event.headers, event.stream_id)
+                elif isinstance(event, DataReceived):
+                    self.receive_data(event.data, event.stream_id)
+                elif isinstance(event, StreamEnded):
+                    self.stream_complete(event.stream_id)
+                elif isinstance(event, ConnectionTerminated):
+                    self.transport.close()
+
+                self.transport.write(self.conn.data_to_send())
 
     def request_received(self, headers: List[Tuple[str, str]], stream_id: int):
         headers = collections.OrderedDict(headers)
@@ -86,7 +95,7 @@ def stream_complete(self, stream_id: int):
         response_headers = (
             (':status', '200'),
             ('content-type', 'application/json'),
-            ('content-length', len(data)),
+            ('content-length', str(len(data))),
             ('server', 'asyncio-h2'),
         )
         self.conn.send_headers(stream_id, response_headers)
@@ -118,7 +127,6 @@ def receive_data(self, data: bytes, stream_id: int):
             stream_data.data.write(data)
 
 
-
 ssl_context = ssl.create_default_context(ssl.Purpose.CLIENT_AUTH)
 ssl_context.options |= (
     ssl.OP_NO_TLSv1 | ssl.OP_NO_TLSv1_1 | ssl.OP_NO_COMPRESSION
