Coverage for tests/test_dependency_class.py: 100%

1from typing import AsyncGenerator, Generator 5 ctx1abcde

2 

3import pytest 5 ctx1abcde

4from fastapi import Depends, FastAPI 5 ctx1abcde

5from fastapi.testclient import TestClient 5 ctx1abcde

6 

7app = FastAPI() 5 ctx1abcde

8 

9 

10class CallableDependency: 5 ctx1abcde

11 def __call__(self, value: str) -> str: 5 ctx1abcde

12 return value 5 ctx1abcde

13 

14 

15class CallableGenDependency: 5 ctx1abcde

16 def __call__(self, value: str) -> Generator[str, None, None]: 5 ctx1abcde

17 yield value 5 ctx1abcde

18 

19 

20class AsyncCallableDependency: 5 ctx1abcde

21 async def __call__(self, value: str) -> str: 5 ctx1abcde

22 return value 5 ctx1abcde

23 

24 

25class AsyncCallableGenDependency: 5 ctx1abcde

26 async def __call__(self, value: str) -> AsyncGenerator[str, None]: 5 ctx1abcde

27 yield value 5 ctx1abcde

28 

29 

30class MethodsDependency: 5 ctx1abcde

31 def synchronous(self, value: str) -> str: 5 ctx1abcde

32 return value 5 ctx1abcde

33 

34 async def asynchronous(self, value: str) -> str: 5 ctx1abcde

35 return value 5 ctx1abcde

36 

37 def synchronous_gen(self, value: str) -> Generator[str, None, None]: 5 ctx1abcde

38 yield value 5 ctx1abcde

39 

40 async def asynchronous_gen(self, value: str) -> AsyncGenerator[str, None]: 5 ctx1abcde

41 yield value 5 ctx1abcde

42 

43 

44callable_dependency = CallableDependency() 5 ctx1abcde

45callable_gen_dependency = CallableGenDependency() 5 ctx1abcde

46async_callable_dependency = AsyncCallableDependency() 5 ctx1abcde

47async_callable_gen_dependency = AsyncCallableGenDependency() 5 ctx1abcde

48methods_dependency = MethodsDependency() 5 ctx1abcde

49 

50 

51@app.get("/callable-dependency") 5 ctx1abcde

52async def get_callable_dependency(value: str = Depends(callable_dependency)): 5 ctx1abcde

53 return value 5 ctx1abcde

54 

55 

56@app.get("/callable-gen-dependency") 5 ctx1abcde

57async def get_callable_gen_dependency(value: str = Depends(callable_gen_dependency)): 5 ctx1abcde

58 return value 5 ctx1abcde

59 

60 

61@app.get("/async-callable-dependency") 5 ctx1abcde

62async def get_async_callable_dependency( 5 ctx1abcde

63 value: str = Depends(async_callable_dependency), 

64): 

65 return value 5 ctx1abcde

66 

67 

68@app.get("/async-callable-gen-dependency") 5 ctx1abcde

69async def get_async_callable_gen_dependency( 5 ctx1abcde

70 value: str = Depends(async_callable_gen_dependency), 

71): 

72 return value 5 ctx1abcde

73 

74 

75@app.get("/synchronous-method-dependency") 5 ctx1abcde

76async def get_synchronous_method_dependency( 5 ctx1abcde

77 value: str = Depends(methods_dependency.synchronous), 

78): 

79 return value 5 ctx1abcde

80 

81 

82@app.get("/synchronous-method-gen-dependency") 5 ctx1abcde

83async def get_synchronous_method_gen_dependency( 5 ctx1abcde

84 value: str = Depends(methods_dependency.synchronous_gen), 

85): 

86 return value 5 ctx1abcde

87 

88 

89@app.get("/asynchronous-method-dependency") 5 ctx1abcde

90async def get_asynchronous_method_dependency( 5 ctx1abcde

91 value: str = Depends(methods_dependency.asynchronous), 

92): 

93 return value 5 ctx1abcde

94 

95 

96@app.get("/asynchronous-method-gen-dependency") 5 ctx1abcde

97async def get_asynchronous_method_gen_dependency( 5 ctx1abcde

98 value: str = Depends(methods_dependency.asynchronous_gen), 

99): 

100 return value 5 ctx1abcde

101 

102 

103client = TestClient(app) 5 ctx1abcde

104 

105 

106@pytest.mark.parametrize( 5 ctx1abcde

107 "route,value", 

108 [ 

109 ("/callable-dependency", "callable-dependency"), 

110 ("/callable-gen-dependency", "callable-gen-dependency"), 

111 ("/async-callable-dependency", "async-callable-dependency"), 

112 ("/async-callable-gen-dependency", "async-callable-gen-dependency"), 

113 ("/synchronous-method-dependency", "synchronous-method-dependency"), 

114 ("/synchronous-method-gen-dependency", "synchronous-method-gen-dependency"), 

115 ("/asynchronous-method-dependency", "asynchronous-method-dependency"), 

116 ("/asynchronous-method-gen-dependency", "asynchronous-method-gen-dependency"), 

117 ], 

118) 

119def test_class_dependency(route, value): 5 ctx1abcde

120 response = client.get(route, params={"value": value}) 5 ctx1abcde

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

122 assert response.json() == value 5 ctx1abcde