Skip to content

Commit b84fecb

Browse files
Update tests/client/transports/test_grpc_client.py
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
1 parent 4768db1 commit b84fecb

1 file changed

Lines changed: 9 additions & 6 deletions

File tree

tests/client/transports/test_grpc_client.py

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -544,17 +544,20 @@ def test_get_grpc_metadata(
544544
assert metadata == expected_metadata
545545

546546

547+
@pytest.mark.parametrize(
548+
'test_extensions',
549+
[
550+
(['ext1']), # Test with explicit extensions
551+
(None), # Test with transport's default extensions
552+
],
553+
)
547554
def test_get_grpc_metadata_uses_lowercase_header_key(
548555
grpc_transport: GrpcTransport,
556+
test_extensions: list[str] | None,
549557
) -> None:
550558
"""Test gRPC metadata header key is always lowercase."""
551559
# Regression: gRPC rejects non-lowercase metadata keys
552-
metadata = grpc_transport._get_grpc_metadata(['ext1'])
553-
assert metadata is not None
554-
key, _ = metadata[0]
555-
assert key == key.lower()
556-
557-
metadata = grpc_transport._get_grpc_metadata()
560+
metadata = grpc_transport._get_grpc_metadata(test_extensions)
558561
assert metadata is not None
559562
key, _ = metadata[0]
560563
assert key == key.lower()

0 commit comments

Comments
 (0)