Fix test determinism

Replace `PlanNodeSearcher.findFirst` with `PlanNodeSearcher.findOnlyElement`. Test does not expect multiple matching plan nodes.
trinodb · Jul 18, 2023 · cd60b26 · cd60b26
1 parent 7cd0ef6
commit cd60b26
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/plugin/trino-base-jdbc/src/test/java/io/trino/plugin/jdbc/BaseJdbcConnectorTest.java b/plugin/trino-base-jdbc/src/test/java/io/trino/plugin/jdbc/BaseJdbcConnectorTest.java
@@ -1711,12 +1711,9 @@ public void testWriteTaskParallelismSessionProperty(int parallelism, int numberO
                     "INSERT INTO " + table.getName() + " (a, b) SELECT clerk, orderkey FROM tpch.sf100.orders LIMIT " + numberOfRows,
                     WarningCollector.NOOP,
                     createPlanOptimizersStatsCollector());
-            TableWriterNode.WriterTarget target = searchFrom(plan.getRoot())
+            TableWriterNode.WriterTarget target = ((TableWriterNode) searchFrom(plan.getRoot())
                     .where(node -> node instanceof TableWriterNode)
-                    .findFirst()
-                    .map(TableWriterNode.class::cast)
-                    .map(TableWriterNode::getTarget)
-                    .orElseThrow();
+                    .findOnlyElement()).getTarget();
 
             assertThat(target.getMaxWriterTasks(getQueryRunner().getMetadata(), getSession()))
                     .hasValue(parallelism);