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

1from fastapi.testclient import TestClient 5 ctx1abcde

2 

3from docs_src.cors.tutorial001 import app 5 ctx1abcde

4 

5 

6def test_cors(): 5 ctx1abcde

7 client = TestClient(app) 5 ctx1abcde

8 # Test pre-flight response 

9 headers = { 5 ctx1abcde

10 "Origin": "https://localhost.tiangolo.com", 

11 "Access-Control-Request-Method": "GET", 

12 "Access-Control-Request-Headers": "X-Example", 

13 } 

14 response = client.options("/", headers=headers) 5 ctx1abcde

15 assert response.status_code == 200, response.text 5 ctx1abcde

16 assert response.text == "OK" 5 ctx1abcde

17 assert ( 5 ctx1abcde

18 response.headers["access-control-allow-origin"] 

19 == "https://localhost.tiangolo.com" 

20 ) 

21 assert response.headers["access-control-allow-headers"] == "X-Example" 5 ctx1abcde

22 

23 # Test standard response 

24 headers = {"Origin": "https://localhost.tiangolo.com"} 5 ctx1abcde

25 response = client.get("/", headers=headers) 5 ctx1abcde

26 assert response.status_code == 200, response.text 5 ctx1abcde

27 assert response.json() == {"message": "Hello World"} 5 ctx1abcde

28 assert ( 5 ctx1abcde

29 response.headers["access-control-allow-origin"] 

30 == "https://localhost.tiangolo.com" 

31 ) 

32 

33 # Test non-CORS response 

34 response = client.get("/") 5 ctx1abcde

35 assert response.status_code == 200, response.text 5 ctx1abcde

36 assert response.json() == {"message": "Hello World"} 5 ctx1abcde

37 assert "access-control-allow-origin" not in response.headers 5 ctx1abcde