Skip to content

Commit dfea053

Browse files
committed
test(java-bigquerystorage): Use regex to match traceID
This should resolve the test failure in the java-bigquerystorage module: https://github.com/googleapis/google-cloud-java/actions/runs/23780714031/job/69292481126?pr=12299 b/492517601
1 parent bba6e32 commit dfea053

File tree

2 files changed

+14
-12
lines changed

2 files changed

+14
-12
lines changed

java-bigquerystorage/google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1/ConnectionWorkerTest.java

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -118,17 +118,17 @@ void testMultiplexedAppendSuccess_NonNullTraceId() throws Exception {
118118
testMultiplexedIngestion(
119119
/* sw1TraceId= */ "header_1:trailer_1",
120120
/* sw2TraceId= */ "header_2:trailer_2",
121-
/* expectedSW1TraceId= */ "java-streamwriter header_1:trailer_1",
122-
/* expectedSW2TraceId= */ "java-streamwriter header_2:trailer_2");
121+
/* expectedSW1TraceId= */ "java-streamwriter(:.+)? header_1:trailer_1",
122+
/* expectedSW2TraceId= */ "java-streamwriter(:.+)? header_2:trailer_2");
123123
}
124124

125125
@Test
126126
void testMultiplexedAppendSuccess_EmptyTraceId() throws Exception {
127127
testMultiplexedIngestion(
128128
/* sw1TraceId= */ "header_1:trailer_1",
129129
/* sw2TraceId= */ "",
130-
/* expectedSW1TraceId= */ "java-streamwriter header_1:trailer_1",
131-
/* expectedSW2TraceId= */ "java-streamwriter");
130+
/* expectedSW1TraceId= */ "java-streamwriter(:.+)? header_1:trailer_1",
131+
/* expectedSW2TraceId= */ "java-streamwriter(:.+)?");
132132
}
133133

134134
private void testMultiplexedIngestion(
@@ -216,7 +216,7 @@ private void testMultiplexedIngestion(
216216
assertThat(
217217
serverRequest.getProtoRows().getWriterSchema().getProtoDescriptor().getName())
218218
.isEqualTo("foo");
219-
assertThat(serverRequest.getTraceId()).isEqualTo(expectedSW1TraceId);
219+
assertThat(serverRequest.getTraceId()).matches(expectedSW1TraceId);
220220
break;
221221
case 1:
222222
// The write stream is empty until we enter multiplexing.
@@ -232,7 +232,7 @@ private void testMultiplexedIngestion(
232232
assertThat(
233233
serverRequest.getProtoRows().getWriterSchema().getProtoDescriptor().getName())
234234
.isEqualTo("complicate");
235-
assertThat(serverRequest.getTraceId()).isEqualTo(expectedSW2TraceId);
235+
assertThat(serverRequest.getTraceId()).matches(expectedSW2TraceId);
236236
break;
237237
case 3:
238238
// Schema is empty if not at the first request after table switch.

java-bigquerystorage/google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1/JsonStreamWriterTest.java

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -248,8 +248,8 @@ void testSingleAppendSimpleJson() throws Exception {
248248
.getRows()
249249
.getSerializedRows(0),
250250
expectedProto.toByteString());
251-
assertEquals(
252-
"java-jsonwriter test:empty", testBigQueryWrite.getAppendRequests().get(0).getTraceId());
251+
assertThat(testBigQueryWrite.getAppendRequests().get(0).getTraceId())
252+
.matches("java-jsonwriter(:.+)? test:empty");
253253
}
254254
}
255255

@@ -291,8 +291,8 @@ void testSingleAppendSimpleGson() throws Exception {
291291
.getRows()
292292
.getSerializedRows(0),
293293
expectedProto.toByteString());
294-
assertEquals(
295-
"java-jsonwriter test:empty", testBigQueryWrite.getAppendRequests().get(0).getTraceId());
294+
assertThat(testBigQueryWrite.getAppendRequests().get(0).getTraceId())
295+
.matches("java-jsonwriter(:.+)? test:empty");
296296
}
297297
}
298298

@@ -339,7 +339,8 @@ void testFlexibleColumnAppend() throws Exception {
339339
.getRows()
340340
.getSerializedRows(0),
341341
expectedProto.toByteString());
342-
assertEquals("java-jsonwriter", testBigQueryWrite.getAppendRequests().get(0).getTraceId());
342+
assertThat(testBigQueryWrite.getAppendRequests().get(0).getTraceId())
343+
.matches("java-jsonwriter(:.+)?");
343344
}
344345
}
345346

@@ -561,7 +562,8 @@ void testSingleAppendMultipleSimpleJson() throws Exception {
561562
.getProtoRows()
562563
.getRows()
563564
.getSerializedRowsCount());
564-
assertEquals("java-jsonwriter", testBigQueryWrite.getAppendRequests().get(0).getTraceId());
565+
assertThat(testBigQueryWrite.getAppendRequests().get(0).getTraceId())
566+
.matches("java-jsonwriter(:.+)?");
565567
for (int i = 0; i < 4; i++) {
566568
assertEquals(
567569
testBigQueryWrite

0 commit comments

Comments
 (0)