This is an automated email from the ASF dual-hosted git repository.
duanzhengqiang pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/shardingsphere.git
The following commit(s) were added to refs/heads/master by this push:
new caf3501 Create SingleTableSQLRouterTest.java (#14527)
caf3501 is described below
commit caf350115842f256e27f3c96083b05d2188832f2
Author: ZiSheng Zhou <[email protected]>
AuthorDate: Wed Jan 5 19:36:21 2022 +0800
Create SingleTableSQLRouterTest.java (#14527)
* Create SingleTableSQLRouterTest.java
* Update SingleTableSQLRouterTest.java
* Update SingleTableSQLRouterTest.java
* Update SingleTableSQLRouterTest.java
---
.../route/SingleTableSQLRouterTest.java | 88 ++++++++++++++++++++++
1 file changed, 88 insertions(+)
diff --git
a/shardingsphere-kernel/shardingsphere-single-table/shardingsphere-single-table-core/src/test/java/org/apache/shardingsphere/singletable/route/SingleTableSQLRouterTest.java
b/shardingsphere-kernel/shardingsphere-single-table/shardingsphere-single-table-core/src/test/java/org/apache/shardingsphere/singletable/route/SingleTableSQLRouterTest.java
new file mode 100644
index 0000000..ee63826
--- /dev/null
+++
b/shardingsphere-kernel/shardingsphere-single-table/shardingsphere-single-table-core/src/test/java/org/apache/shardingsphere/singletable/route/SingleTableSQLRouterTest.java
@@ -0,0 +1,88 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.shardingsphere.singletable.route;
+
+import org.apache.shardingsphere.infra.binder.LogicSQL;
+import org.apache.shardingsphere.infra.binder.statement.SQLStatementContext;
+import
org.apache.shardingsphere.infra.binder.statement.ddl.CreateTableStatementContext;
+import
org.apache.shardingsphere.infra.config.properties.ConfigurationProperties;
+import org.apache.shardingsphere.infra.database.type.DatabaseType;
+import org.apache.shardingsphere.infra.datanode.DataNode;
+import org.apache.shardingsphere.infra.metadata.ShardingSphereMetaData;
+import org.apache.shardingsphere.infra.route.context.RouteContext;
+import org.apache.shardingsphere.infra.route.context.RouteMapper;
+import org.apache.shardingsphere.infra.route.context.RouteUnit;
+import
org.apache.shardingsphere.singletable.config.SingleTableRuleConfiguration;
+import org.apache.shardingsphere.singletable.rule.SingleTableRule;
+import
org.apache.shardingsphere.sql.parser.sql.common.segment.generic.table.SimpleTableSegment;
+import
org.apache.shardingsphere.sql.parser.sql.common.segment.generic.table.TableNameSegment;
+import
org.apache.shardingsphere.sql.parser.sql.common.statement.ddl.CreateTableStatement;
+import
org.apache.shardingsphere.sql.parser.sql.common.value.identifier.IdentifierValue;
+import
org.apache.shardingsphere.sql.parser.sql.dialect.statement.mysql.ddl.MySQLCreateTableStatement;
+import org.junit.Test;
+
+import javax.sql.DataSource;
+
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.LinkedList;
+import java.util.List;
+import java.util.Map;
+import java.util.Properties;
+
+import static org.hamcrest.CoreMatchers.is;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertThat;
+import static org.mockito.Mockito.RETURNS_DEEP_STUBS;
+import static org.mockito.Mockito.mock;
+
+public final class SingleTableSQLRouterTest {
+
+ @Test
+ public void assertCreateTableSuccessfully() {
+ SingleTableSQLRouter singleTableSQLRouter = new SingleTableSQLRouter();
+ IdentifierValue identifierValue = new IdentifierValue("t_order");
+ TableNameSegment tableNameSegment = new TableNameSegment(1, 2,
identifierValue);
+ SimpleTableSegment simpleTableSegment = new
SimpleTableSegment(tableNameSegment);
+ CreateTableStatement createTableStatement = new
MySQLCreateTableStatement();
+ createTableStatement.setTable(simpleTableSegment);
+ List<Object> parametersList = new LinkedList<>();
+ SQLStatementContext<CreateTableStatement> sqlStatementContext = new
CreateTableStatementContext(createTableStatement);
+ LogicSQL logicSQL = new LogicSQL(sqlStatementContext, "create table",
parametersList);
+ SingleTableRule singleTableRule = new SingleTableRule(new
SingleTableRuleConfiguration(), mock(DatabaseType.class),
+ createDataSourceMap(), Collections.emptyList(), new
ConfigurationProperties(new Properties()));
+ singleTableRule.getSingleTableDataNodes().put("t_order",
Collections.singletonList(new DataNode("ds_0", "t_order")));
+ RouteContext routeContext =
singleTableSQLRouter.createRouteContext(logicSQL,
mock(ShardingSphereMetaData.class), singleTableRule, new
ConfigurationProperties(new Properties()));
+ List<RouteUnit> routeUnits = new
ArrayList<>(routeContext.getRouteUnits());
+ assertThat(routeContext.getRouteUnits().size(), is(1));
+ assertThat(routeUnits.get(0).getDataSourceMapper().getActualName(),
is("ds_0"));
+ assertThat(routeUnits.get(0).getTableMappers().size(), is(1));
+ RouteMapper tableMapper0 =
routeUnits.get(0).getTableMappers().iterator().next();
+ assertThat(tableMapper0.getActualName(), is("t_order"));
+ assertThat(tableMapper0.getLogicName(), is("t_order"));
+ assertFalse(routeContext.isFederated());
+ }
+
+ private Map<String, DataSource> createDataSourceMap() {
+ Map<String, DataSource> result = new HashMap<>(2, 1);
+ result.put("ds_0", mock(DataSource.class, RETURNS_DEEP_STUBS));
+ result.put("ds_1", mock(DataSource.class, RETURNS_DEEP_STUBS));
+ return result;
+ }
+}
