NO-SNOW Fix flaky test using multiple threads (snowflakedb#790)

sfc-gh-kgaputis · Sep 12, 2024 · f0100f1 · f0100f1
1 parent c98252e
commit f0100f1
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/test/java/net/snowflake/ingest/streaming/internal/StreamingIngestBigFilesIT.java b/src/test/java/net/snowflake/ingest/streaming/internal/StreamingIngestBigFilesIT.java
@@ -130,15 +130,16 @@ private void ingestRandomRowsToTable(
       boolean isNullable)
       throws ExecutionException, InterruptedException {
 
-    List<Map<String, Object>> rows = new ArrayList<>();
+    final List<Map<String, Object>> rows = Collections.synchronizedList(new ArrayList<>());
     for (int i = 0; i < batchSize; i++) {
       Random r = new Random();
       rows.add(TestUtils.getRandomRow(r, isNullable));
     }
 
     ExecutorService testThreadPool = Executors.newFixedThreadPool(numChannels);
     CompletableFuture[] futures = new CompletableFuture[numChannels];
-    List<SnowflakeStreamingIngestChannel> channelList = new ArrayList<>();
+    List<SnowflakeStreamingIngestChannel> channelList =
+        Collections.synchronizedList(new ArrayList<>());
     for (int i = 0; i < numChannels; i++) {
       final String channelName = "CHANNEL" + i;
       int finalI = i;