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

1import importlib 6 ctx1abcdef

2import os 6 ctx1abcdef

3from pathlib import Path 6 ctx1abcdef

4 

5import pytest 6 ctx1abcdef

6from fastapi.testclient import TestClient 6 ctx1abcdef

7 

8from ...utils import needs_py39, needs_py310 6 ctx1abcdef

9 

10 

11@pytest.fixture( 6 ctx1abcdef

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): 6 ctx1abcdef

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

23 

24 client = TestClient(mod.app) 6 ctx1abcdef

25 return client 6 ctx1abcdef

26 

27 

28def test(client: TestClient): 6 ctx1abcdef

29 log = Path("log.txt") 6 ctx1ghijkl

30 if log.is_file(): 6 ctx1ghijkl

31 os.remove(log) # pragma: no cover 6 ctx1ghijkl

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

33 assert response.status_code == 200, response.text 6 ctx1ghijkl

34 assert response.json() == {"message": "Message sent"} 6 ctx1ghijkl

35 with open("./log.txt") as f: 6 ctx1ghijkl

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