This is an automated email from the ASF dual-hosted git repository.
duanzhengqiang 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 40f2377ac28 Add Encrypt test (#19502)
40f2377ac28 is described below
commit 40f2377ac2884c16f5c05998678da16847000b34
Author: skai <[email protected]>
AuthorDate: Wed Jul 27 11:44:41 2022 +0800
Add Encrypt test (#19502)
* Add Encrypt test
* EncryptPredicateRightValueTokenGeneratorTest（#15824）
* EncryptPredicateColumnTokenGeneratorTest（#15823）
* revise
* fix checkstyle
* change from review
* fix bug
* remove blank line
Co-authored-by: skai <[email protected]>
---
...rUseDefaultInsertColumnsTokenGeneratorTest.java | 7 ++-
.../token/generator/EncryptGeneratorBaseTest.java | 43 +++++++++++++--
.../EncryptInsertValuesTokenGeneratorTest.java | 8 +--
.../EncryptPredicateColumnTokenGeneratorTest.java | 52 ++++++++++++++++++
...cryptPredicateRightValueTokenGeneratorTest.java | 62 ++++++++++++++++++++++
.../InsertCipherNameTokenGeneratorTest.java | 3 +-
6 files changed, 161 insertions(+), 14 deletions(-)
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
index be092f88ffc..9e2922f1c87 100644
---
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
@@ -17,7 +17,6 @@
package org.apache.shardingsphere.encrypt.rewrite.token.generator;
-import org.apache.shardingsphere.encrypt.rule.EncryptRule;
import org.junit.Test;
import java.util.Collections;
@@ -31,14 +30,14 @@ public final class
EncryptForUseDefaultInsertColumnsTokenGeneratorTest extends E
@Test
public void assertIsGenerateSQLToken() {
EncryptForUseDefaultInsertColumnsTokenGenerator tokenGenerator = new
EncryptForUseDefaultInsertColumnsTokenGenerator();
- tokenGenerator.setEncryptRule(new
EncryptRule(createEncryptRuleConfiguration()));
+ tokenGenerator.setEncryptRule(createEncryptRule());
assertFalse(tokenGenerator.isGenerateSQLToken(createInsertStatementContext(Collections.emptyList())));
}
@Test
public void assertGenerateSQLTokenFromGenerateNewSQLToken() {
EncryptForUseDefaultInsertColumnsTokenGenerator tokenGenerator = new
EncryptForUseDefaultInsertColumnsTokenGenerator();
- tokenGenerator.setEncryptRule(new
EncryptRule(createEncryptRuleConfiguration()));
+ tokenGenerator.setEncryptRule(createEncryptRule());
tokenGenerator.setPreviousSQLTokens(Collections.emptyList());
assertThat(tokenGenerator.generateSQLToken(createInsertStatementContext(Collections.emptyList())).toString(),
is("(id, name, status, pwd_cipher, pwd_assist, pwd_plain)"));
}
@@ -46,7 +45,7 @@ public final class
EncryptForUseDefaultInsertColumnsTokenGeneratorTest extends E
@Test
public void assertGenerateSQLTokenFromPreviousSQLTokens() {
EncryptForUseDefaultInsertColumnsTokenGenerator tokenGenerator = new
EncryptForUseDefaultInsertColumnsTokenGenerator();
- tokenGenerator.setEncryptRule(new
EncryptRule(createEncryptRuleConfiguration()));
+ tokenGenerator.setEncryptRule(createEncryptRule());
tokenGenerator.setPreviousSQLTokens(getPreviousSQLTokens());
assertThat(tokenGenerator.generateSQLToken(createInsertStatementContext(Collections.emptyList())).toString(),
is("(id, name, status, pwd_cipher, pwd_assist, pwd_plain)"));
}
diff --git
a/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-core/src/test/java/org/apache/shardingsphere/encrypt/rewrite/token/generator/EncryptGeneratorBaseTest.java
b/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-core/src/test/java/org/apache/shardingsphere/encrypt/rewrite/token/generator/EncryptGeneratorBaseTest.java
index c12e34028bd..b2d74dba617 100644
---
a/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-core/src/test/java/org/apache/shardingsphere/encrypt/rewrite/token/generator/EncryptGeneratorBaseTest.java
+++
b/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-core/src/test/java/org/apache/shardingsphere/encrypt/rewrite/token/generator/EncryptGeneratorBaseTest.java
@@ -30,30 +30,40 @@ 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.rewrite.token.pojo.EncryptInsertValuesToken;
+import org.apache.shardingsphere.encrypt.rule.EncryptRule;
import
org.apache.shardingsphere.infra.binder.statement.dml.InsertStatementContext;
+import
org.apache.shardingsphere.infra.binder.statement.dml.UpdateStatementContext;
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.SQLToken;
import
org.apache.shardingsphere.infra.rewrite.sql.token.pojo.generic.InsertValue;
+import
org.apache.shardingsphere.sql.parser.sql.common.segment.dml.assignment.ColumnAssignmentSegment;
import
org.apache.shardingsphere.sql.parser.sql.common.segment.dml.assignment.InsertValuesSegment;
+import
org.apache.shardingsphere.sql.parser.sql.common.segment.dml.assignment.SetAssignmentSegment;
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.BinaryOperationExpression;
import
org.apache.shardingsphere.sql.parser.sql.common.segment.dml.expr.ExpressionSegment;
+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.dml.predicate.WhereSegment;
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.apache.shardingsphere.sql.parser.sql.dialect.statement.mysql.dml.MySQLUpdateStatement;
public abstract class EncryptGeneratorBaseTest {
- protected static EncryptRuleConfiguration createEncryptRuleConfiguration()
{
+ private static final String TABLE_NAME = "t_user";
+
+ protected static EncryptRule createEncryptRule() {
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)),
- Collections.singletonMap("test_encryptor", new
ShardingSphereAlgorithmConfiguration("CORE.QUERY_ASSISTED.FIXTURE", new
Properties())));
+ return new EncryptRule(new
EncryptRuleConfiguration(Collections.singleton(new
EncryptTableRuleConfiguration(TABLE_NAME,
Collections.singletonList(pwdColumnConfig), null)),
+ Collections.singletonMap("test_encryptor", new
ShardingSphereAlgorithmConfiguration("CORE.QUERY_ASSISTED.FIXTURE", new
Properties()))));
}
protected static InsertStatementContext createInsertStatementContext(final
List<Object> parameters) {
@@ -67,7 +77,7 @@ public abstract class EncryptGeneratorBaseTest {
protected static InsertStatement createInsertStatement() {
InsertStatement result = new MySQLInsertStatement();
- result.setTable(new SimpleTableSegment(new TableNameSegment(0, 0, new
IdentifierValue("tbl"))));
+ result.setTable(createTableSegment(TABLE_NAME));
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"))));
@@ -76,6 +86,31 @@ public abstract class EncryptGeneratorBaseTest {
return result;
}
+ protected static UpdateStatementContext createUpdatesStatementContext() {
+ MySQLUpdateStatement mySQLUpdateStatement = new MySQLUpdateStatement();
+ mySQLUpdateStatement.setTableSegment(createTableSegment(TABLE_NAME));
+ mySQLUpdateStatement.setWhere(createWhereSegment());
+ mySQLUpdateStatement.setSetAssignment(createSetAssignmentSegment());
+ return new UpdateStatementContext(mySQLUpdateStatement);
+ }
+
+ private static WhereSegment createWhereSegment() {
+ BinaryOperationExpression nameExpression = new
BinaryOperationExpression(10, 24,
+ new ColumnSegment(10, 13, new IdentifierValue("name")), new
LiteralExpressionSegment(18, 22, "LiLei"), "=", "name = 'LiLei'");
+ BinaryOperationExpression pwdExpression = new
BinaryOperationExpression(30, 44,
+ new ColumnSegment(30, 32, new IdentifierValue("pwd")), new
LiteralExpressionSegment(40, 45, "123456"), "=", "pwd = '123456'");
+ return new WhereSegment(0, 0, new BinaryOperationExpression(0, 0,
nameExpression, pwdExpression, "AND", "name = 'LiLei' AND pwd = '123456'"));
+ }
+
+ private static SetAssignmentSegment createSetAssignmentSegment() {
+ List<ColumnSegment> columnSegment = Collections.singletonList(new
ColumnSegment(0, 0, new IdentifierValue("pwd")));
+ return new SetAssignmentSegment(0, 0, Collections.singletonList(new
ColumnAssignmentSegment(0, 0, columnSegment, new LiteralExpressionSegment(0, 0,
"654321"))));
+ }
+
+ private static SimpleTableSegment createTableSegment(final String
tableName) {
+ return new SimpleTableSegment(new TableNameSegment(0, 0, new
IdentifierValue(tableName)));
+ }
+
private static List<ExpressionSegment> createValueExpressions() {
List<ExpressionSegment> result = new ArrayList<>(4);
result.add(new ParameterMarkerExpressionSegment(0, 0, 1));
diff --git
a/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-core/src/test/java/org/apache/shardingsphere/encrypt/rewrite/token/generator/EncryptInsertValuesTokenGeneratorTest.java
b/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-core/src/test/java/org/apache/shardingsphere/encrypt/rewrite/token/generator/EncryptInsertValuesTokenGeneratorTest.java
index 9819eccaf16..24a3257a2a3 100644
---
a/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-core/src/test/java/org/apache/shardingsphere/encrypt/rewrite/token/generator/EncryptInsertValuesTokenGeneratorTest.java
+++
b/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-core/src/test/java/org/apache/shardingsphere/encrypt/rewrite/token/generator/EncryptInsertValuesTokenGeneratorTest.java
@@ -19,7 +19,6 @@ package
org.apache.shardingsphere.encrypt.rewrite.token.generator;
import java.util.Arrays;
import java.util.Collections;
-import org.apache.shardingsphere.encrypt.rule.EncryptRule;
import org.junit.Test;
import static org.hamcrest.CoreMatchers.is;
@@ -31,14 +30,14 @@ public final class EncryptInsertValuesTokenGeneratorTest
extends EncryptGenerato
@Test
public void assertIsGenerateSQLToken() {
EncryptInsertValuesTokenGenerator encryptInsertValuesTokenGenerator =
new EncryptInsertValuesTokenGenerator();
- encryptInsertValuesTokenGenerator.setEncryptRule(new
EncryptRule(createEncryptRuleConfiguration()));
+ encryptInsertValuesTokenGenerator.setEncryptRule(createEncryptRule());
assertTrue(encryptInsertValuesTokenGenerator.isGenerateSQLToken(createInsertStatementContext(Collections.emptyList())));
}
@Test
public void assertGenerateSQLTokenFromGenerateNewSQLToken() {
EncryptInsertValuesTokenGenerator encryptInsertValuesTokenGenerator =
new EncryptInsertValuesTokenGenerator();
- encryptInsertValuesTokenGenerator.setEncryptRule(new
EncryptRule(createEncryptRuleConfiguration()));
+ encryptInsertValuesTokenGenerator.setEncryptRule(createEncryptRule());
encryptInsertValuesTokenGenerator.setPreviousSQLTokens(Collections.emptyList());
encryptInsertValuesTokenGenerator.setDatabaseName("db_schema");
assertThat(encryptInsertValuesTokenGenerator.generateSQLToken(createInsertStatementContext(Arrays.asList(1,
"Tom", 0, "123456"))).toString(), is("(?, ?, ?, ?, ?, ?)"));
@@ -47,7 +46,8 @@ public final class EncryptInsertValuesTokenGeneratorTest
extends EncryptGenerato
@Test
public void assertGenerateSQLTokenFromPreviousSQLTokens() {
EncryptInsertValuesTokenGenerator encryptInsertValuesTokenGenerator =
new EncryptInsertValuesTokenGenerator();
- encryptInsertValuesTokenGenerator.setEncryptRule(new
EncryptRule(createEncryptRuleConfiguration()));
+ encryptInsertValuesTokenGenerator.setDatabaseName("db-001");
+ encryptInsertValuesTokenGenerator.setEncryptRule(createEncryptRule());
encryptInsertValuesTokenGenerator.setPreviousSQLTokens(getPreviousSQLTokens());
encryptInsertValuesTokenGenerator.setDatabaseName("db_schema");
assertThat(encryptInsertValuesTokenGenerator.generateSQLToken(createInsertStatementContext(Arrays.asList(1,
"Tom", 0, "123456"))).toString(), is("(?, ?, ?, ?, ?, ?)"));
diff --git
a/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-core/src/test/java/org/apache/shardingsphere/encrypt/rewrite/token/generator/EncryptPredicateColumnTokenGeneratorTest.java
b/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-core/src/test/java/org/apache/shardingsphere/encrypt/rewrite/token/generator/EncryptPredicateColumnTokenGeneratorTest.java
new file mode 100644
index 00000000000..6dcd10c011f
--- /dev/null
+++
b/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-core/src/test/java/org/apache/shardingsphere/encrypt/rewrite/token/generator/EncryptPredicateColumnTokenGeneratorTest.java
@@ -0,0 +1,52 @@
+/*
+ * 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 org.apache.shardingsphere.infra.database.DefaultDatabase;
+import
org.apache.shardingsphere.infra.rewrite.sql.token.pojo.generic.SubstitutableColumnNameToken;
+import org.junit.Test;
+
+import java.util.Collection;
+import java.util.Collections;
+
+import static org.hamcrest.CoreMatchers.is;
+import static org.junit.Assert.assertThat;
+import static org.junit.Assert.assertTrue;
+
+public final class EncryptPredicateColumnTokenGeneratorTest extends
EncryptGeneratorBaseTest {
+
+ @Test
+ public void assertIsGenerateSQLToken() {
+ EncryptPredicateColumnTokenGenerator tokenGenerator = new
EncryptPredicateColumnTokenGenerator();
+ tokenGenerator.setDatabaseName(DefaultDatabase.LOGIC_NAME);
+ tokenGenerator.setEncryptRule(createEncryptRule());
+ tokenGenerator.setSchemas(Collections.emptyMap());
+
assertTrue(tokenGenerator.isGenerateSQLToken(createUpdatesStatementContext()));
+ }
+
+ @Test
+ public void assertGenerateSQLTokenFromGenerateNewSQLToken() {
+ EncryptPredicateColumnTokenGenerator tokenGenerator = new
EncryptPredicateColumnTokenGenerator();
+ tokenGenerator.setDatabaseName(DefaultDatabase.LOGIC_NAME);
+ tokenGenerator.setEncryptRule(createEncryptRule());
+ tokenGenerator.setSchemas(Collections.emptyMap());
+ Collection<SubstitutableColumnNameToken> substitutableColumnNameTokens
= tokenGenerator.generateSQLTokens(createUpdatesStatementContext());
+ assertThat(substitutableColumnNameTokens.size(), is(1));
+
assertThat(substitutableColumnNameTokens.iterator().next().toString(null),
is("pwd_plain"));
+ }
+}
diff --git
a/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-core/src/test/java/org/apache/shardingsphere/encrypt/rewrite/token/generator/EncryptPredicateRightValueTokenGeneratorTest.java
b/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-core/src/test/java/org/apache/shardingsphere/encrypt/rewrite/token/generator/EncryptPredicateRightValueTokenGeneratorTest.java
new file mode 100644
index 00000000000..5e1457f6da8
--- /dev/null
+++
b/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-core/src/test/java/org/apache/shardingsphere/encrypt/rewrite/token/generator/EncryptPredicateRightValueTokenGeneratorTest.java
@@ -0,0 +1,62 @@
+/*
+ * 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 org.apache.shardingsphere.encrypt.rewrite.condition.EncryptCondition;
+import
org.apache.shardingsphere.encrypt.rewrite.condition.EncryptConditionEngine;
+import
org.apache.shardingsphere.infra.binder.statement.dml.UpdateStatementContext;
+import org.apache.shardingsphere.infra.database.DefaultDatabase;
+import
org.apache.shardingsphere.infra.metadata.database.schema.decorator.model.ShardingSphereSchema;
+import org.apache.shardingsphere.infra.rewrite.sql.token.pojo.SQLToken;
+import org.junit.Test;
+
+import java.util.Collection;
+import java.util.Collections;
+
+import static org.hamcrest.CoreMatchers.is;
+import static org.junit.Assert.assertThat;
+import static org.junit.Assert.assertTrue;
+import static org.mockito.Mockito.mock;
+
+public final class EncryptPredicateRightValueTokenGeneratorTest extends
EncryptGeneratorBaseTest {
+
+ @Test
+ public void assertIsGenerateSQLToken() {
+ EncryptPredicateRightValueTokenGenerator tokenGenerator = new
EncryptPredicateRightValueTokenGenerator();
+ tokenGenerator.setDatabaseName(DefaultDatabase.LOGIC_NAME);
+ tokenGenerator.setEncryptRule(createEncryptRule());
+
assertTrue(tokenGenerator.isGenerateSQLToken(createUpdatesStatementContext()));
+ }
+
+ @Test
+ public void assertGenerateSQLTokenFromGenerateNewSQLToken() {
+ UpdateStatementContext updatesStatementContext =
createUpdatesStatementContext();
+ EncryptPredicateRightValueTokenGenerator tokenGenerator = new
EncryptPredicateRightValueTokenGenerator();
+ tokenGenerator.setDatabaseName(DefaultDatabase.LOGIC_NAME);
+ tokenGenerator.setEncryptRule(createEncryptRule());
+
tokenGenerator.setEncryptConditions(getEncryptConditions(updatesStatementContext));
+ Collection<SQLToken> sqlTokens =
tokenGenerator.generateSQLTokens(updatesStatementContext);
+ assertThat(sqlTokens.size(), is(1));
+ assertThat(sqlTokens.iterator().next().toString(), is("'123456'"));
+ }
+
+ private Collection<EncryptCondition> getEncryptConditions(final
UpdateStatementContext updatesStatementContext) {
+ return new EncryptConditionEngine(createEncryptRule(),
Collections.singletonMap(DefaultDatabase.LOGIC_NAME,
mock(ShardingSphereSchema.class)))
+
.createEncryptConditions(updatesStatementContext.getWhereSegments(),
updatesStatementContext.getColumnSegments(), updatesStatementContext,
DefaultDatabase.LOGIC_NAME);
+ }
+}
diff --git
a/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-core/src/test/java/org/apache/shardingsphere/encrypt/rewrite/token/generator/InsertCipherNameTokenGeneratorTest.java
b/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-core/src/test/java/org/apache/shardingsphere/encrypt/rewrite/token/generator/InsertCipherNameTokenGeneratorTest.java
index 6c0d505cde6..58b662eab21 100644
---
a/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-core/src/test/java/org/apache/shardingsphere/encrypt/rewrite/token/generator/InsertCipherNameTokenGeneratorTest.java
+++
b/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-core/src/test/java/org/apache/shardingsphere/encrypt/rewrite/token/generator/InsertCipherNameTokenGeneratorTest.java
@@ -17,7 +17,6 @@
package org.apache.shardingsphere.encrypt.rewrite.token.generator;
-import org.apache.shardingsphere.encrypt.rule.EncryptRule;
import
org.apache.shardingsphere.infra.binder.statement.dml.SelectStatementContext;
import org.junit.Before;
import org.junit.Test;
@@ -36,7 +35,7 @@ public final class InsertCipherNameTokenGeneratorTest extends
EncryptGeneratorBa
@Before
public void setup() {
- generator.setEncryptRule(new
EncryptRule(createEncryptRuleConfiguration()));
+ generator.setEncryptRule(createEncryptRule());
}
@Test
