[shardingsphere] branch master updated: #12495 - Added unit test for ShadowUpdateStatementRoutingEngine in shadow project (#13133)

[me...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

menghaoran 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 ed88342  #12495 - Added unit test for ShadowUpdateStatementRoutingEngine in shadow project (#13133)
ed88342 is described below

commit ed88342028bb6753b4cb55ee479555d9a9301a9f
Author: Oresztesz Margaritisz <gi...@users.noreply.github.com>
AuthorDate: Tue Oct 19 13:37:18 2021 +0200

    #12495 - Added unit test for ShadowUpdateStatementRoutingEngine in shadow project (#13133)
    
    * #12495 - Added ShadowUpdateStatementRoutingEngineTest
    
    * #12495 - Fixed review items
---
 .../ShadowUpdateStatementRoutingEngineTest.java    | 162 +++++++++++++++++++++
 1 file changed, 162 insertions(+)

diff --git a/shardingsphere-features/shardingsphere-shadow/shardingsphere-shadow-core/src/test/java/org/apache/shardingsphere/shadow/route/engine/dml/ShadowUpdateStatementRoutingEngineTest.java b/shardingsphere-features/shardingsphere-shadow/shardingsphere-shadow-core/src/test/java/org/apache/shardingsphere/shadow/route/engine/dml/ShadowUpdateStatementRoutingEngineTest.java
new file mode 100644
index 0000000..0f97458
--- /dev/null
+++ b/shardingsphere-features/shardingsphere-shadow/shardingsphere-shadow-core/src/test/java/org/apache/shardingsphere/shadow/route/engine/dml/ShadowUpdateStatementRoutingEngineTest.java
@@ -0,0 +1,162 @@
+/*
+ * 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.shadow.route.engine.dml;
+
+import org.apache.shardingsphere.infra.binder.statement.dml.UpdateStatementContext;
+import org.apache.shardingsphere.infra.route.context.RouteContext;
+import org.apache.shardingsphere.infra.route.context.RouteMapper;
+import org.apache.shardingsphere.infra.route.context.RouteUnit;
+import org.apache.shardingsphere.shadow.algorithm.config.AlgorithmProvidedShadowRuleConfiguration;
+import org.apache.shardingsphere.shadow.algorithm.shadow.column.ColumnRegexMatchShadowAlgorithm;
+import org.apache.shardingsphere.shadow.api.config.datasource.ShadowDataSourceConfiguration;
+import org.apache.shardingsphere.shadow.api.config.table.ShadowTableConfiguration;
+import org.apache.shardingsphere.shadow.api.shadow.ShadowOperationType;
+import org.apache.shardingsphere.shadow.condition.ShadowColumnCondition;
+import org.apache.shardingsphere.shadow.rule.ShadowRule;
+import org.apache.shardingsphere.shadow.spi.ShadowAlgorithm;
+import org.apache.shardingsphere.sql.parser.sql.common.segment.dml.column.ColumnSegment;
+import org.apache.shardingsphere.sql.parser.sql.common.segment.dml.expr.BinaryOperationExpression;
+import org.apache.shardingsphere.sql.parser.sql.common.segment.dml.expr.simple.LiteralExpressionSegment;
+import org.apache.shardingsphere.sql.parser.sql.common.segment.dml.predicate.WhereSegment;
+import org.apache.shardingsphere.sql.parser.sql.common.segment.generic.CommentSegment;
+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.value.identifier.IdentifierValue;
+import org.apache.shardingsphere.sql.parser.sql.dialect.statement.mysql.dml.MySQLUpdateStatement;
+import org.junit.Before;
+import org.junit.Test;
+
+import java.util.Collection;
+import java.util.Collections;
+import java.util.Iterator;
+import java.util.LinkedHashMap;
+import java.util.LinkedList;
+import java.util.Map;
+import java.util.Optional;
+import java.util.Properties;
+
+import static org.hamcrest.CoreMatchers.is;
+import static org.junit.Assert.assertThat;
+import static org.junit.Assert.assertTrue;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
+
+public final class ShadowUpdateStatementRoutingEngineTest {
+    
+    private ShadowUpdateStatementRoutingEngine shadowRouteEngine;
+    
+    @Before
+    public void init() {
+        shadowRouteEngine = new ShadowUpdateStatementRoutingEngine(createUpdateStatementContext(), Collections.emptyList());
+    }
+    
+    private UpdateStatementContext createUpdateStatementContext() {
+        UpdateStatementContext result = mock(UpdateStatementContext.class);
+        Collection<SimpleTableSegment> allTables = new LinkedList<>();
+        allTables.add(new SimpleTableSegment(new TableNameSegment(20, 25, new IdentifierValue("t_order"))));
+        when(result.getAllTables()).thenReturn(allTables);
+        BinaryOperationExpression binaryOperationExpression = mock(BinaryOperationExpression.class);
+        when(binaryOperationExpression.getLeft()).thenReturn(new ColumnSegment(0, 0, new IdentifierValue("user_id")));
+        when(binaryOperationExpression.getRight()).thenReturn(new LiteralExpressionSegment(0, 0, "1"));
+        WhereSegment whereSegment = new WhereSegment(0, 0, binaryOperationExpression);
+        when(result.getWhere()).thenReturn(Optional.of(whereSegment));
+        MySQLUpdateStatement updateStatement = new MySQLUpdateStatement();
+        Collection<CommentSegment> commentSegments = new LinkedList<>();
+        commentSegments.add(new CommentSegment("/*shadow:true,foo:bar*/", 0, 20));
+        commentSegments.add(new CommentSegment("/*aaa:bbb*/", 21, 30));
+        updateStatement.setCommentSegments(commentSegments);
+        when(result.getSqlStatement()).thenReturn(updateStatement);
+        return result;
+    }
+    
+    @Test
+    public void assertRouteAndParseShadowColumnConditions() {
+        RouteContext routeContext = mock(RouteContext.class);
+        Collection<RouteUnit> routeUnits = new LinkedList<>();
+        routeUnits.add(new RouteUnit(new RouteMapper("ds", "ds_shadow"), new LinkedList<>()));
+        when(routeContext.getRouteUnits()).thenReturn(routeUnits);
+        shadowRouteEngine.route(routeContext, new ShadowRule(createAlgorithmProvidedShadowRuleConfiguration()));
+        Optional<Collection<ShadowColumnCondition>> shadowColumnConditions = shadowRouteEngine.parseShadowColumnConditions();
+        assertTrue(shadowColumnConditions.isPresent());
+        Collection<ShadowColumnCondition> shadowColumns = shadowColumnConditions.get();
+        assertThat(shadowColumns.size(), is(1));
+        Iterator<ShadowColumnCondition> iterator = shadowColumns.iterator();
+        ShadowColumnCondition userId = iterator.next();
+        assertThat(userId.getColumn(), is("user_id"));
+        assertThat(userId.getTable(), is("t_order"));
+        assertThat(userId.getValues().iterator().next(), is("1"));
+        Optional<Collection<String>> sqlNotes = shadowRouteEngine.parseSqlNotes();
+        assertTrue(sqlNotes.isPresent());
+        assertThat(sqlNotes.get().size(), is(2));
+        Iterator<String> sqlNotesIt = sqlNotes.get().iterator();
+        assertThat(sqlNotesIt.next(), is("/*shadow:true,foo:bar*/"));
+        assertThat(sqlNotesIt.next(), is("/*aaa:bbb*/"));
+    }
+
+    private AlgorithmProvidedShadowRuleConfiguration createAlgorithmProvidedShadowRuleConfiguration() {
+        AlgorithmProvidedShadowRuleConfiguration result = new AlgorithmProvidedShadowRuleConfiguration();
+        result.setEnable(true);
+        result.setDataSources(createDataSources());
+        result.setTables(createTables());
+        result.setShadowAlgorithms(createShadowAlgorithms());
+        return result;
+    }
+
+    private Map<String, ShadowAlgorithm> createShadowAlgorithms() {
+        Map<String, ShadowAlgorithm> result = new LinkedHashMap<>();
+        result.put("user-id-update-regex-algorithm", createColumnShadowAlgorithm());
+        return result;
+    }
+
+    private ShadowAlgorithm createColumnShadowAlgorithm() {
+        Properties properties = new Properties();
+        properties.setProperty("column", "user_id");
+        properties.setProperty("operation", "update");
+        properties.setProperty("regex", "[1]");
+        ColumnRegexMatchShadowAlgorithm columnRegexMatchShadowAlgorithm = new ColumnRegexMatchShadowAlgorithm();
+        columnRegexMatchShadowAlgorithm.setProps(properties);
+        columnRegexMatchShadowAlgorithm.init();
+        return columnRegexMatchShadowAlgorithm;
+    }
+
+    private Map<String, ShadowTableConfiguration> createTables() {
+        Map<String, ShadowTableConfiguration> result = new LinkedHashMap<>();
+        Collection<String> shadowAlgorithmNames = new LinkedList<>();
+        shadowAlgorithmNames.add("user-id-update-regex-algorithm");
+        result.put("t_order", new ShadowTableConfiguration(Collections.singletonList("shadow-data-source-0"), shadowAlgorithmNames));
+        return result;
+    }
+
+    private Map<String, ShadowDataSourceConfiguration> createDataSources() {
+        Map<String, ShadowDataSourceConfiguration> result = new LinkedHashMap<>();
+        result.put("shadow-data-source-0", new ShadowDataSourceConfiguration("ds", "ds_shadow"));
+        return result;
+    }
+    
+    @Test
+    public void assertCreateShadowDetermineCondition() {
+        assertThat(shadowRouteEngine.createShadowDetermineCondition().getShadowOperationType(), is(ShadowOperationType.UPDATE));
+    }
+    
+    @Test
+    public void assertGetAllTables() {
+        Collection<SimpleTableSegment> allTables = shadowRouteEngine.getAllTables();
+        assertThat(allTables.size(), is(1));
+        assertThat(allTables.iterator().next().getTableName().getIdentifier().getValue(), is("t_order"));
+    }
+}