[shardingsphere] branch master updated: test: add test for KernelProcessor class. (#9664)

[zh...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

zhangliang 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 b29e6ab  test: add test for KernelProcessor class. (#9664)
b29e6ab is described below

commit b29e6abe6cf9708a93a0d87195a4f403b5321606
Author: Asxing Yang（阿行） <ho...@163.com>
AuthorDate: Mon Mar 15 12:56:02 2021 +0800

    test: add test for KernelProcessor class. (#9664)
---
 .../infra/context/kernel/KernelProcessorTest.java  | 71 ++++++++++++++++++++++
 1 file changed, 71 insertions(+)

diff --git a/shardingsphere-infra/shardingsphere-infra-context/src/test/java/org/apache/shardingsphere/infra/context/kernel/KernelProcessorTest.java b/shardingsphere-infra/shardingsphere-infra-context/src/test/java/org/apache/shardingsphere/infra/context/kernel/KernelProcessorTest.java
new file mode 100644
index 0000000..6b0d4e7
--- /dev/null
+++ b/shardingsphere-infra/shardingsphere-infra-context/src/test/java/org/apache/shardingsphere/infra/context/kernel/KernelProcessorTest.java
@@ -0,0 +1,71 @@
+/*
+ * 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.infra.context.kernel;
+
+import org.apache.shardingsphere.infra.binder.LogicSQL;
+import org.apache.shardingsphere.infra.binder.statement.SQLStatementContext;
+import org.apache.shardingsphere.infra.config.properties.ConfigurationProperties;
+import org.apache.shardingsphere.infra.config.properties.ConfigurationPropertyKey;
+import org.apache.shardingsphere.infra.executor.sql.context.ExecutionContext;
+import org.apache.shardingsphere.infra.metadata.ShardingSphereMetaData;
+import org.apache.shardingsphere.infra.metadata.resource.ShardingSphereResource;
+import org.apache.shardingsphere.infra.metadata.rule.ShardingSphereRuleMetaData;
+import org.apache.shardingsphere.infra.metadata.schema.ShardingSphereSchema;
+import org.apache.shardingsphere.sql.parser.sql.common.statement.SQLStatement;
+import org.apache.shardingsphere.sql.parser.sql.dialect.statement.mysql.dml.MySQLSelectStatement;
+import org.hamcrest.Matchers;
+import org.junit.Before;
+import org.junit.Test;
+
+import java.util.Collections;
+import java.util.Optional;
+import java.util.Properties;
+
+import static org.junit.Assert.assertThat;
+import static org.mockito.Mockito.RETURNS_DEEP_STUBS;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
+
+public class KernelProcessorTest {
+
+    private KernelProcessor kernelProcessor;
+
+    @Before
+    public void setUp() throws Exception {
+        kernelProcessor = new KernelProcessor();
+    }
+
+    @Test
+    public void testGenerateExecutionContext() {
+        SQLStatementContext<SQLStatement> sqlStatementContext = mock(SQLStatementContext.class);
+        MySQLSelectStatement selectStatement = mock(MySQLSelectStatement.class);
+        when(selectStatement.getLock()).thenReturn(Optional.empty());
+        when(sqlStatementContext.getSqlStatement()).thenReturn(selectStatement);
+        final LogicSQL logicSQL = new LogicSQL(sqlStatementContext, "select * from t_order", Collections.emptyList());
+        final ShardingSphereMetaData metaData = new ShardingSphereMetaData("logic_schema",
+                mock(ShardingSphereResource.class, RETURNS_DEEP_STUBS),
+                new ShardingSphereRuleMetaData(Collections.emptyList(),
+                        Collections.emptyList()),
+                mock(ShardingSphereSchema.class));
+        Properties properties = new Properties();
+        properties.setProperty(ConfigurationPropertyKey.SQL_SHOW.getKey(), Boolean.TRUE.toString());
+        final ConfigurationProperties props = new ConfigurationProperties(properties);
+        final ExecutionContext result = kernelProcessor.generateExecutionContext(logicSQL, metaData, props);
+        assertThat(1, Matchers.is(result.getExecutionUnits().size()));
+    }
+}