Skip to content

Commit 921f20b

Browse files
committed
address review feedback
1 parent 0f68169 commit 921f20b

File tree

1 file changed

+18
-18
lines changed

1 file changed

+18
-18
lines changed

tests/system/test_streams.py

Lines changed: 18 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -19,14 +19,14 @@
1919
from google import showcase
2020

2121

22-
metadata = (("showcase-trailer", "hello world"),)
22+
_METADATA = (("showcase-trailer", "hello world"),)
2323

2424

2525
def test_unary_stream(echo):
2626
content = 'The hail in Wales falls mainly on the snails.'
2727
responses = echo.expand({
2828
'content': content,
29-
}, metadata=metadata)
29+
}, metadata=_METADATA)
3030

3131
# Consume the response and ensure it matches what we expect.
3232
# with pytest.raises(exceptions.NotFound) as exc:
@@ -38,11 +38,11 @@ def test_unary_stream(echo):
3838
(metadata.key, metadata.value)
3939
for metadata in responses.trailing_metadata()
4040
]
41-
assert metadata[0] in response_metadata
41+
assert _METADATA[0] in response_metadata
4242
else:
43-
showcase_header = f"X-Showcase-Request-{metadata[0][0]}"
43+
showcase_header = f"X-Showcase-Request-{_METADATA[0][0]}"
4444
assert showcase_header in responses._response.headers
45-
assert responses._response.headers[showcase_header] == metadata[0][1]
45+
assert responses._response.headers[showcase_header] == _METADATA[0][1]
4646

4747

4848
def test_stream_unary(echo):
@@ -75,7 +75,7 @@ def test_stream_stream(echo):
7575
requests = []
7676
requests.append(showcase.EchoRequest(content="hello"))
7777
requests.append(showcase.EchoRequest(content="world!"))
78-
responses = echo.chat(iter(requests), metadata=metadata)
78+
responses = echo.chat(iter(requests), metadata=_METADATA)
7979

8080
contents = []
8181
for response in responses:
@@ -86,7 +86,7 @@ def test_stream_stream(echo):
8686
(metadata.key, metadata.value)
8787
for metadata in responses.trailing_metadata()
8888
]
89-
assert metadata[0] in response_metadata
89+
assert _METADATA[0] in response_metadata
9090

9191

9292
def test_stream_stream_passing_dict(echo):
@@ -95,7 +95,7 @@ def test_stream_stream_passing_dict(echo):
9595
return
9696

9797
requests = [{'content': 'hello'}, {'content': 'world!'}]
98-
responses = echo.chat(iter(requests), metadata=metadata)
98+
responses = echo.chat(iter(requests), metadata=_METADATA)
9999

100100
contents = []
101101
for response in responses:
@@ -106,7 +106,7 @@ def test_stream_stream_passing_dict(echo):
106106
(metadata.key, metadata.value)
107107
for metadata in responses.trailing_metadata()
108108
]
109-
assert metadata[0] in response_metadata
109+
assert _METADATA[0] in response_metadata
110110

111111

112112
if os.environ.get("GAPIC_PYTHON_ASYNC", "true") == "true":
@@ -117,7 +117,7 @@ async def test_async_unary_stream_reader(async_echo):
117117
content = 'The hail in Wales falls mainly on the snails.'
118118
call = await async_echo.expand({
119119
'content': content,
120-
}, metadata=metadata)
120+
}, metadata=_METADATA)
121121

122122
# Consume the response and ensure it matches what we expect.
123123
# with pytest.raises(exceptions.NotFound) as exc:
@@ -127,14 +127,14 @@ async def test_async_unary_stream_reader(async_echo):
127127
assert ground_truth == 'snails.'
128128

129129
trailing_metadata = await call.trailing_metadata()
130-
assert metadata[0] in trailing_metadata.items()
130+
assert _METADATA[0] in trailing_metadata.items()
131131

132132
@pytest.mark.asyncio
133133
async def test_async_unary_stream_async_generator(async_echo):
134134
content = 'The hail in Wales falls mainly on the snails.'
135135
call = await async_echo.expand({
136136
'content': content,
137-
}, metadata=metadata)
137+
}, metadata=_METADATA)
138138

139139
# Consume the response and ensure it matches what we expect.
140140
# with pytest.raises(exceptions.NotFound) as exc:
@@ -145,7 +145,7 @@ async def test_async_unary_stream_async_generator(async_echo):
145145
assert ground_truth == 'snails.'
146146

147147
trailing_metadata = await call.trailing_metadata()
148-
assert metadata[0] in trailing_metadata.items()
148+
assert _METADATA[0] in trailing_metadata.items()
149149

150150
@pytest.mark.asyncio
151151
async def test_async_stream_unary_iterable(async_echo):
@@ -187,7 +187,7 @@ async def test_async_stream_unary_passing_dict(async_echo):
187187

188188
@pytest.mark.asyncio
189189
async def test_async_stream_stream_reader_writier(async_echo):
190-
call = await async_echo.chat(metadata=metadata)
190+
call = await async_echo.chat(metadata=_METADATA)
191191
await call.write(showcase.EchoRequest(content="hello"))
192192
await call.write(showcase.EchoRequest(content="world!"))
193193
await call.done_writing()
@@ -199,7 +199,7 @@ async def test_async_stream_stream_reader_writier(async_echo):
199199
assert contents == ['hello', 'world!']
200200

201201
trailing_metadata = await call.trailing_metadata()
202-
assert metadata[0] in trailing_metadata.items()
202+
assert _METADATA[0] in trailing_metadata.items()
203203

204204
@pytest.mark.asyncio
205205
async def test_async_stream_stream_async_generator(async_echo):
@@ -208,7 +208,7 @@ async def async_generator():
208208
yield showcase.EchoRequest(content="hello")
209209
yield showcase.EchoRequest(content="world!")
210210

211-
call = await async_echo.chat(async_generator(), metadata=metadata)
211+
call = await async_echo.chat(async_generator(), metadata=_METADATA)
212212

213213
contents = []
214214
async for response in call:
@@ -221,12 +221,12 @@ async def async_generator():
221221
@pytest.mark.asyncio
222222
async def test_async_stream_stream_passing_dict(async_echo):
223223
requests = [{'content': 'hello'}, {'content': 'world!'}]
224-
call = await async_echo.chat(iter(requests), metadata=metadata)
224+
call = await async_echo.chat(iter(requests), metadata=_METADATA)
225225

226226
contents = []
227227
async for response in call:
228228
contents.append(response.content)
229229
assert contents == ['hello', 'world!']
230230

231231
trailing_metadata = await call.trailing_metadata()
232-
assert metadata[0] in trailing_metadata.items()
232+
assert _METADATA[0] in trailing_metadata.items()

0 commit comments

Comments
 (0)