@@ -0,0 +1,2407 @@
+/*%0A * Licensed to the Apache Software Foundation (ASF) under one or more%0A * contributor license agreements.  See the NOTICE file distributed with%0A * this work for additional information regarding copyright ownership.%0A * The ASF licenses this file to You under the Apache License, Version 2.0%0A * (the %22License%22); you may not use this file except in compliance with%0A * the License.  You may obtain a copy of the License at%0A *%0A *     http://www.apache.org/licenses/LICENSE-2.0%0A *%0A * Unless required by applicable law or agreed to in writing, software%0A * distributed under the License is distributed on an %22AS IS%22 BASIS,%0A * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.%0A * See the License for the specific language governing permissions and%0A * limitations under the License.%0A */%0A%0Apackage org.apache.shardingsphere.encrypt.merge.dql;%0A%0Aimport org.apache.shardingsphere.infra.executor.sql.QueryResult;%0Aimport org.apache.shardingsphere.infra.merge.result.MergedResult;%0Aimport org.apache.shardingsphere.sql.parser.binder.metadata.schema.SchemaMetaData;%0Aimport org.apache.shardingsphere.sql.parser.binder.statement.SQLStatementContext;%0Aimport org.junit.Test;%0A%0Aimport java.sql.SQLException;%0A%0Aimport static org.junit.Assert.assertTrue;%0Aimport static org.mockito.Mockito.mock;%0Aimport static org.mockito.Mockito.when;%0A%0Apublic final class EncryptDQLResultDecoratorTest %7B%0A%0A    private EncryptAlgorithmMetaData metaData;%0A%0A    private boolean queryWithCipherColumn;%0A%0A    @Test%0A    public void assertDecorateQueryResult() throws SQLException %7B%0A        QueryResult queryResult = mock(QueryResult.class);%0A        when(queryResult.next()).thenReturn(true);%0A        EncryptDQLResultDecorator decorator = new EncryptDQLResultDecorator(metaData, queryWithCipherColumn);%0A        MergedResult actual = decorator.decorate(queryResult, mock(SQLStatementContext.class), mock(SchemaMetaData.class));%0A        assertTrue(actual.next());%0A    %7D%0A%0A    @Test%0A    public void assertDecorateMergedResult() throws SQLException %7B%0A        MergedResult mergedResult = mock(MergedResult.class);%0A        when(mergedResult.next()).thenReturn(true);%0A        EncryptDQLResultDecorator decorator = new EncryptDQLResultDecorator(metaData, queryWithCipherColumn);%0A        MergedResult actual = decorator.decorate(mergedResult, mock(SQLStatementContext.class), mock(SchemaMetaData.class));%0A        assertTrue(actual.next());%0A    %7D%0A%7D%0A

Add files via upload (#7058)