capture an error code and a reason when the server closes the connection

examples/ping_pong/server.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python
 
+import os
 import asyncio
 import websockets
 
@@ -8,6 +9,13 @@ async def echo(websocket, path):
         print(message)
         await websocket.send(message)
 
+        if os.getenv('TEST_CLOSE'):
+            print('Closing')
+            # breakpoint()
+            await websocket.close(1001, 'close message')
+            # await websocket.close()
+            break
+
 asyncio.get_event_loop().run_until_complete(
     websockets.serve(echo, 'localhost', 5678))
 asyncio.get_event_loop().run_forever()