2018-08-28 13:58:03 +00:00
|
|
|
|
|
|
|
|
The test client allows you to make requests against your ASGI application,
|
|
|
|
|
using the `requests` library.
|
|
|
|
|
|
|
|
|
|
```python
|
2018-08-28 14:18:17 +00:00
|
|
|
from starlette.response import HTMLResponse
|
|
|
|
|
from starlette.testclient import TestClient
|
2018-08-28 13:58:03 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
class App:
|
|
|
|
|
def __init__(self, scope):
|
|
|
|
|
self.scope = scope
|
|
|
|
|
|
|
|
|
|
async def __call__(self, receive, send):
|
|
|
|
|
response = HTMLResponse('<html><body>Hello, world!</body></html>')
|
|
|
|
|
await response(receive, send)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_app():
|
|
|
|
|
client = TestClient(App)
|
|
|
|
|
response = client.get('/')
|
|
|
|
|
assert response.status_code == 200
|
|
|
|
|
```
|
|
|
|
|
|
|
|
|
|
The test client exposes the same interface as any other `requests` session.
|
|
|
|
|
In particular, note that the calls to make a request are just standard
|
|
|
|
|
function calls, not awaitables.
|
|
|
|
|
|
|
|
|
|
### Testing WebSocket applications
|
|
|
|
|
|
|
|
|
|
You can also test websocket applications with the test client.
|
|
|
|
|
|
|
|
|
|
The `requests` library will be used to build the initial handshake, meaning you
|
|
|
|
|
can use the same authentication options and other headers between both http and
|
|
|
|
|
websocket testing.
|
|
|
|
|
|
|
|
|
|
```python
|
|
|
|
|
from starlette.testclient import TestClient
|
|
|
|
|
from starlette.websockets import WebSocketSession
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class App:
|
|
|
|
|
def __init__(self, scope):
|
|
|
|
|
self.scope = scope
|
|
|
|
|
|
|
|
|
|
async def __call__(self, receive, send):
|
|
|
|
|
session = WebSocketSession(self.scope, receive=receive, send=send)
|
|
|
|
|
await session.accept()
|
|
|
|
|
await session.send_text('Hello, world!')
|
|
|
|
|
await session.close()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def test_app():
|
|
|
|
|
client = TestClient(App)
|
2018-08-28 14:18:17 +00:00
|
|
|
with client.websocket_connect('/') as session:
|
2018-08-28 13:58:03 +00:00
|
|
|
data = session.receive_text()
|
|
|
|
|
assert data == 'Hello, world!'
|
|
|
|
|
```
|
|
|
|
|
|
|
|
|
|
The operations on session are standard function calls, not awaitables.
|
|
|
|
|
|
|
|
|
|
It's important to use the session within a context-managed `with` block. This
|
|
|
|
|
ensure that the background thread on which the ASGI application is properly
|
|
|
|
|
terminated, and that any exceptions that occur within the application are
|
|
|
|
|
always raised by the test client.
|
|
|
|
|
|
|
|
|
|
#### Establishing a test session
|
|
|
|
|
|
2018-08-28 14:18:17 +00:00
|
|
|
* `.websocket_connect(url, subprotocols=None, **options)` - Takes the same set of arguments as `requests.get()`.
|
2018-08-28 13:58:03 +00:00
|
|
|
|
|
|
|
|
May raise `starlette.websockets.Disconnect` if the application does not accept the websocket connection.
|
|
|
|
|
|
2018-08-28 14:18:17 +00:00
|
|
|
#### Sending data
|
|
|
|
|
|
|
|
|
|
* `.send_text(data)` - Send the given text to the application.
|
|
|
|
|
* `.send_bytes(data)` - Send the given bytes to the application.
|
|
|
|
|
* `.send_json(data)` - Send the given data to the application.
|
|
|
|
|
|
2018-08-28 13:58:03 +00:00
|
|
|
#### Receiving data
|
|
|
|
|
|
|
|
|
|
* `.receive_text()` - Wait for incoming text sent by the application and return it.
|
|
|
|
|
* `.receive_bytes()` - Wait for incoming bytestring sent by the application and return it.
|
|
|
|
|
* `.receive_json()` - Wait for incoming json data sent by the application and return it.
|
|
|
|
|
|
|
|
|
|
May raise `starlette.websockets.Disconnect`.
|
|
|
|
|
|
|
|
|
|
#### Closing the connection
|
|
|
|
|
|
|
|
|
|
* `.close(code=1000)` - Perform a client-side close of the websocket connection.
|
