.

googleapis · Nov 10, 2023 · 388ee54 · 388ee54
1 parent cebe9b9
commit 388ee54
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/...rc/test/java/com/google/cloud/bigquery/storage/v1/it/ITBigQueryWriteManualClientTest.java b/...rc/test/java/com/google/cloud/bigquery/storage/v1/it/ITBigQueryWriteManualClientTest.java
@@ -1573,17 +1573,18 @@ public void testLargeRequest() throws IOException, InterruptedException, Executi
         StreamWriter.newBuilder(parent.toString() + "/_default")
             .setWriterSchema(CreateProtoSchemaWithColField())
             .build()) {
-      List<Integer> sizeSet = Arrays.asList(15 * 1024 * 1024);
+      List<Integer> sizeSet = Arrays.asList(15 * 1024 * 1024, 1024);
       List<ApiFuture<AppendRowsResponse>> responseList =
           new ArrayList<ApiFuture<AppendRowsResponse>>();
       Random r = new Random();
       for (int i = 0; i < 50; i++) {
+        int size = sizeSet.get(r.nextInt(2));
+        LOG.info("Sending size: " + size);
         responseList.add(
             streamWriter.append(
                 CreateProtoRows(
                     new String[] {
-                      new String(new char[sizeSet.get(0)])
-                          .replace('\u0000', (char) (r.nextInt(26) + 'a'))
+                      new String(new char[size]).replace('\u0000', (char) (r.nextInt(26) + 'a'))
                     })));
       }
       for (int i = 0; i < 50; i++) {