+import importlib
+
import pytest
+from fastapi import FastAPI
from fastapi.testclient import TestClient
from fastapi.websockets import WebSocketDisconnect
-from docs_src.websockets.tutorial002 import app
+from ...utils import needs_py39, needs_py310
+
+
+@pytest.fixture(
+ name="app",
+ params=[
+ "tutorial002",
+ pytest.param("tutorial002_py310", marks=needs_py310),
+ "tutorial002_an",
+ pytest.param("tutorial002_an_py39", marks=needs_py39),
+ pytest.param("tutorial002_an_py310", marks=needs_py310),
+ ],
+)
+def get_app(request: pytest.FixtureRequest):
+ mod = importlib.import_module(f"docs_src.websockets.{request.param}")
+
+ return mod.app
-def test_main():
+def test_main(app: FastAPI):
client = TestClient(app)
response = client.get("/")
assert response.status_code == 200, response.text
assert b"<!DOCTYPE html>" in response.content
-def test_websocket_with_cookie():
+def test_websocket_with_cookie(app: FastAPI):
client = TestClient(app, cookies={"session": "fakesession"})
with pytest.raises(WebSocketDisconnect):
with client.websocket_connect("/items/foo/ws") as websocket:
assert data == f"Message text was: {message}, for item ID: foo"
-def test_websocket_with_header():
+def test_websocket_with_header(app: FastAPI):
client = TestClient(app)
with pytest.raises(WebSocketDisconnect):
with client.websocket_connect("/items/bar/ws?token=some-token") as websocket:
assert data == f"Message text was: {message}, for item ID: bar"
-def test_websocket_with_header_and_query():
+def test_websocket_with_header_and_query(app: FastAPI):
client = TestClient(app)
with pytest.raises(WebSocketDisconnect):
with client.websocket_connect("/items/2/ws?q=3&token=some-token") as websocket:
assert data == f"Message text was: {message}, for item ID: 2"
-def test_websocket_no_credentials():
+def test_websocket_no_credentials(app: FastAPI):
client = TestClient(app)
with pytest.raises(WebSocketDisconnect):
with client.websocket_connect("/items/foo/ws"):
) # pragma: no cover
-def test_websocket_invalid_data():
+def test_websocket_invalid_data(app: FastAPI):
client = TestClient(app)
with pytest.raises(WebSocketDisconnect):
with client.websocket_connect("/items/foo/ws?q=bar&token=some-token"):
+++ /dev/null
-import pytest
-from fastapi.testclient import TestClient
-from fastapi.websockets import WebSocketDisconnect
-
-from docs_src.websockets.tutorial002_an import app
-
-
-def test_main():
- client = TestClient(app)
- response = client.get("/")
- assert response.status_code == 200, response.text
- assert b"<!DOCTYPE html>" in response.content
-
-
-def test_websocket_with_cookie():
- client = TestClient(app, cookies={"session": "fakesession"})
- with pytest.raises(WebSocketDisconnect):
- with client.websocket_connect("/items/foo/ws") as websocket:
- message = "Message one"
- websocket.send_text(message)
- data = websocket.receive_text()
- assert data == "Session cookie or query token value is: fakesession"
- data = websocket.receive_text()
- assert data == f"Message text was: {message}, for item ID: foo"
- message = "Message two"
- websocket.send_text(message)
- data = websocket.receive_text()
- assert data == "Session cookie or query token value is: fakesession"
- data = websocket.receive_text()
- assert data == f"Message text was: {message}, for item ID: foo"
-
-
-def test_websocket_with_header():
- client = TestClient(app)
- with pytest.raises(WebSocketDisconnect):
- with client.websocket_connect("/items/bar/ws?token=some-token") as websocket:
- message = "Message one"
- websocket.send_text(message)
- data = websocket.receive_text()
- assert data == "Session cookie or query token value is: some-token"
- data = websocket.receive_text()
- assert data == f"Message text was: {message}, for item ID: bar"
- message = "Message two"
- websocket.send_text(message)
- data = websocket.receive_text()
- assert data == "Session cookie or query token value is: some-token"
- data = websocket.receive_text()
- assert data == f"Message text was: {message}, for item ID: bar"
-
-
-def test_websocket_with_header_and_query():
- client = TestClient(app)
- with pytest.raises(WebSocketDisconnect):
- with client.websocket_connect("/items/2/ws?q=3&token=some-token") as websocket:
- message = "Message one"
- websocket.send_text(message)
- data = websocket.receive_text()
- assert data == "Session cookie or query token value is: some-token"
- data = websocket.receive_text()
- assert data == "Query parameter q is: 3"
- data = websocket.receive_text()
- assert data == f"Message text was: {message}, for item ID: 2"
- message = "Message two"
- websocket.send_text(message)
- data = websocket.receive_text()
- assert data == "Session cookie or query token value is: some-token"
- data = websocket.receive_text()
- assert data == "Query parameter q is: 3"
- data = websocket.receive_text()
- assert data == f"Message text was: {message}, for item ID: 2"
-
-
-def test_websocket_no_credentials():
- client = TestClient(app)
- with pytest.raises(WebSocketDisconnect):
- with client.websocket_connect("/items/foo/ws"):
- pytest.fail(
- "did not raise WebSocketDisconnect on __enter__"
- ) # pragma: no cover
-
-
-def test_websocket_invalid_data():
- client = TestClient(app)
- with pytest.raises(WebSocketDisconnect):
- with client.websocket_connect("/items/foo/ws?q=bar&token=some-token"):
- pytest.fail(
- "did not raise WebSocketDisconnect on __enter__"
- ) # pragma: no cover
+++ /dev/null
-import pytest
-from fastapi import FastAPI
-from fastapi.testclient import TestClient
-from fastapi.websockets import WebSocketDisconnect
-
-from ...utils import needs_py310
-
-
-@pytest.fixture(name="app")
-def get_app():
- from docs_src.websockets.tutorial002_an_py310 import app
-
- return app
-
-
-@needs_py310
-def test_main(app: FastAPI):
- client = TestClient(app)
- response = client.get("/")
- assert response.status_code == 200, response.text
- assert b"<!DOCTYPE html>" in response.content
-
-
-@needs_py310
-def test_websocket_with_cookie(app: FastAPI):
- client = TestClient(app, cookies={"session": "fakesession"})
- with pytest.raises(WebSocketDisconnect):
- with client.websocket_connect("/items/foo/ws") as websocket:
- message = "Message one"
- websocket.send_text(message)
- data = websocket.receive_text()
- assert data == "Session cookie or query token value is: fakesession"
- data = websocket.receive_text()
- assert data == f"Message text was: {message}, for item ID: foo"
- message = "Message two"
- websocket.send_text(message)
- data = websocket.receive_text()
- assert data == "Session cookie or query token value is: fakesession"
- data = websocket.receive_text()
- assert data == f"Message text was: {message}, for item ID: foo"
-
-
-@needs_py310
-def test_websocket_with_header(app: FastAPI):
- client = TestClient(app)
- with pytest.raises(WebSocketDisconnect):
- with client.websocket_connect("/items/bar/ws?token=some-token") as websocket:
- message = "Message one"
- websocket.send_text(message)
- data = websocket.receive_text()
- assert data == "Session cookie or query token value is: some-token"
- data = websocket.receive_text()
- assert data == f"Message text was: {message}, for item ID: bar"
- message = "Message two"
- websocket.send_text(message)
- data = websocket.receive_text()
- assert data == "Session cookie or query token value is: some-token"
- data = websocket.receive_text()
- assert data == f"Message text was: {message}, for item ID: bar"
-
-
-@needs_py310
-def test_websocket_with_header_and_query(app: FastAPI):
- client = TestClient(app)
- with pytest.raises(WebSocketDisconnect):
- with client.websocket_connect("/items/2/ws?q=3&token=some-token") as websocket:
- message = "Message one"
- websocket.send_text(message)
- data = websocket.receive_text()
- assert data == "Session cookie or query token value is: some-token"
- data = websocket.receive_text()
- assert data == "Query parameter q is: 3"
- data = websocket.receive_text()
- assert data == f"Message text was: {message}, for item ID: 2"
- message = "Message two"
- websocket.send_text(message)
- data = websocket.receive_text()
- assert data == "Session cookie or query token value is: some-token"
- data = websocket.receive_text()
- assert data == "Query parameter q is: 3"
- data = websocket.receive_text()
- assert data == f"Message text was: {message}, for item ID: 2"
-
-
-@needs_py310
-def test_websocket_no_credentials(app: FastAPI):
- client = TestClient(app)
- with pytest.raises(WebSocketDisconnect):
- with client.websocket_connect("/items/foo/ws"):
- pytest.fail(
- "did not raise WebSocketDisconnect on __enter__"
- ) # pragma: no cover
-
-
-@needs_py310
-def test_websocket_invalid_data(app: FastAPI):
- client = TestClient(app)
- with pytest.raises(WebSocketDisconnect):
- with client.websocket_connect("/items/foo/ws?q=bar&token=some-token"):
- pytest.fail(
- "did not raise WebSocketDisconnect on __enter__"
- ) # pragma: no cover
+++ /dev/null
-import pytest
-from fastapi import FastAPI
-from fastapi.testclient import TestClient
-from fastapi.websockets import WebSocketDisconnect
-
-from ...utils import needs_py39
-
-
-@pytest.fixture(name="app")
-def get_app():
- from docs_src.websockets.tutorial002_an_py39 import app
-
- return app
-
-
-@needs_py39
-def test_main(app: FastAPI):
- client = TestClient(app)
- response = client.get("/")
- assert response.status_code == 200, response.text
- assert b"<!DOCTYPE html>" in response.content
-
-
-@needs_py39
-def test_websocket_with_cookie(app: FastAPI):
- client = TestClient(app, cookies={"session": "fakesession"})
- with pytest.raises(WebSocketDisconnect):
- with client.websocket_connect("/items/foo/ws") as websocket:
- message = "Message one"
- websocket.send_text(message)
- data = websocket.receive_text()
- assert data == "Session cookie or query token value is: fakesession"
- data = websocket.receive_text()
- assert data == f"Message text was: {message}, for item ID: foo"
- message = "Message two"
- websocket.send_text(message)
- data = websocket.receive_text()
- assert data == "Session cookie or query token value is: fakesession"
- data = websocket.receive_text()
- assert data == f"Message text was: {message}, for item ID: foo"
-
-
-@needs_py39
-def test_websocket_with_header(app: FastAPI):
- client = TestClient(app)
- with pytest.raises(WebSocketDisconnect):
- with client.websocket_connect("/items/bar/ws?token=some-token") as websocket:
- message = "Message one"
- websocket.send_text(message)
- data = websocket.receive_text()
- assert data == "Session cookie or query token value is: some-token"
- data = websocket.receive_text()
- assert data == f"Message text was: {message}, for item ID: bar"
- message = "Message two"
- websocket.send_text(message)
- data = websocket.receive_text()
- assert data == "Session cookie or query token value is: some-token"
- data = websocket.receive_text()
- assert data == f"Message text was: {message}, for item ID: bar"
-
-
-@needs_py39
-def test_websocket_with_header_and_query(app: FastAPI):
- client = TestClient(app)
- with pytest.raises(WebSocketDisconnect):
- with client.websocket_connect("/items/2/ws?q=3&token=some-token") as websocket:
- message = "Message one"
- websocket.send_text(message)
- data = websocket.receive_text()
- assert data == "Session cookie or query token value is: some-token"
- data = websocket.receive_text()
- assert data == "Query parameter q is: 3"
- data = websocket.receive_text()
- assert data == f"Message text was: {message}, for item ID: 2"
- message = "Message two"
- websocket.send_text(message)
- data = websocket.receive_text()
- assert data == "Session cookie or query token value is: some-token"
- data = websocket.receive_text()
- assert data == "Query parameter q is: 3"
- data = websocket.receive_text()
- assert data == f"Message text was: {message}, for item ID: 2"
-
-
-@needs_py39
-def test_websocket_no_credentials(app: FastAPI):
- client = TestClient(app)
- with pytest.raises(WebSocketDisconnect):
- with client.websocket_connect("/items/foo/ws"):
- pytest.fail(
- "did not raise WebSocketDisconnect on __enter__"
- ) # pragma: no cover
-
-
-@needs_py39
-def test_websocket_invalid_data(app: FastAPI):
- client = TestClient(app)
- with pytest.raises(WebSocketDisconnect):
- with client.websocket_connect("/items/foo/ws?q=bar&token=some-token"):
- pytest.fail(
- "did not raise WebSocketDisconnect on __enter__"
- ) # pragma: no cover
+++ /dev/null
-import pytest
-from fastapi import FastAPI
-from fastapi.testclient import TestClient
-from fastapi.websockets import WebSocketDisconnect
-
-from ...utils import needs_py310
-
-
-@pytest.fixture(name="app")
-def get_app():
- from docs_src.websockets.tutorial002_py310 import app
-
- return app
-
-
-@needs_py310
-def test_main(app: FastAPI):
- client = TestClient(app)
- response = client.get("/")
- assert response.status_code == 200, response.text
- assert b"<!DOCTYPE html>" in response.content
-
-
-@needs_py310
-def test_websocket_with_cookie(app: FastAPI):
- client = TestClient(app, cookies={"session": "fakesession"})
- with pytest.raises(WebSocketDisconnect):
- with client.websocket_connect("/items/foo/ws") as websocket:
- message = "Message one"
- websocket.send_text(message)
- data = websocket.receive_text()
- assert data == "Session cookie or query token value is: fakesession"
- data = websocket.receive_text()
- assert data == f"Message text was: {message}, for item ID: foo"
- message = "Message two"
- websocket.send_text(message)
- data = websocket.receive_text()
- assert data == "Session cookie or query token value is: fakesession"
- data = websocket.receive_text()
- assert data == f"Message text was: {message}, for item ID: foo"
-
-
-@needs_py310
-def test_websocket_with_header(app: FastAPI):
- client = TestClient(app)
- with pytest.raises(WebSocketDisconnect):
- with client.websocket_connect("/items/bar/ws?token=some-token") as websocket:
- message = "Message one"
- websocket.send_text(message)
- data = websocket.receive_text()
- assert data == "Session cookie or query token value is: some-token"
- data = websocket.receive_text()
- assert data == f"Message text was: {message}, for item ID: bar"
- message = "Message two"
- websocket.send_text(message)
- data = websocket.receive_text()
- assert data == "Session cookie or query token value is: some-token"
- data = websocket.receive_text()
- assert data == f"Message text was: {message}, for item ID: bar"
-
-
-@needs_py310
-def test_websocket_with_header_and_query(app: FastAPI):
- client = TestClient(app)
- with pytest.raises(WebSocketDisconnect):
- with client.websocket_connect("/items/2/ws?q=3&token=some-token") as websocket:
- message = "Message one"
- websocket.send_text(message)
- data = websocket.receive_text()
- assert data == "Session cookie or query token value is: some-token"
- data = websocket.receive_text()
- assert data == "Query parameter q is: 3"
- data = websocket.receive_text()
- assert data == f"Message text was: {message}, for item ID: 2"
- message = "Message two"
- websocket.send_text(message)
- data = websocket.receive_text()
- assert data == "Session cookie or query token value is: some-token"
- data = websocket.receive_text()
- assert data == "Query parameter q is: 3"
- data = websocket.receive_text()
- assert data == f"Message text was: {message}, for item ID: 2"
-
-
-@needs_py310
-def test_websocket_no_credentials(app: FastAPI):
- client = TestClient(app)
- with pytest.raises(WebSocketDisconnect):
- with client.websocket_connect("/items/foo/ws"):
- pytest.fail(
- "did not raise WebSocketDisconnect on __enter__"
- ) # pragma: no cover
-
-
-@needs_py310
-def test_websocket_invalid_data(app: FastAPI):
- client = TestClient(app)
- with pytest.raises(WebSocketDisconnect):
- with client.websocket_connect("/items/foo/ws?q=bar&token=some-token"):
- pytest.fail(
- "did not raise WebSocketDisconnect on __enter__"
- ) # pragma: no cover
projects / thirdparty / fastapi / fastapi.git / commitdiff
