Skip to content

Commit 6eece31

Browse files
committed
add test for bug where requuest body is not passed
1 parent 0068d8d commit 6eece31

1 file changed

Lines changed: 18 additions & 1 deletion

File tree

‎packages/gapic-generator/tests/system/test_resumable_upload_progress.py‎

Lines changed: 18 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
ResumableUploadSession,
2222
UploadProgress,
2323
)
24-
from google.showcase import UploadMediaResponse
24+
from google.showcase import UploadMediaRequest, UploadMediaResponse
2525

2626
from conftest import make_resumable_upload
2727

@@ -254,3 +254,20 @@ def test_standalone_finalize_unseekable_stream(intercepted_resumable_upload_rest
254254
]
255255
assert offsets == [0, 262_144, 524_288, 786_432, 786_432]
256256
assert all(p.total_bytes is None for p in progress_records)
257+
258+
259+
def test_client_upload_media_passes_request_body(intercepted_resumable_upload_rest):
260+
"""End-to-end test invoking client.upload_media(request=...) and verifying request body is sent on start."""
261+
client, _ = intercepted_resumable_upload_rest
262+
payload = b"0123456789" * 100
263+
stream = io.BytesIO(payload)
264+
265+
session = client.upload_media(
266+
request=UploadMediaRequest(name="client_upload_media.txt"),
267+
)
268+
response = session.upload(stream=stream)
269+
270+
assert isinstance(response, UploadMediaResponse)
271+
assert response.name == "client_upload_media.txt"
272+
assert response.size == len(payload)
273+

0 commit comments

Comments
 (0)