Simplify assertion in testJoinPushdownDisabled

trinodb · Nov 7, 2022 · ed8bc5b · ed8bc5b
1 parent 2e702c0
commit ed8bc5b
Showing 1 changed file with 3 additions and 8 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
@@ -1125,15 +1125,10 @@ public void testJoinPushdownDisabled()
                 .setSystemProperty("optimize_hash_generation", "false")
                 .build();
 
-        PlanMatchPattern partitionedJoinOverTableScans = node(JoinNode.class,
-                exchange(ExchangeNode.Scope.REMOTE, ExchangeNode.Type.REPARTITION,
-                        node(TableScanNode.class)),
-                exchange(ExchangeNode.Scope.LOCAL,
-                        exchange(ExchangeNode.Scope.REMOTE, ExchangeNode.Type.REPARTITION,
-                                node(TableScanNode.class))));
-
         assertThat(query(noJoinPushdown, "SELECT r.name, n.name FROM nation n JOIN region r ON n.regionkey = r.regionkey"))
-                .isNotFullyPushedDown(partitionedJoinOverTableScans);
+                .isNotFullyPushedDown(node(JoinNode.class,
+                        anyTree(node(TableScanNode.class)),
+                        anyTree(node(TableScanNode.class))));
     }
 
     /**