Coverage for tests/test_custom_swagger_ui_redirect.py: 100%

1from fastapi import FastAPI 7 ctx1abcdefg

2from fastapi.testclient import TestClient 7 ctx1abcdefg

3 

4swagger_ui_oauth2_redirect_url = "/docs/redirect" 7 ctx1abcdefg

5 

6app = FastAPI(swagger_ui_oauth2_redirect_url=swagger_ui_oauth2_redirect_url) 7 ctx1abcdefg

7 

8 

9@app.get("/items/") 7 ctx1abcdefg

10async def read_items(): 7 ctx1abcdefg

11 return {"id": "foo"} 7 ctx1vwxyzAB

12 

13 

14client = TestClient(app) 7 ctx1abcdefg

15 

16 

17def test_swagger_ui(): 7 ctx1abcdefg

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

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

20 assert response.headers["content-type"] == "text/html; charset=utf-8" 7 ctx1hijklmn

21 assert "swagger-ui-dist" in response.text 7 ctx1hijklmn

22 print(client.base_url) 7 ctx1hijklmn

23 assert ( 7 ctx1hijklmn

24 f"oauth2RedirectUrl: window.location.origin + '{swagger_ui_oauth2_redirect_url}'" 

25 in response.text 

26 ) 

27 

28 

29def test_swagger_ui_oauth2_redirect(): 7 ctx1abcdefg

30 response = client.get(swagger_ui_oauth2_redirect_url) 7 ctx1opqrstu

31 assert response.status_code == 200, response.text 7 ctx1opqrstu

32 assert response.headers["content-type"] == "text/html; charset=utf-8" 7 ctx1opqrstu

33 assert "window.opener.swaggerUIRedirectOauth2" in response.text 7 ctx1opqrstu

34 

35 

36def test_response(): 7 ctx1abcdefg

37 response = client.get("/items/") 7 ctx1vwxyzAB

38 assert response.json() == {"id": "foo"} 7 ctx1vwxyzAB