Coverage for tests/test_dependency_class.py: 100%

1from typing import AsyncGenerator, Generator 6 ctx1abcdef

2 

3import pytest 6 ctx1abcdef

4from fastapi import Depends, FastAPI 6 ctx1abcdef

5from fastapi.testclient import TestClient 6 ctx1abcdef

6 

7app = FastAPI() 6 ctx1abcdef

8 

9 

10class CallableDependency: 6 ctx1abcdef

11 def __call__(self, value: str) -> str: 6 ctx1abcdef

12 return value 6 ctx1ghijkl

13 

14 

15class CallableGenDependency: 6 ctx1abcdef

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

17 yield value 6 ctx1ghijkl

18 

19 

20class AsyncCallableDependency: 6 ctx1abcdef

21 async def __call__(self, value: str) -> str: 6 ctx1abcdef

22 return value 6 ctx1ghijkl

23 

24 

25class AsyncCallableGenDependency: 6 ctx1abcdef

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

27 yield value 6 ctx1ghijkl

28 

29 

30class MethodsDependency: 6 ctx1abcdef

31 def synchronous(self, value: str) -> str: 6 ctx1abcdef

32 return value 6 ctx1ghijkl

33 

34 async def asynchronous(self, value: str) -> str: 6 ctx1abcdef

35 return value 6 ctx1ghijkl

36 

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

38 yield value 6 ctx1ghijkl

39 

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

41 yield value 6 ctx1ghijkl

42 

43 

44callable_dependency = CallableDependency() 6 ctx1abcdef

45callable_gen_dependency = CallableGenDependency() 6 ctx1abcdef

46async_callable_dependency = AsyncCallableDependency() 6 ctx1abcdef

47async_callable_gen_dependency = AsyncCallableGenDependency() 6 ctx1abcdef

48methods_dependency = MethodsDependency() 6 ctx1abcdef

49 

50 

51@app.get("/callable-dependency") 6 ctx1abcdef

52async def get_callable_dependency(value: str = Depends(callable_dependency)): 6 ctx1abcdef

53 return value 6 ctx1ghijkl

54 

55 

56@app.get("/callable-gen-dependency") 6 ctx1abcdef

57async def get_callable_gen_dependency(value: str = Depends(callable_gen_dependency)): 6 ctx1abcdef

58 return value 6 ctx1ghijkl

59 

60 

61@app.get("/async-callable-dependency") 6 ctx1abcdef

62async def get_async_callable_dependency( 6 ctx1abcdef

63 value: str = Depends(async_callable_dependency), 

64): 

65 return value 6 ctx1ghijkl

66 

67 

68@app.get("/async-callable-gen-dependency") 6 ctx1abcdef

69async def get_async_callable_gen_dependency( 6 ctx1abcdef

70 value: str = Depends(async_callable_gen_dependency), 

71): 

72 return value 6 ctx1ghijkl

73 

74 

75@app.get("/synchronous-method-dependency") 6 ctx1abcdef

76async def get_synchronous_method_dependency( 6 ctx1abcdef

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

78): 

79 return value 6 ctx1ghijkl

80 

81 

82@app.get("/synchronous-method-gen-dependency") 6 ctx1abcdef

83async def get_synchronous_method_gen_dependency( 6 ctx1abcdef

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

85): 

86 return value 6 ctx1ghijkl

87 

88 

89@app.get("/asynchronous-method-dependency") 6 ctx1abcdef

90async def get_asynchronous_method_dependency( 6 ctx1abcdef

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

92): 

93 return value 6 ctx1ghijkl

94 

95 

96@app.get("/asynchronous-method-gen-dependency") 6 ctx1abcdef

97async def get_asynchronous_method_gen_dependency( 6 ctx1abcdef

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

99): 

100 return value 6 ctx1ghijkl

101 

102 

103client = TestClient(app) 6 ctx1abcdef

104 

105 

106@pytest.mark.parametrize( 6 ctx1abcdef

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): 6 ctx1abcdef

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

121 assert response.status_code == 200, response.text 6 ctx1ghijkl

122 assert response.json() == value 6 ctx1ghijkl