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

1import importlib 4 ctx1jklm

2 

3from fastapi.testclient import TestClient 4 ctx1jklm

4from inline_snapshot import snapshot 4 ctx1jklm

5 

6 

7def get_client() -> TestClient: 4 ctx1jklm

8 from docs_src.conditional_openapi import tutorial001_py310 9 ctx1dagebhfci

9 

10 importlib.reload(tutorial001_py310) 9 ctx1dagebhfci

11 

12 client = TestClient(tutorial001_py310.app) 9 ctx1dagebhfci

13 return client 9 ctx1dagebhfci

14 

15 

16def test_disable_openapi(monkeypatch): 4 ctx1jklm

17 monkeypatch.setenv("OPENAPI_URL", "") 3 ctx1abc

18 # Load the client after setting the env var 

19 client = get_client() 3 ctx1abc

20 response = client.get("/openapi.json") 3 ctx1abc

21 assert response.status_code == 404, response.text 3 ctx1abc

22 response = client.get("/docs") 3 ctx1abc

23 assert response.status_code == 404, response.text 3 ctx1abc

24 response = client.get("/redoc") 3 ctx1abc

25 assert response.status_code == 404, response.text 3 ctx1abc

26 

27 

28def test_root(): 4 ctx1jklm

29 client = get_client() 3 ctx1ghi

30 response = client.get("/") 3 ctx1ghi

31 assert response.status_code == 200 3 ctx1ghi

32 assert response.json() == {"message": "Hello World"} 3 ctx1ghi

33 

34 

35def test_default_openapi(): 4 ctx1jklm

36 client = get_client() 3 ctx1def

37 response = client.get("/docs") 3 ctx1def

38 assert response.status_code == 200, response.text 3 ctx1def

39 response = client.get("/redoc") 3 ctx1def

40 assert response.status_code == 200, response.text 3 ctx1def

41 response = client.get("/openapi.json") 3 ctx1def

42 assert response.json() == snapshot( 3 ctx1def

43 { 

44 "openapi": "3.1.0", 

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

46 "paths": { 

47 "/": { 

48 "get": { 

49 "summary": "Root", 

50 "operationId": "root__get", 

51 "responses": { 

52 "200": { 

53 "description": "Successful Response", 

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

55 } 

56 }, 

57 } 

58 } 

59 }, 

60 } 

61 )