✅ Simplify tests for background_tasks (#13166)

tests/test_tutorial/test_background_tasks/test_tutorial002.py

@@ -1,14 +1,31 @@
+import importlib
 import os
 from pathlib import Path
 
+import pytest
 from fastapi.testclient import TestClient
 
-from docs_src.background_tasks.tutorial002 import app
+from ...utils import needs_py39, needs_py310
-
-client = TestClient(app)
 
 
-def test():
+@pytest.fixture(
+    name="client",
+    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_client(request: pytest.FixtureRequest):
+    mod = importlib.import_module(f"docs_src.background_tasks.{request.param}")
+
+    client = TestClient(mod.app)
+    return client
+
+
+def test(client: TestClient):
     log = Path("log.txt")
     if log.is_file():
         os.remove(log)  # pragma: no cover

tests/test_tutorial/test_background_tasks/test_tutorial002_an.py

@@ -1,19 +0,0 @@
-import os
-from pathlib import Path
-
-from fastapi.testclient import TestClient
-
-from docs_src.background_tasks.tutorial002_an import app
-
-client = TestClient(app)
-
-
-def test():
-    log = Path("log.txt")
-    if log.is_file():
-        os.remove(log)  # pragma: no cover
-    response = client.post("/send-notification/foo@example.com?q=some-query")
-    assert response.status_code == 200, response.text
-    assert response.json() == {"message": "Message sent"}
-    with open("./log.txt") as f:
-        assert "found query: some-query\nmessage to foo@example.com" in f.read()

tests/test_tutorial/test_background_tasks/test_tutorial002_an_py310.py

@@ -1,21 +0,0 @@
-import os
-from pathlib import Path
-
-from fastapi.testclient import TestClient
-
-from ...utils import needs_py310
-
-
-@needs_py310
-def test():
-    from docs_src.background_tasks.tutorial002_an_py310 import app
-
-    client = TestClient(app)
-    log = Path("log.txt")
-    if log.is_file():
-        os.remove(log)  # pragma: no cover
-    response = client.post("/send-notification/foo@example.com?q=some-query")
-    assert response.status_code == 200, response.text
-    assert response.json() == {"message": "Message sent"}
-    with open("./log.txt") as f:
-        assert "found query: some-query\nmessage to foo@example.com" in f.read()

tests/test_tutorial/test_background_tasks/test_tutorial002_an_py39.py

@@ -1,21 +0,0 @@
-import os
-from pathlib import Path
-
-from fastapi.testclient import TestClient
-
-from ...utils import needs_py39
-
-
-@needs_py39
-def test():
-    from docs_src.background_tasks.tutorial002_an_py39 import app
-
-    client = TestClient(app)
-    log = Path("log.txt")
-    if log.is_file():
-        os.remove(log)  # pragma: no cover
-    response = client.post("/send-notification/foo@example.com?q=some-query")
-    assert response.status_code == 200, response.text
-    assert response.json() == {"message": "Message sent"}
-    with open("./log.txt") as f:
-        assert "found query: some-query\nmessage to foo@example.com" in f.read()

tests/test_tutorial/test_background_tasks/test_tutorial002_py310.py

@@ -1,21 +0,0 @@
-import os
-from pathlib import Path
-
-from fastapi.testclient import TestClient
-
-from ...utils import needs_py310
-
-
-@needs_py310
-def test():
-    from docs_src.background_tasks.tutorial002_py310 import app
-
-    client = TestClient(app)
-    log = Path("log.txt")
-    if log.is_file():
-        os.remove(log)  # pragma: no cover
-    response = client.post("/send-notification/foo@example.com?q=some-query")
-    assert response.status_code == 200, response.text
-    assert response.json() == {"message": "Message sent"}
-    with open("./log.txt") as f:
-        assert "found query: some-query\nmessage to foo@example.com" in f.read()