From 93d3539548b2f335dc63e94a39b7f2c55668def7 Mon Sep 17 00:00:00 2001 From: zhangliang Date: Mon, 7 Oct 2024 19:17:31 +0800 Subject: [PATCH] Fix SingleSQLRouterTest --- .../shardingsphere/single/route/SingleSQLRouterTest.java | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/kernel/single/core/src/test/java/org/apache/shardingsphere/single/route/SingleSQLRouterTest.java b/kernel/single/core/src/test/java/org/apache/shardingsphere/single/route/SingleSQLRouterTest.java index 762c6eae7e152..e21b72da467be 100644 --- a/kernel/single/core/src/test/java/org/apache/shardingsphere/single/route/SingleSQLRouterTest.java +++ b/kernel/single/core/src/test/java/org/apache/shardingsphere/single/route/SingleSQLRouterTest.java @@ -55,8 +55,10 @@ import java.sql.Connection; import java.sql.SQLException; import java.util.ArrayList; +import java.util.Arrays; import java.util.Collections; import java.util.HashMap; +import java.util.HashSet; import java.util.Iterator; import java.util.LinkedList; import java.util.List; @@ -195,9 +197,7 @@ void assertDecorateRouteContextWithReadwriteSplittingDataSource() { ShardingSphereDatabase database = new ShardingSphereDatabase(DefaultDatabase.LOGIC_NAME, mock(DatabaseType.class), mock(ResourceMetaData.class, RETURNS_DEEP_STUBS), new RuleMetaData(Collections.singleton(rule)), Collections.emptyMap()); sqlRouter.decorateRouteContext(routeContext, createQueryContext(), database, rule, new ConfigurationProperties(new Properties())); - Iterator routedDataSourceNames = routeContext.getActualDataSourceNames().iterator(); - assertThat(routedDataSourceNames.next(), is("write_ds")); - assertThat(routedDataSourceNames.next(), is("readwrite_ds")); + assertThat(routeContext.getActualDataSourceNames(), is(new HashSet<>(Arrays.asList("write_ds", "readwrite_ds")))); } @Test