44from hello_legacy_grpc .v1 import greeter_pb2 , greeter_pb2_grpc
55from hello_legacy_grpc .v1 .greeter_rbt import RebootGreeter
66from proxy_greeter_servicer import ProxyGreeterServicer
7+ from reboot .aio .applications import Application
78from reboot .aio .tests import Reboot
89from reboot_greeter_servicer import RebootGreeterServicer
910
@@ -19,8 +20,10 @@ async def asyncTearDown(self) -> None:
1920
2021 async def test_reboot_greeter (self ) -> None :
2122 await self .rbt .up (
22- servicers = [RebootGreeterServicer ],
23- legacy_grpc_servicers = [DeprecatedGreeterServicer ],
23+ Application (
24+ servicers = [RebootGreeterServicer ],
25+ legacy_grpc_servicers = [DeprecatedGreeterServicer ],
26+ ),
2427 )
2528
2629 context = self .rbt .create_external_context (name = f"test-{ self .id ()} " )
@@ -46,7 +49,9 @@ async def test_reboot_greeter(self) -> None:
4649 )
4750
4851 async def test_deprecated_greeter (self ) -> None :
49- await self .rbt .up (legacy_grpc_servicers = [DeprecatedGreeterServicer ])
52+ await self .rbt .up (
53+ Application (legacy_grpc_servicers = [DeprecatedGreeterServicer ])
54+ )
5055
5156 context = self .rbt .create_external_context (name = f"test-{ self .id ()} " )
5257
@@ -66,10 +71,13 @@ async def test_deprecated_greeter(self) -> None:
6671 )
6772 async def test_proxy_greeter (self ) -> None :
6873 await self .rbt .up (
69- servicers = [RebootGreeterServicer ],
70- legacy_grpc_servicers = [
71- DeprecatedGreeterServicer , ProxyGreeterServicer
72- ],
74+ Application (
75+ servicers = [RebootGreeterServicer ],
76+ legacy_grpc_servicers = [
77+ DeprecatedGreeterServicer ,
78+ ProxyGreeterServicer ,
79+ ],
80+ ),
7381 local_envoy = True ,
7482 )
7583
0 commit comments