6
6
7
7
from testsuite .databases .pgsql import discover
8
8
9
- USERVER_CONFIG_HOOKS = ['_prepare_service_config' ]
9
+ import handlers .hello_pb2_grpc as hello_services # noqa: E402, E501
10
+
11
+ USERVER_CONFIG_HOOKS = ['prepare_service_config' ]
10
12
pytest_plugins = [
11
13
'pytest_userver.plugins.postgresql' ,
12
14
'pytest_userver.plugins.grpc' ,
13
15
]
14
16
15
17
16
- @pytest .fixture (scope = 'session' )
17
- def hello_protos ():
18
- return grpc .protos ('hello.proto' )
19
-
20
-
21
- @pytest .fixture (scope = 'session' )
22
- def hello_services ():
23
- return grpc .services ('hello.proto' )
24
-
25
-
26
18
@pytest .fixture
27
- def grpc_service (pgsql , hello_services , grpc_channel , service_client ):
19
+ def grpc_service (pgsql , grpc_channel , service_client ):
28
20
return hello_services .HelloServiceStub (grpc_channel )
29
21
30
22
31
23
@pytest .fixture (scope = 'session' )
32
- def mock_grpc_hello_session (
33
- hello_services , grpc_mockserver , create_grpc_mock ,
34
- ):
24
+ def mock_grpc_hello_session (grpc_mockserver , create_grpc_mock ):
35
25
mock = create_grpc_mock (hello_services .HelloServiceServicer )
36
26
hello_services .add_HelloServiceServicer_to_server (
37
27
mock .servicer , grpc_mockserver ,
@@ -46,7 +36,7 @@ def mock_grpc_server(mock_grpc_hello_session):
46
36
47
37
48
38
@pytest .fixture (scope = 'session' )
49
- def _prepare_service_config (grpc_mockserver_endpoint ):
39
+ def prepare_service_config (grpc_mockserver_endpoint ):
50
40
def patch_config (config , config_vars ):
51
41
components = config ['components_manager' ]['components' ]
52
42
components ['hello-client' ]['endpoint' ] = grpc_mockserver_endpoint
0 commit comments