diff --git a/java-bigquery/google-cloud-bigquery/src/test/java/com/google/cloud/bigquery/it/ITBigQueryTest.java b/java-bigquery/google-cloud-bigquery/src/test/java/com/google/cloud/bigquery/it/ITBigQueryTest.java index dd7319af2e4c..0fba4f9c8241 100644 --- a/java-bigquery/google-cloud-bigquery/src/test/java/com/google/cloud/bigquery/it/ITBigQueryTest.java +++ b/java-bigquery/google-cloud-bigquery/src/test/java/com/google/cloud/bigquery/it/ITBigQueryTest.java @@ -2981,6 +2981,8 @@ void testInsertAll() throws IOException { builder2.put("FloatField", 1.2); builder2.put("GeographyField", "POINT(-122.350220 47.649154)"); builder2.put("NumericField", new BigDecimal("123456789.123456789")); + // Set insertId on all rows to enable automatic retries by the client library on transient + // errors. InsertAllRequest request = InsertAllRequest.newBuilder(tableInfo.getTableId()) .addRow(UUID.randomUUID().toString(), builder1.build()) @@ -3040,6 +3042,8 @@ void testInsertAllWithSuffix() throws InterruptedException { builder2.put("FloatField", 1.2); builder2.put("GeographyField", "POINT(-122.350220 47.649154)"); builder2.put("NumericField", new BigDecimal("123456789.123456789")); + // Set insertId on all rows to enable automatic retries by the client library on transient + // errors. InsertAllRequest request = InsertAllRequest.newBuilder(tableInfo.getTableId()) .addRow(UUID.randomUUID().toString(), builder1.build()) @@ -3114,6 +3118,8 @@ void testInsertAllWithErrors() { builder3.put("IntegerArrayField", ImmutableList.of(0, 1)); builder3.put("BooleanField", false); builder3.put("BytesField", BYTES_BASE64); + // Set insertId on all rows to enable automatic retries by the client library on transient + // errors. InsertAllRequest request = InsertAllRequest.newBuilder(tableInfo.getTableId()) .addRow(UUID.randomUUID().toString(), builder1.build())