Skip to content

Commit 66b1bb0

Browse files
Simplify test (#146)
1 parent c5e5404 commit 66b1bb0

1 file changed

Lines changed: 3 additions & 19 deletions

File tree

betterproto2/tests/grpc/test_grpclib_client.py

Lines changed: 3 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -153,11 +153,7 @@ async def test_service_call_lower_level_with_overrides(requires_grpclib, require
153153
import grpclib.metadata
154154
from grpclib.testing import ChannelFor
155155

156-
from tests.outputs.service.service import (
157-
DoThingRequest,
158-
DoThingResponse,
159-
TestStub as ThingServiceClient,
160-
)
156+
from tests.outputs.service.service import DoThingRequest, TestStub as ThingServiceClient
161157

162158
from .thing_service import ThingService
163159

@@ -170,13 +166,7 @@ async def test_service_call_lower_level_with_overrides(requires_grpclib, require
170166
kwarg_metadata = {"authorization": "12345"}
171167
async with ChannelFor([ThingService(test_hook=_assert_request_meta_received(deadline, metadata))]) as channel:
172168
client = ThingServiceClient(channel, deadline=deadline, metadata=metadata)
173-
response = await client._unary_unary(
174-
"/service.Test/DoThing",
175-
DoThingRequest(THING_TO_DO),
176-
DoThingResponse,
177-
deadline=kwarg_deadline,
178-
metadata=kwarg_metadata,
179-
)
169+
response = await client.do_thing(DoThingRequest(THING_TO_DO), deadline=kwarg_deadline, metadata=kwarg_metadata)
180170
assert response.names == [THING_TO_DO]
181171

182172
# Setting timeout
@@ -194,13 +184,7 @@ async def test_service_call_lower_level_with_overrides(requires_grpclib, require
194184
]
195185
) as channel:
196186
client = ThingServiceClient(channel, deadline=deadline, metadata=metadata)
197-
response = await client._unary_unary(
198-
"/service.Test/DoThing",
199-
DoThingRequest(THING_TO_DO),
200-
DoThingResponse,
201-
timeout=kwarg_timeout,
202-
metadata=kwarg_metadata,
203-
)
187+
response = await client.do_thing(DoThingRequest(THING_TO_DO), timeout=kwarg_timeout, metadata=kwarg_metadata)
204188
assert response.names == [THING_TO_DO]
205189

206190

0 commit comments

Comments
 (0)