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

1import importlib 4 ctx1lmno

2 

3from fastapi.testclient import TestClient 4 ctx1lmno

4from inline_snapshot import snapshot 4 ctx1lmno

5 

6 

7def get_client() -> TestClient: 4 ctx1lmno

8 from docs_src.conditional_openapi import tutorial001_py310 11 ctx1eahbifcjgdk

9 

10 importlib.reload(tutorial001_py310) 11 ctx1eahbifcjgdk

11 

12 client = TestClient(tutorial001_py310.app) 11 ctx1eahbifcjgdk

13 return client 11 ctx1eahbifcjgdk

14 

15 

16def test_disable_openapi(monkeypatch): 4 ctx1lmno

17 monkeypatch.setenv("OPENAPI_URL", "") 4 ctx1abcd

18 # Load the client after setting the env var 

19 client = get_client() 4 ctx1abcd

20 response = client.get("/openapi.json") 4 ctx1abcd

21 assert response.status_code == 404, response.text 4 ctx1abcd

22 response = client.get("/docs") 4 ctx1abcd

23 assert response.status_code == 404, response.text 4 ctx1abcd

24 response = client.get("/redoc") 4 ctx1abcd

25 assert response.status_code == 404, response.text 4 ctx1abcd

26 

27 

28def test_root(): 4 ctx1lmno

29 client = get_client() 4 ctx1hijk

30 response = client.get("/") 4 ctx1hijk

31 assert response.status_code == 200 4 ctx1hijk

32 assert response.json() == {"message": "Hello World"} 4 ctx1hijk

33 

34 

35def test_default_openapi(): 4 ctx1lmno

36 client = get_client() 4 ctx1epfg

37 response = client.get("/docs") 4 ctx1epfg

38 assert response.status_code == 200, response.text 4 ctx1epfg

39 response = client.get("/redoc") 4 ctx1epfg

40 assert response.status_code == 200, response.text 4 ctx1epfg

41 response = client.get("/openapi.json") 4 ctx1epfg

42 assert response.json() == snapshot( 4 ctx1epfg

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 )