Coverage for tests/test_tutorial/test_conditional_openapi/test_tutorial001.py: 100%

1import importlib 7 ctx1opqrstu

2 

3from fastapi.testclient import TestClient 7 ctx1opqrstu

4 

5from ...utils import needs_pydanticv2 7 ctx1opqrstu

6 

7 

8def get_client() -> TestClient: 7 ctx1opqrstu

9 from docs_src.conditional_openapi import tutorial001 21 ctx1havibwjcxkdylezmfAngB

10 

11 importlib.reload(tutorial001) 21 ctx1havibwjcxkdylezmfAngB

12 

13 client = TestClient(tutorial001.app) 21 ctx1havibwjcxkdylezmfAngB

14 return client 21 ctx1havibwjcxkdylezmfAngB

15 

16 

17@needs_pydanticv2 7 ctx1opqrstu

18def test_disable_openapi(monkeypatch): 7 ctx1opqrstu

19 monkeypatch.setenv("OPENAPI_URL", "") 7 ctx1abcdefg

20 # Load the client after setting the env var 

21 client = get_client() 7 ctx1abcdefg

22 response = client.get("/openapi.json") 7 ctx1abcdefg

23 assert response.status_code == 404, response.text 7 ctx1abcdefg

24 response = client.get("/docs") 7 ctx1abcdefg

25 assert response.status_code == 404, response.text 7 ctx1abcdefg

26 response = client.get("/redoc") 7 ctx1abcdefg

27 assert response.status_code == 404, response.text 7 ctx1abcdefg

28 

29 

30@needs_pydanticv2 7 ctx1opqrstu

31def test_root(): 7 ctx1opqrstu

32 client = get_client() 7 ctx1vwxyzAB

33 response = client.get("/") 7 ctx1vwxyzAB

34 assert response.status_code == 200 7 ctx1vwxyzAB

35 assert response.json() == {"message": "Hello World"} 7 ctx1vwxyzAB

36 

37 

38@needs_pydanticv2 7 ctx1opqrstu

39def test_default_openapi(): 7 ctx1opqrstu

40 client = get_client() 7 ctx1hijklmn

41 response = client.get("/docs") 7 ctx1hijklmn

42 assert response.status_code == 200, response.text 7 ctx1hijklmn

43 response = client.get("/redoc") 7 ctx1hijklmn

44 assert response.status_code == 200, response.text 7 ctx1hijklmn

45 response = client.get("/openapi.json") 7 ctx1hijklmn

46 assert response.json() == { 7 ctx1hijklmn

47 "openapi": "3.1.0", 

48 "info": {"title": "FastAPI", "version": "0.1.0"}, 

49 "paths": { 

50 "/": { 

51 "get": { 

52 "summary": "Root", 

53 "operationId": "root__get", 

54 "responses": { 

55 "200": { 

56 "description": "Successful Response", 

57 "content": {"application/json": {"schema": {}}}, 

58 } 

59 }, 

60 } 

61 } 

62 }, 

63 }