lqs
d20566fe07
Some checks failed
Test / lint (push) Has been cancelled
Test / test (pydantic-v1, 3.10) (push) Has been cancelled
Test / test (pydantic-v1, 3.11) (push) Has been cancelled
Test / test (pydantic-v1, 3.12) (push) Has been cancelled
Test / test (pydantic-v1, 3.8) (push) Has been cancelled
Test / test (pydantic-v1, 3.9) (push) Has been cancelled
Test / test (pydantic-v2, 3.10) (push) Has been cancelled
Test / test (pydantic-v2, 3.11) (push) Has been cancelled
Test / test (pydantic-v2, 3.12) (push) Has been cancelled
Test / test (pydantic-v2, 3.8) (push) Has been cancelled
Test / test (pydantic-v2, 3.9) (push) Has been cancelled
Test / coverage-combine (push) Has been cancelled
Test / check (push) Has been cancelled
Issue Manager / issue-manager (push) Has been cancelled
Label Approved / label-approved (push) Has been cancelled
28 lines
872 B
Python
28 lines
872 B
Python
from fastapi.testclient import TestClient
|
|
|
|
from docs_src.websockets.tutorial003 import app, html
|
|
|
|
client = TestClient(app)
|
|
|
|
|
|
def test_get():
|
|
response = client.get("/")
|
|
assert response.text == html
|
|
|
|
|
|
def test_websocket_handle_disconnection():
|
|
with client.websocket_connect("/ws/1234") as connection, client.websocket_connect(
|
|
"/ws/5678"
|
|
) as connection_two:
|
|
connection.send_text("Hello from 1234")
|
|
data1 = connection.receive_text()
|
|
assert data1 == "You wrote: Hello from 1234"
|
|
data2 = connection_two.receive_text()
|
|
client1_says = "Client #1234 says: Hello from 1234"
|
|
assert data2 == client1_says
|
|
data1 = connection.receive_text()
|
|
assert data1 == client1_says
|
|
connection_two.close()
|
|
data1 = connection.receive_text()
|
|
assert data1 == "Client #5678 left the chat"
|