1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18 package org.apache.shardingsphere.test.it.sql.parser.internal.asserts.statement.rdl.alter.impl;
19
20 import lombok.AccessLevel;
21 import lombok.NoArgsConstructor;
22 import org.apache.shardingsphere.shadow.distsql.segment.ShadowRuleSegment;
23 import org.apache.shardingsphere.shadow.distsql.statement.AlterShadowRuleStatement;
24 import org.apache.shardingsphere.test.it.sql.parser.internal.asserts.SQLCaseAssertContext;
25 import org.apache.shardingsphere.test.it.sql.parser.internal.asserts.segment.distsql.rdl.ShadowRuleAssert;
26 import org.apache.shardingsphere.test.it.sql.parser.internal.asserts.statement.ExistingAssert;
27 import org.apache.shardingsphere.test.it.sql.parser.internal.cases.parser.jaxb.segment.impl.distsql.rdl.ExpectedShadowRule;
28 import org.apache.shardingsphere.test.it.sql.parser.internal.cases.parser.jaxb.statement.rdl.rule.shadow.AlterShadowRuleStatementTestCase;
29
30 import java.util.Collection;
31 import java.util.List;
32
33 import static org.junit.jupiter.api.Assertions.assertNotNull;
34 import static org.junit.jupiter.api.Assertions.assertNull;
35
36
37
38
39 @NoArgsConstructor(access = AccessLevel.PRIVATE)
40 public final class AlterShadowRuleStatementAssert {
41
42
43
44
45
46
47
48
49 public static void assertIs(final SQLCaseAssertContext assertContext, final AlterShadowRuleStatement actual, final AlterShadowRuleStatementTestCase expected) {
50 if (ExistingAssert.assertIs(assertContext, actual, expected)) {
51 assertShadowRule(assertContext, actual.getRules(), expected.getRules());
52 }
53 }
54
55 private static void assertShadowRule(final SQLCaseAssertContext assertContext, final Collection<ShadowRuleSegment> actual, final List<ExpectedShadowRule> expected) {
56 if (null == expected) {
57 assertNull(actual, assertContext.getText("Actual shadow rule should not exist."));
58 } else {
59 assertNotNull(actual, assertContext.getText("Actual shadow rule should exist."));
60 int count = 0;
61 for (ShadowRuleSegment each : actual) {
62 ShadowRuleAssert.assertIs(assertContext, each, expected.get(count));
63 count++;
64 }
65 }
66 }
67 }