Merge 280ea92 into e83962e

shardingsphere-proxy/shardingsphere-proxy-backend/src/test/java/org/apache/shardingsphere/proxy/backend/text/admin/DALBackendHandlerFactoryTest.java

@@ -0,0 +1,103 @@
+/*
+ * 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.backend.text.admin;
+
+import lombok.SneakyThrows;
+import org.apache.shardingsphere.proxy.backend.communication.jdbc.connection.BackendConnection;
+import org.apache.shardingsphere.proxy.backend.text.TextProtocolBackendHandler;
+import org.apache.shardingsphere.sql.parser.sql.common.statement.dal.DALStatement;
+import org.apache.shardingsphere.sql.parser.sql.common.statement.dal.SetStatement;
+import org.apache.shardingsphere.sql.parser.sql.dialect.statement.mysql.dal.MySQLShowDatabasesStatement;
+import org.apache.shardingsphere.sql.parser.sql.dialect.statement.mysql.dal.MySQLShowTablesStatement;
+import org.apache.shardingsphere.sql.parser.sql.dialect.statement.mysql.dal.MySQLUseStatement;
+import org.hamcrest.Matcher;
+import org.junit.Test;
+
+import java.lang.reflect.Field;
+
+import static org.hamcrest.CoreMatchers.instanceOf;
+import static org.hamcrest.CoreMatchers.is;
+import static org.junit.Assert.assertThat;
+import static org.mockito.Mockito.mock;
+
+public final class DALBackendHandlerFactoryTest {
+
+    @Test
+    public void assertShowTablesBackendHandlerReturnedWhenMySQLShowTablesStatement() {
+        MySQLShowTablesStatement mySQLShowTablesStatement = mock(MySQLShowTablesStatement.class);
+        BackendConnection backendConnection = mock(BackendConnection.class);
+        TextProtocolBackendHandler textProtocolBackendHandler = DALBackendHandlerFactory.newInstance("", mySQLShowTablesStatement, backendConnection);
+        assertThat(textProtocolBackendHandler, instanceOf(ShowTablesBackendHandler.class));
+        ShowTablesBackendHandler showTablesBackendHandler = (ShowTablesBackendHandler) textProtocolBackendHandler;
+        assertFieldOfInstance(showTablesBackendHandler, "sqlStatement", is(mySQLShowTablesStatement));
+        assertFieldOfInstance(showTablesBackendHandler, "sql", is(""));
+        assertFieldOfInstance(showTablesBackendHandler, "backendConnection", is(backendConnection));
+    }
+
+    @Test
+    public void assertUseDatabaseBackendHandlerReturnedWhenMySQLUseStatement() {
+        MySQLUseStatement mySQLUseStatement = mock(MySQLUseStatement.class);
+        BackendConnection backendConnection = mock(BackendConnection.class);
+        TextProtocolBackendHandler textProtocolBackendHandler = DALBackendHandlerFactory.newInstance("", mySQLUseStatement, backendConnection);
+        assertThat(textProtocolBackendHandler, instanceOf(UseDatabaseBackendHandler.class));
+        UseDatabaseBackendHandler useDatabaseBackendHandler = (UseDatabaseBackendHandler) textProtocolBackendHandler;
+        assertFieldOfInstance(useDatabaseBackendHandler, "useStatement", is(mySQLUseStatement));
+        assertFieldOfInstance(useDatabaseBackendHandler, "backendConnection", is(backendConnection));
+    }
+
+    @Test
+    public void assertShowDatabasesBackendHandlerReturnedWhenMySQLShowDatabasesStatement() {
+        BackendConnection backendConnection = mock(BackendConnection.class);
+        TextProtocolBackendHandler textProtocolBackendHandler = DALBackendHandlerFactory.newInstance("", mock(MySQLShowDatabasesStatement.class), backendConnection);
+        assertThat(textProtocolBackendHandler, instanceOf(ShowDatabasesBackendHandler.class));
+        ShowDatabasesBackendHandler showDatabasesBackendHandler = (ShowDatabasesBackendHandler) textProtocolBackendHandler;
+        assertFieldOfInstance(showDatabasesBackendHandler, "backendConnection", is(backendConnection));
+    }
+
+    @Test
+    public void assertBroadcastBackendHandlerReturnedWhenSetStatement() {
+        SetStatement setStatement = mock(SetStatement.class);
+        BackendConnection backendConnection = mock(BackendConnection.class);
+        TextProtocolBackendHandler textProtocolBackendHandler = DALBackendHandlerFactory.newInstance("", setStatement, backendConnection);
+        assertThat(textProtocolBackendHandler, instanceOf(BroadcastBackendHandler.class));
+        BroadcastBackendHandler broadcastBackendHandler = (BroadcastBackendHandler) textProtocolBackendHandler;
+        assertFieldOfInstance(broadcastBackendHandler, "sqlStatement", is(setStatement));
+        assertFieldOfInstance(broadcastBackendHandler, "sql", is(""));
+        assertFieldOfInstance(broadcastBackendHandler, "backendConnection", is(backendConnection));
+    }
+
+    @Test
+    public void assertUnicastBackendHandlerReturnedWhenOtherDALStatement() {
+        DALStatement dalStatement = mock(DALStatement.class);
+        BackendConnection backendConnection = mock(BackendConnection.class);
+        TextProtocolBackendHandler textProtocolBackendHandler = DALBackendHandlerFactory.newInstance("", dalStatement, backendConnection);
+        assertThat(textProtocolBackendHandler, instanceOf(UnicastBackendHandler.class));
+        UnicastBackendHandler unicastBackendHandler = (UnicastBackendHandler) textProtocolBackendHandler;
+        assertFieldOfInstance(unicastBackendHandler, "sqlStatement", is(dalStatement));
+        assertFieldOfInstance(unicastBackendHandler, "sql", is(""));
+        assertFieldOfInstance(unicastBackendHandler, "backendConnection", is(backendConnection));
+    }
+
+    @SneakyThrows
+    private <S, T> void assertFieldOfInstance(final S classInstance, final String fieldName, final Matcher<T> matcher) {
+        Field field = classInstance.getClass().getDeclaredField(fieldName);
+        field.setAccessible(true);
+        T value = (T) field.get(classInstance);
+        assertThat(value, matcher);
+    }
+}