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 06834bb add unit test for EncryptAssignmentTokenGenerator (#15701)
(#15917)
06834bb is described below
commit 06834bbe4079a25b256c2467c5d857192162b8b0
Author: Panagiotis Parthenis <[email protected]>
AuthorDate: Thu Mar 10 03:00:33 2022 +0200
add unit test for EncryptAssignmentTokenGenerator (#15701) (#15917)
---
.../token/EncryptAssignmentTokenGeneratorTest.java | 130 +++++++++++++++++++++
1 file changed, 130 insertions(+)
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/EncryptAssignmentTokenGeneratorTest.java
new file mode 100644
index 0000000..c1571c0
--- /dev/null
+++
b/shardingsphere-features/shardingsphere-encrypt/shardingsphere-encrypt-core/src/test/java/org/apache/shardingsphere/encrypt/rewrite/token/EncryptAssignmentTokenGeneratorTest.java
@@ -0,0 +1,130 @@
+/*
+ * 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;
+
+import
org.apache.shardingsphere.encrypt.rewrite.token.generator.EncryptAssignmentTokenGenerator;
+import
org.apache.shardingsphere.encrypt.rewrite.token.pojo.EncryptAssignmentToken;
+import org.apache.shardingsphere.encrypt.rule.EncryptRule;
+import org.apache.shardingsphere.encrypt.spi.EncryptAlgorithm;
+import
org.apache.shardingsphere.infra.binder.statement.dml.InsertStatementContext;
+import
org.apache.shardingsphere.infra.binder.statement.dml.UpdateStatementContext;
+import
org.apache.shardingsphere.sql.parser.sql.common.segment.dml.assignment.AssignmentSegment;
+import
org.apache.shardingsphere.sql.parser.sql.common.segment.dml.assignment.SetAssignmentSegment;
+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.dialect.handler.dml.InsertStatementHandler;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.MockedStatic;
+import org.mockito.junit.MockitoJUnitRunner;
+import java.util.Arrays;
+import java.util.Collection;
+import java.util.Optional;
+
+import static org.junit.Assert.assertThat;
+import static org.junit.Assert.assertTrue;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.ArgumentMatchers.eq;
+import static org.hamcrest.CoreMatchers.is;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.mockStatic;
+import static org.mockito.Mockito.when;
+import static org.mockito.Mockito.RETURNS_DEEP_STUBS;
+
+@RunWith(MockitoJUnitRunner.class)
+public final class EncryptAssignmentTokenGeneratorTest {
+
+ private EncryptAssignmentTokenGenerator tokenGenerator;
+
+ private EncryptRule encryptRule;
+
+ private UpdateStatementContext updateStatement;
+
+ private InsertStatementContext insertStatement;
+
+ private AssignmentSegment assignmentSegment;
+
+ private SetAssignmentSegment setAssignmentSegment;
+
+ private LiteralExpressionSegment literalExpression;
+
+ private ParameterMarkerExpressionSegment parameterMarkerExpression;
+
+ private EncryptAlgorithm encryptAlgorithm;
+
+ @Before
+ public void setup() {
+ encryptAlgorithm = mock(EncryptAlgorithm.class);
+ encryptRule = mock(EncryptRule.class, RETURNS_DEEP_STUBS);
+ updateStatement = mock(UpdateStatementContext.class,
RETURNS_DEEP_STUBS);
+ insertStatement = mock(InsertStatementContext.class,
RETURNS_DEEP_STUBS);
+ assignmentSegment = mock(AssignmentSegment.class, RETURNS_DEEP_STUBS);
+ setAssignmentSegment = mock(SetAssignmentSegment.class,
RETURNS_DEEP_STUBS);
+ literalExpression = mock(LiteralExpressionSegment.class,
RETURNS_DEEP_STUBS);
+ parameterMarkerExpression =
mock(ParameterMarkerExpressionSegment.class, RETURNS_DEEP_STUBS);
+ tokenGenerator = new EncryptAssignmentTokenGenerator();
+ tokenGenerator.setEncryptRule(encryptRule);
+
when(updateStatement.getAllTables().iterator().next().getTableName().getIdentifier().getValue()).thenReturn("table");
+
when(updateStatement.getSqlStatement().getSetAssignment().getAssignments()).thenReturn(Arrays.asList(assignmentSegment));
+
when(assignmentSegment.getColumns().get(0).getIdentifier().getValue()).thenReturn("columns");
+ when(encryptRule.findEncryptor(eq("table"),
eq("columns"))).thenReturn(Optional.of(encryptAlgorithm));
+
when(insertStatement.getAllTables().iterator().next().getTableName().getIdentifier().getValue()).thenReturn("table");
+
when(setAssignmentSegment.getAssignments()).thenReturn(Arrays.asList(assignmentSegment));
+ }
+
+ @Test
+ public void assertIsGenerateSQLTokenUpdateSQLSuccess() {
+ assertTrue(tokenGenerator.isGenerateSQLToken(updateStatement));
+ }
+
+ @Test
+ public void assertIsGenerateSQLTokenUpdateSQLFail() {
+ assertTrue(tokenGenerator.isGenerateSQLToken(insertStatement));
+ }
+
+ @Test
+ public void
assertGenerateSQLTokenWithUpdateParameterMarkerExpressionSegment() {
+
when(assignmentSegment.getValue()).thenReturn(parameterMarkerExpression);
+ Collection<EncryptAssignmentToken> resultCollection =
tokenGenerator.generateSQLTokens(updateStatement);
+ assertThat(resultCollection.size(), is(1));
+ }
+
+ @Test
+ public void assertGenerateSQLTokenWithUpdateLiteralExpressionSegment() {
+ when(assignmentSegment.getValue()).thenReturn(literalExpression);
+ Collection<EncryptAssignmentToken> resultCollection =
tokenGenerator.generateSQLTokens(updateStatement);
+ assertThat(resultCollection.size(), is(1));
+ }
+
+ @Test
+ public void assertGenerateSQLTokenWithUpdateEmpty() {
+ when(assignmentSegment.getValue()).thenReturn(null);
+ Collection<EncryptAssignmentToken> resultCollection =
tokenGenerator.generateSQLTokens(updateStatement);
+ assertThat(resultCollection.size(), is(0));
+ }
+
+ @Test
+ public void assertGenerateSQLTokenWithInsertLiteralExpressionSegment() {
+ MockedStatic<InsertStatementHandler>
insertStatementHandlerMockedStatic = mockStatic(InsertStatementHandler.class);
+ insertStatementHandlerMockedStatic.when(() ->
InsertStatementHandler.getSetAssignmentSegment(any())).thenReturn(Optional.of(setAssignmentSegment));
+ when(assignmentSegment.getValue()).thenReturn(literalExpression);
+ Collection<EncryptAssignmentToken> resultCollection =
tokenGenerator.generateSQLTokens(insertStatement);
+ assertThat(resultCollection.size(), is(1));
+ }
+}
