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 3c812e6 Issue#8592 (#8602)
3c812e6 is described below
commit 3c812e60eb949fcf6ccb15ea6c112a622fc232bb
Author: huanghao495430759 <[email protected]>
AuthorDate: Tue Dec 15 15:42:50 2020 +0800
Issue#8592 (#8602)
Fixes #8592.
Add test case for FrontDatabaseProtocolTypeFactory
---
.../FrontDatabaseProtocolTypeFactoryTest.java | 104 +++++++++++++++++++++
1 file changed, 104 insertions(+)
diff --git
a/shardingsphere-proxy/shardingsphere-proxy-frontend/shardingsphere-proxy-frontend-core/src/test/java/org/apache/shardingsphere/proxy/frontend/protocol/FrontDatabaseProtocolTypeFactoryTest.java
b/shardingsphere-proxy/shardingsphere-proxy-frontend/shardingsphere-proxy-frontend-core/src/test/java/org/apache/shardingsphere/proxy/frontend/protocol/FrontDatabaseProtocolTypeFactoryTest.java
new file mode 100644
index 0000000..a2f1b7d
--- /dev/null
+++
b/shardingsphere-proxy/shardingsphere-proxy-frontend/shardingsphere-proxy-frontend-core/src/test/java/org/apache/shardingsphere/proxy/frontend/protocol/FrontDatabaseProtocolTypeFactoryTest.java
@@ -0,0 +1,104 @@
+/*
+ * 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.proxy.frontend.protocol;
+
+import lombok.SneakyThrows;
+import org.apache.shardingsphere.infra.auth.Authentication;
+import
org.apache.shardingsphere.infra.config.exception.ShardingSphereConfigurationException;
+import
org.apache.shardingsphere.infra.config.properties.ConfigurationProperties;
+import
org.apache.shardingsphere.infra.config.properties.ConfigurationPropertyKey;
+import
org.apache.shardingsphere.infra.context.metadata.impl.StandardMetaDataContexts;
+import org.apache.shardingsphere.infra.database.type.DatabaseType;
+import org.apache.shardingsphere.infra.database.type.dialect.MySQLDatabaseType;
+import org.apache.shardingsphere.infra.executor.kernel.ExecutorEngine;
+import org.apache.shardingsphere.infra.metadata.ShardingSphereMetaData;
+import
org.apache.shardingsphere.infra.metadata.resource.ShardingSphereResource;
+import org.apache.shardingsphere.proxy.backend.context.ProxyContext;
+import org.junit.Test;
+
+import java.lang.reflect.Field;
+import java.util.Collections;
+import java.util.Map;
+import java.util.Properties;
+
+import static org.hamcrest.CoreMatchers.instanceOf;
+import static org.hamcrest.CoreMatchers.is;
+import static org.junit.Assert.assertThat;
+import static org.junit.Assert.assertTrue;
+import static org.mockito.Mockito.RETURNS_DEEP_STUBS;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
+
+public final class FrontDatabaseProtocolTypeFactoryTest {
+
+ @Test(expected = ShardingSphereConfigurationException.class)
+ public void
assertGetDatabaseTypeFromMetaDataContextsThrowShardingSphereConfigurationException()
{
+ StandardMetaDataContexts standardMetaDataContexts = new
StandardMetaDataContexts(Collections.emptyMap(), mock(ExecutorEngine.class),
mock(Authentication.class),
+ new ConfigurationProperties(new Properties()));
+ setMetaDataContexts(standardMetaDataContexts);
+ assertTrue(standardMetaDataContexts.getMetaDataMap().isEmpty());
+ FrontDatabaseProtocolTypeFactory.getDatabaseType();
+ }
+
+ @Test
+ public void
assertGetDatabaseTypeInstanceOfMySQLDatabaseTypeFromMetaDataContextsSchemaName()
{
+ StandardMetaDataContexts standardMetaDataContexts = new
StandardMetaDataContexts(getMetaDataMap(), mock(ExecutorEngine.class),
mock(Authentication.class),
+ new ConfigurationProperties(new Properties()));
+ setMetaDataContexts(standardMetaDataContexts);
+ assertTrue(!standardMetaDataContexts.getMetaDataMap().isEmpty());
+ String configuredDatabaseType =
standardMetaDataContexts.getProps().getValue(ConfigurationPropertyKey.PROXY_FRONTEND_DATABASE_PROTOCOL_TYPE);
+ assertTrue(configuredDatabaseType.isEmpty());
+
assertTrue(standardMetaDataContexts.getAllSchemaNames().contains("mysql"));
+ DatabaseType databaseType =
FrontDatabaseProtocolTypeFactory.getDatabaseType();
+ assertThat(databaseType, instanceOf(DatabaseType.class));
+ assertThat(databaseType.getName(), is("MySQL"));
+ }
+
+ @Test
+ public void
assertGetDatabaseTypeOfPostgreSQLDatabaseTypeFromMetaDataContextsProps() {
+ Properties properties = new Properties();
+
properties.setProperty(ConfigurationPropertyKey.PROXY_FRONTEND_DATABASE_PROTOCOL_TYPE.getKey(),
"PostgreSQL");
+ StandardMetaDataContexts standardMetaDataContexts = new
StandardMetaDataContexts(getMetaDataMap(), mock(ExecutorEngine.class),
mock(Authentication.class),
+ new ConfigurationProperties(properties));
+ setMetaDataContexts(standardMetaDataContexts);
+ assertTrue(!standardMetaDataContexts.getMetaDataMap().isEmpty());
+ String configuredDatabaseType =
standardMetaDataContexts.getProps().getValue(ConfigurationPropertyKey.PROXY_FRONTEND_DATABASE_PROTOCOL_TYPE);
+ assertThat(configuredDatabaseType, is("PostgreSQL"));
+
assertTrue(standardMetaDataContexts.getAllSchemaNames().contains("mysql"));
+ DatabaseType databaseType =
FrontDatabaseProtocolTypeFactory.getDatabaseType();
+ assertThat(databaseType, instanceOf(DatabaseType.class));
+ assertThat(databaseType.getName(), is("PostgreSQL"));
+
assertThat(standardMetaDataContexts.getMetaData("mysql").getResource().getDatabaseType(),
instanceOf(MySQLDatabaseType.class));
+ }
+
+ @SneakyThrows(ReflectiveOperationException.class)
+ private void setMetaDataContexts(final StandardMetaDataContexts
standardMetaDataContexts) {
+ Field field =
ProxyContext.getInstance().getClass().getDeclaredField("metaDataContexts");
+ field.setAccessible(true);
+ field.set(ProxyContext.getInstance(), standardMetaDataContexts);
+ }
+
+ private Map<String, ShardingSphereMetaData> getMetaDataMap() {
+ ShardingSphereResource shardingSphereResource =
mock(ShardingSphereResource.class);
+ when(shardingSphereResource.getDatabaseType()).thenReturn(new
MySQLDatabaseType());
+ ShardingSphereMetaData shardingSphereMetaData =
mock(ShardingSphereMetaData.class, RETURNS_DEEP_STUBS);
+
when(shardingSphereMetaData.getResource()).thenReturn(shardingSphereResource);
+ Map<String, ShardingSphereMetaData> metaDataMap =
Collections.singletonMap("mysql", shardingSphereMetaData);
+ return metaDataMap;
+ }
+}
