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 a856e11042c Add unit test for
EncryptForUseDefaultInsertColumnsTokenGenerator (#19368)
a856e11042c is described below
commit a856e11042c9c4ba58eb8c29c16daa4f8b690ea2
Author: skai <[email protected]>
AuthorDate: Wed Jul 20 10:30:23 2022 +0800
Add unit test for EncryptForUseDefaultInsertColumnsTokenGenerator (#19368)
* Add unit test for EncryptForUseDefaultInsertColumnsTokenGenerator
* add License
---
...ptForUseDefaultInsertColumnsTokenGenerator.java | 3 +
.../EncryptAssignmentTokenGeneratorTest.java | 3 +-
...rUseDefaultInsertColumnsTokenGeneratorTest.java | 102 +++++++++++++++++++++
3 files changed, 106 insertions(+), 2 deletions(-)
diff --git
a/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-core/src/main/java/org/apache/shardingsphere/encrypt/rewrite/token/generator/EncryptForUseDefaultInsertColumnsTokenGenerator.java
b/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-core/src/main/java/org/apache/shardingsphere/encrypt/rewrite/token/generator/EncryptForUseDefaultInsertColumnsTokenGenerator.java
index 4c6c9db880c..8f013f1b54f 100644
---
a/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-core/src/main/java/org/apache/shardingsphere/encrypt/rewrite/token/generator/EncryptForUseDefaultInsertColumnsTokenGenerator.java
+++
b/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-core/src/main/java/org/apache/shardingsphere/encrypt/rewrite/token/generator/EncryptForUseDefaultInsertColumnsTokenGenerator.java
@@ -62,6 +62,9 @@ public final class
EncryptForUseDefaultInsertColumnsTokenGenerator implements Op
}
private Optional<UseDefaultInsertColumnsToken> findInsertColumnsToken() {
+ if (null == previousSQLTokens) {
+ return Optional.empty();
+ }
for (SQLToken each : previousSQLTokens) {
if (each instanceof UseDefaultInsertColumnsToken) {
return Optional.of((UseDefaultInsertColumnsToken) each);
diff --git
a/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-core/src/test/java/org/apache/shardingsphere/encrypt/rewrite/token/EncryptAssignmentTokenGeneratorTest.java
b/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-core/src/test/java/org/apache/shardingsphere/encrypt/rewrite/token/generator/EncryptAssignmentTokenGeneratorTest.java
similarity index 97%
rename from
shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-core/src/test/java/org/apache/shardingsphere/encrypt/rewrite/token/EncryptAssignmentTokenGeneratorTest.java
rename to
shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-core/src/test/java/org/apache/shardingsphere/encrypt/rewrite/token/generator/EncryptAssignmentTokenGeneratorTest.java
index b330bc27bbc..f299ac856bb 100644
---
a/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-core/src/test/java/org/apache/shardingsphere/encrypt/rewrite/token/EncryptAssignmentTokenGeneratorTest.java
+++
b/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-core/src/test/java/org/apache/shardingsphere/encrypt/rewrite/token/generator/EncryptAssignmentTokenGeneratorTest.java
@@ -15,9 +15,8 @@
* limitations under the License.
*/
-package org.apache.shardingsphere.encrypt.rewrite.token;
+package org.apache.shardingsphere.encrypt.rewrite.token.generator;
-import
org.apache.shardingsphere.encrypt.rewrite.token.generator.EncryptAssignmentTokenGenerator;
import org.apache.shardingsphere.encrypt.rule.EncryptRule;
import org.apache.shardingsphere.encrypt.spi.EncryptAlgorithm;
import
org.apache.shardingsphere.infra.binder.statement.dml.InsertStatementContext;
diff --git
a/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-core/src/test/java/org/apache/shardingsphere/encrypt/rewrite/token/generator/EncryptForUseDefaultInsertColumnsTokenGeneratorTest.java
b/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-core/src/test/java/org/apache/shardingsphere/encrypt/rewrite/token/generator/EncryptForUseDefaultInsertColumnsTokenGeneratorTest.java
new file mode 100644
index 00000000000..7c767d5580b
--- /dev/null
+++
b/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-core/src/test/java/org/apache/shardingsphere/encrypt/rewrite/token/generator/EncryptForUseDefaultInsertColumnsTokenGeneratorTest.java
@@ -0,0 +1,102 @@
+/*
+ * 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.rewrite.token.generator;
+
+import com.google.common.collect.ImmutableMap;
+import org.apache.shardingsphere.encrypt.api.config.EncryptRuleConfiguration;
+import
org.apache.shardingsphere.encrypt.api.config.rule.EncryptColumnRuleConfiguration;
+import
org.apache.shardingsphere.encrypt.api.config.rule.EncryptTableRuleConfiguration;
+import org.apache.shardingsphere.encrypt.rule.EncryptRule;
+import
org.apache.shardingsphere.infra.binder.statement.dml.InsertStatementContext;
+import
org.apache.shardingsphere.infra.config.algorithm.ShardingSphereAlgorithmConfiguration;
+import org.apache.shardingsphere.infra.database.DefaultDatabase;
+import
org.apache.shardingsphere.infra.metadata.database.ShardingSphereDatabase;
+import
org.apache.shardingsphere.infra.metadata.database.schema.decorator.model.ShardingSphereSchema;
+import
org.apache.shardingsphere.infra.rewrite.sql.token.pojo.generic.UseDefaultInsertColumnsToken;
+import
org.apache.shardingsphere.sql.parser.sql.common.segment.dml.assignment.InsertValuesSegment;
+import
org.apache.shardingsphere.sql.parser.sql.common.segment.dml.column.ColumnSegment;
+import
org.apache.shardingsphere.sql.parser.sql.common.segment.dml.column.InsertColumnsSegment;
+import
org.apache.shardingsphere.sql.parser.sql.common.segment.dml.expr.simple.LiteralExpressionSegment;
+import
org.apache.shardingsphere.sql.parser.sql.common.segment.dml.expr.simple.ParameterMarkerExpressionSegment;
+import
org.apache.shardingsphere.sql.parser.sql.common.segment.generic.table.SimpleTableSegment;
+import
org.apache.shardingsphere.sql.parser.sql.common.segment.generic.table.TableNameSegment;
+import
org.apache.shardingsphere.sql.parser.sql.common.statement.dml.InsertStatement;
+import
org.apache.shardingsphere.sql.parser.sql.common.value.identifier.IdentifierValue;
+import
org.apache.shardingsphere.sql.parser.sql.dialect.statement.mysql.dml.MySQLInsertStatement;
+import org.junit.Test;
+import java.util.Arrays;
+import java.util.Collections;
+import java.util.List;
+import java.util.Properties;
+
+import static org.hamcrest.CoreMatchers.is;
+import static org.junit.Assert.assertThat;
+import static org.junit.Assert.assertFalse;
+import static org.mockito.Mockito.RETURNS_DEEP_STUBS;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
+
+public final class EncryptForUseDefaultInsertColumnsTokenGeneratorTest {
+
+ @Test
+ public void assertIsGenerateSQLToken() {
+ EncryptForUseDefaultInsertColumnsTokenGenerator
encryptForUseDefaultInsertColumnsTokenGenerator = new
EncryptForUseDefaultInsertColumnsTokenGenerator();
+ encryptForUseDefaultInsertColumnsTokenGenerator.setEncryptRule(new
EncryptRule(createEncryptRuleConfiguration()));
+ InsertStatementContext insertStatementContext =
createInsertStatementContext(Collections.emptyList());
+
assertFalse(encryptForUseDefaultInsertColumnsTokenGenerator.isGenerateSQLToken(insertStatementContext));
+ }
+
+ @Test
+ public void assertGenerateSQLToken() {
+ EncryptForUseDefaultInsertColumnsTokenGenerator
encryptForUseDefaultInsertColumnsTokenGenerator = new
EncryptForUseDefaultInsertColumnsTokenGenerator();
+ encryptForUseDefaultInsertColumnsTokenGenerator.setEncryptRule(new
EncryptRule(createEncryptRuleConfiguration()));
+ InsertStatementContext insertStatementContext =
createInsertStatementContext(Collections.emptyList());
+ UseDefaultInsertColumnsToken useDefaultInsertColumnsToken =
encryptForUseDefaultInsertColumnsTokenGenerator.generateSQLToken(insertStatementContext);
+ assertThat(useDefaultInsertColumnsToken.toString(), is("(id, name,
status, pwd_cipher, pwd_assist, pwd_plain)"));
+ }
+
+ private InsertStatementContext createInsertStatementContext(final
List<Object> parameters) {
+ InsertStatement insertStatement = createInsertStatement();
+ ShardingSphereDatabase database = mock(ShardingSphereDatabase.class,
RETURNS_DEEP_STUBS);
+ ShardingSphereSchema schema = mock(ShardingSphereSchema.class);
+ String defaultSchemaName = DefaultDatabase.LOGIC_NAME;
+ when(database.getSchemas().get(defaultSchemaName)).thenReturn(schema);
+ when(schema.getAllColumnNames("tbl")).thenReturn(Arrays.asList("id",
"name", "status", "pwd"));
+ return new
InsertStatementContext(Collections.singletonMap(DefaultDatabase.LOGIC_NAME,
database), parameters, insertStatement, DefaultDatabase.LOGIC_NAME);
+ }
+
+ private InsertStatement createInsertStatement() {
+ InsertStatement result = new MySQLInsertStatement();
+ result.setTable(new SimpleTableSegment(new TableNameSegment(0, 0, new
IdentifierValue("tbl"))));
+ InsertColumnsSegment insertColumnsSegment = new
InsertColumnsSegment(0, 0, Arrays.asList(
+ new ColumnSegment(0, 0, new IdentifierValue("id")), new
ColumnSegment(0, 0, new IdentifierValue("name")),
+ new ColumnSegment(0, 0, new IdentifierValue("status")), new
ColumnSegment(0, 0, new IdentifierValue("pwd"))));
+ result.setInsertColumns(insertColumnsSegment);
+ result.getValues().add(new InsertValuesSegment(0, 0, Arrays.asList(
+ new ParameterMarkerExpressionSegment(0, 0, 1), new
ParameterMarkerExpressionSegment(0, 0, 2), new LiteralExpressionSegment(0, 0,
"init"))));
+ result.getValues().add(new InsertValuesSegment(0, 0, Arrays.asList(
+ new ParameterMarkerExpressionSegment(0, 0, 3), new
ParameterMarkerExpressionSegment(0, 0, 4), new LiteralExpressionSegment(0, 0,
"init"))));
+ return result;
+ }
+
+ private EncryptRuleConfiguration createEncryptRuleConfiguration() {
+ EncryptColumnRuleConfiguration pwdColumnConfig = new
EncryptColumnRuleConfiguration("pwd", "pwd_cipher", "pwd_assist", "pwd_plain",
"test_encryptor", "test_encryptor", false);
+ return new EncryptRuleConfiguration(Collections.singleton(new
EncryptTableRuleConfiguration("tbl",
Collections.singletonList(pwdColumnConfig), null)),
+ ImmutableMap.of("test_encryptor", new
ShardingSphereAlgorithmConfiguration("CORE.QUERY_ASSISTED.FIXTURE", new
Properties())));
+ }
+}
