Coverage for tests/test_tutorial/test_background_tasks/test_tutorial002.py: 100%

1import importlib 5 ctx1abcde

2import os 5 ctx1abcde

3from pathlib import Path 5 ctx1abcde

4 

5import pytest 5 ctx1abcde

6from fastapi.testclient import TestClient 5 ctx1abcde

7 

8from ...utils import needs_py39, needs_py310 5 ctx1abcde

9 

10 

11@pytest.fixture( 5 ctx1abcde

12 name="client", 

13 params=[ 

14 "tutorial002", 

15 pytest.param("tutorial002_py310", marks=needs_py310), 

16 "tutorial002_an", 

17 pytest.param("tutorial002_an_py39", marks=needs_py39), 

18 pytest.param("tutorial002_an_py310", marks=needs_py310), 

19 ], 

20) 

21def get_client(request: pytest.FixtureRequest): 5 ctx1abcde

22 mod = importlib.import_module(f"docs_src.background_tasks.{request.param}") 5 ctx1abcde

23 

24 client = TestClient(mod.app) 5 ctx1abcde

25 return client 5 ctx1abcde

26 

27 

28def test(client: TestClient): 5 ctx1abcde

29 log = Path("log.txt") 5 ctx1fghij

30 if log.is_file(): 5 ctx1fghij

31 os.remove(log) # pragma: no cover 5 ctx1fghij

32 response = client.post("/send-notification/foo@example.com?q=some-query") 5 ctx1fghij

33 assert response.status_code == 200, response.text 5 ctx1fghij

34 assert response.json() == {"message": "Message sent"} 5 ctx1fghij

35 with open("./log.txt") as f: 5 ctx1fghij

36 assert "found query: some-query\nmessage to foo@example.com" in f.read() 5 ctx1fghij