kennethreitz/responder
1
0
Fork 0
mirror of https://github.com/kennethreitz/responder.git synced 2026-06-05 23:00:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

kinda test websocket support

Browse Source
This commit is contained in:
Kenneth Reitz
2018-10-22 10:05:20 -04:00
parent 6fc02964ba
commit ec13df75d0
2 changed files with 27 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
responder/models.py
+17 -17
View File
@@ -314,26 +314,26 @@ class WebSocket:
return QueryDict({})
async def accept(self):
return (await self._starlette.accept())
return await self._starlette.accept()
async def receive(self):
return (await self._starlette.receive())
return await self._starlette.receive()
async def content(self):
"""Receive bytes"""
return (await self._starlette.receive_bytes())
"""Receive bytes."""
return await self._starlette.receive_bytes()
async def text(self):
"""Receive text"""
return (await self._starlette.receive_text())
"""Receive text."""
return await self._starlette.receive_text()
async def json(self):
"""Receive json"""
return (await self._starlette.receive_json())
# async def json(self):
# """Receive json"""
# return await self._starlette.receive_json()
async def media(self):
"""Receive json"""
return (await self.receive_json())
# async def media(self):
# """Receive json"""
# return (await self.receive_json())
async def send(self):
await self._starlette.send()
@@ -341,11 +341,11 @@ class WebSocket:
async def send_text(self, data):
await self._starlette.send_text(data)
async def send_json(self, data):
await self._starlette.send_json(data)
# async def send_json(self, data):
# await self._starlette.send_json(data)
async def send_media(self, data):
await self.send_json(data)
# async def send_media(self, data):
# await self.send_json(data)
async def close(self, code=1000) -> None:
async def close(self, code=1000):
await self._starlette.close(code)
tests/test_responder.py
+10
View File
@@ -439,3 +439,13 @@ def test_404(session):
r = session.get("/foo")
assert r.status_code == responder.status_codes.HTTP_404
def test_kinda_websockets(api):
@api.route("/ws", websocket=True)
async def websocket(ws):
await ws.accept()
await ws.send_text("Hello via websocket!")
await ws.close()