From fa149bc49cd8dc4b57a9ca35798cf135218933d6 Mon Sep 17 00:00:00 2001
From: Slawomir Pajak <slawomir.pajak@starburstdata.com>
Date: Mon, 16 Dec 2024 15:22:15 +0100
Subject: [PATCH] Test concurrent update without partition

---
 .../iceberg/TestIcebergLocalConcurrentWrites.java     | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/plugin/trino-iceberg/src/test/java/io/trino/plugin/iceberg/TestIcebergLocalConcurrentWrites.java b/plugin/trino-iceberg/src/test/java/io/trino/plugin/iceberg/TestIcebergLocalConcurrentWrites.java
index 65253a37868d..903fff33da5e 100644
--- a/plugin/trino-iceberg/src/test/java/io/trino/plugin/iceberg/TestIcebergLocalConcurrentWrites.java
+++ b/plugin/trino-iceberg/src/test/java/io/trino/plugin/iceberg/TestIcebergLocalConcurrentWrites.java
@@ -393,13 +393,22 @@ void testConcurrentNonOverlappingUpdate()
     @Test
     void testConcurrentOverlappingUpdate()
             throws Exception
+    {
+        testConcurrentOverlappingUpdate(false);
+        testConcurrentOverlappingUpdate(true);
+    }
+
+    private void testConcurrentOverlappingUpdate(boolean partitioned)
+            throws Exception
     {
         int threads = 3;
         CyclicBarrier barrier = new CyclicBarrier(threads);
         ExecutorService executor = newFixedThreadPool(threads);
         String tableName = "test_concurrent_overlapping_updates_table_" + randomNameSuffix();
 
-        assertUpdate("CREATE TABLE " + tableName + " (a, part)  WITH (partitioning = ARRAY['part']) AS VALUES (1, 10), (11, 20), (21, 30), (31, 40)", 4);
+        assertUpdate("CREATE TABLE " + tableName + " (a, part) " +
+                (partitioned ? " WITH (partitioning = ARRAY['part'])" : "") +
+                " AS VALUES (1, 10), (11, 20), (21, 30), (31, 40)", 4);
 
         try {
             List<Future<Boolean>> futures = IntStream.range(0, threads)