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 40b24de Add test case for EncryptAlgorithmMetaData (#6871)
40b24de is described below
commit 40b24de0fc40a56abc1766caa1dcd50edeb1be15
Author: Serendipity <[email protected]>
AuthorDate: Sun Aug 16 14:27:32 2020 +0800
Add test case for EncryptAlgorithmMetaData (#6871)
* Add test case for EncryptAlgorithmMetaData
* modify method name
---
.../merge/dql/EncryptAlgorithmMetaDataTest.java | 110 +++++++++++++++++++++
1 file changed, 110 insertions(+)
diff --git
a/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-merge/src/test/java/org/apache/shardingsphere/encrypt/merge/dql/EncryptAlgorithmMetaDataTest.java
b/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-merge/src/test/java/org/apache/shardingsphere/encrypt/merge/dql/EncryptAlgorithmMetaDataTest.java
new file mode 100644
index 0000000..7a8f6da
--- /dev/null
+++
b/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-merge/src/test/java/org/apache/shardingsphere/encrypt/merge/dql/EncryptAlgorithmMetaDataTest.java
@@ -0,0 +1,110 @@
+/*
+ * 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.encrypt.merge.dql;
+
+import java.util.Arrays;
+import java.util.Collections;
+import java.util.List;
+import java.util.Optional;
+import java.util.Properties;
+import org.apache.shardingsphere.encrypt.rule.EncryptRule;
+import org.apache.shardingsphere.encrypt.spi.EncryptAlgorithm;
+import
org.apache.shardingsphere.infra.config.algorithm.ShardingSphereAlgorithmConfiguration;
+import
org.apache.shardingsphere.infra.config.algorithm.ShardingSphereAlgorithmFactory;
+import
org.apache.shardingsphere.sql.parser.binder.metadata.schema.SchemaMetaData;
+import
org.apache.shardingsphere.sql.parser.binder.segment.select.projection.Projection;
+import
org.apache.shardingsphere.sql.parser.binder.segment.select.projection.ProjectionsContext;
+import
org.apache.shardingsphere.sql.parser.binder.segment.select.projection.impl.ColumnProjection;
+import
org.apache.shardingsphere.sql.parser.binder.segment.select.projection.impl.DerivedProjection;
+import org.apache.shardingsphere.sql.parser.binder.segment.table.TablesContext;
+import
org.apache.shardingsphere.sql.parser.binder.statement.dml.SelectStatementContext;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.Mock;
+import org.mockito.junit.MockitoJUnitRunner;
+
+import static org.hamcrest.CoreMatchers.is;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertThat;
+import static org.junit.Assert.assertTrue;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
+
+@RunWith(MockitoJUnitRunner.class)
+public final class EncryptAlgorithmMetaDataTest {
+
+ @Mock
+ private SchemaMetaData schemaMetaData;
+
+ @Mock
+ private EncryptRule encryptRule;
+
+ @Mock
+ private SelectStatementContext selectStatementContext;
+
+ @Mock
+ private TablesContext tablesContext;
+
+ @Mock
+ private ColumnProjection columnProjection;
+
+ @Mock
+ private ProjectionsContext projectionsContext;
+
+ private EncryptAlgorithm encryptAlgorithm;
+
+ @Before
+ public void setUp() {
+
when(selectStatementContext.getProjectionsContext()).thenReturn(projectionsContext);
+ List<Projection> columnProjectionList =
Collections.singletonList(columnProjection);
+
when(projectionsContext.getExpandProjections()).thenReturn(columnProjectionList);
+ when(columnProjection.getName()).thenReturn("id");
+
when(selectStatementContext.getTablesContext()).thenReturn(tablesContext);
+ encryptAlgorithm = ShardingSphereAlgorithmFactory.createAlgorithm(new
ShardingSphereAlgorithmConfiguration("Md5", new Properties()),
EncryptAlgorithm.class);
+ }
+
+ @Test
+ public void assertFindEncryptorByTableNameAndColumnName() {
+ when(tablesContext.findTableName(columnProjection,
schemaMetaData)).thenReturn(Optional.of("t_order"));
+ when(encryptRule.findEncryptor("t_order",
"id")).thenReturn(Optional.of(encryptAlgorithm));
+ EncryptAlgorithmMetaData encryptAlgorithmMetaData = new
EncryptAlgorithmMetaData(schemaMetaData, encryptRule, selectStatementContext);
+ Optional<EncryptAlgorithm> actualEncryptor =
encryptAlgorithmMetaData.findEncryptor(1);
+ assertTrue(actualEncryptor.isPresent());
+ assertThat(actualEncryptor.get().getType(), is("MD5"));
+ }
+
+ @Test
+ public void assertFindEncryptorByColumnName() {
+ when(tablesContext.findTableName(columnProjection,
schemaMetaData)).thenReturn(Optional.empty());
+ when(tablesContext.getTableNames()).thenReturn(Arrays.asList("t_user",
"t_user_item", "t_order_item"));
+ when(encryptRule.findEncryptor("t_order_item",
"id")).thenReturn(Optional.of(encryptAlgorithm));
+ EncryptAlgorithmMetaData encryptAlgorithmMetaData = new
EncryptAlgorithmMetaData(schemaMetaData, encryptRule, selectStatementContext);
+ Optional<EncryptAlgorithm> actualEncryptor =
encryptAlgorithmMetaData.findEncryptor(1);
+ assertTrue(actualEncryptor.isPresent());
+ assertThat(actualEncryptor.get().getType(), is("MD5"));
+ }
+
+ @Test
+ public void assertFindEncryptorWhenColumnProjectionIsNotExist() {
+
when(projectionsContext.getExpandProjections()).thenReturn(Collections.singletonList(mock(DerivedProjection.class)));
+ EncryptAlgorithmMetaData encryptAlgorithmMetaData = new
EncryptAlgorithmMetaData(schemaMetaData, encryptRule, selectStatementContext);
+ Optional<EncryptAlgorithm> actualEncryptor =
encryptAlgorithmMetaData.findEncryptor(1);
+ assertFalse(actualEncryptor.isPresent());
+ }
+}
