Fix flaky test in memory connector caused by scaled writer

unix280 · Nov 26, 2024 · ecc08db · ecc08db
1 parent b7501f9
commit ecc08db
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/presto-memory/src/test/java/com/facebook/presto/plugin/memory/TestMemoryWorkerCrash.java b/presto-memory/src/test/java/com/facebook/presto/plugin/memory/TestMemoryWorkerCrash.java
@@ -17,9 +17,12 @@
 import com.facebook.presto.testing.QueryRunner;
 import com.facebook.presto.tests.AbstractTestQueryFramework;
 import com.facebook.presto.tests.DistributedQueryRunner;
+import com.google.common.collect.ImmutableMap;
 import io.airlift.units.Duration;
 import org.testng.annotations.Test;
 
+import java.util.Map;
+
 import static com.facebook.airlift.testing.Assertions.assertLessThan;
 import static io.airlift.units.Duration.nanosSince;
 import static java.util.concurrent.TimeUnit.MILLISECONDS;
@@ -33,7 +36,8 @@ public class TestMemoryWorkerCrash
     protected QueryRunner createQueryRunner()
             throws Exception
     {
-        return MemoryQueryRunner.createQueryRunner();
+        Map<String, String> coordinatorProperties = ImmutableMap.of("scale-writers", "false", "redistribute-writes", "true");
+        return MemoryQueryRunner.createQueryRunner(coordinatorProperties, ImmutableMap.of());
     }
 
     @Test