From 14cc0f333f8851935f9b135214ebec5ceb134913 Mon Sep 17 00:00:00 2001 From: Tomo Suzuki Date: Fri, 13 Mar 2026 19:03:47 -0400 Subject: [PATCH] test: StreamWriterTest to allow version in trace ID When the source tree of an integration build has non-SNAPSHOT version, the unit test "testAppendWithMissingValueMap" was failing because the trace ID contained the version of the artifact. b/492517601 --- .../google/cloud/bigquery/storage/v1/StreamWriterTest.java | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/java-bigquerystorage/google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1/StreamWriterTest.java b/java-bigquerystorage/google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1/StreamWriterTest.java index 614ec7533358..70912102ab80 100644 --- a/java-bigquerystorage/google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1/StreamWriterTest.java +++ b/java-bigquerystorage/google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1/StreamWriterTest.java @@ -403,7 +403,10 @@ private void verifyAppendRequests(long appendCount) { // First request received by server should have schema and stream name. assertTrue(serverRequest.getProtoRows().hasWriterSchema()); assertEquals(TEST_STREAM_1, serverRequest.getWriteStream()); - assertEquals("java-streamwriter " + TEST_TRACE_ID, serverRequest.getTraceId()); + // The trace ID would look like "java-streamwriter:3.23.0 DATAFLOW:job_id" if + // StreamWriter.getFullTraceId contains the version (non SNAPSHOT version); + // otherwise it's "java-streamwriter DATAFLOW:job_id". + assertThat(serverRequest.getTraceId()).matches("java-streamwriter(:.+)? " + TEST_TRACE_ID); } else { // Following request should not have schema and stream name. assertFalse(serverRequest.getProtoRows().hasWriterSchema());