File tree Expand file tree Collapse file tree 1 file changed +4
-5
lines changed
tests/opentelemetry-test-utils/src/opentelemetry/test Expand file tree Collapse file tree 1 file changed +4
-5
lines changed Original file line number Diff line number Diff line change @@ -100,15 +100,14 @@ async def send_input(self, message):
100
100
async def send_default_request (self ):
101
101
await self .send_input ({"type" : "http.request" , "body" : b"" })
102
102
103
- async def get_output (self ):
104
- output = await self .communicator .receive_output (0 )
105
- return output
103
+ async def get_output (self , timeout = 1 ):
104
+ return await self .communicator .receive_output (timeout )
106
105
107
- async def get_all_output (self ):
106
+ async def get_all_output (self , timeout = 1 ):
108
107
outputs = []
109
108
while True :
110
109
try :
111
- outputs .append (await self .get_output ( ))
110
+ outputs .append (await self .communicator . receive_output ( timeout ))
112
111
except asyncio .TimeoutError :
113
112
break
114
113
return outputs
You can’t perform that action at this time.
0 commit comments