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.create.impl;
19
20 import lombok.AccessLevel;
21 import lombok.NoArgsConstructor;
22 import org.apache.shardingsphere.encrypt.distsql.segment.EncryptRuleSegment;
23 import org.apache.shardingsphere.encrypt.distsql.statement.CreateEncryptRuleStatement;
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.EncryptRuleAssert;
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.ExpectedEncryptRule;
28 import org.apache.shardingsphere.test.it.sql.parser.internal.cases.parser.jaxb.statement.rdl.rule.encrypt.CreateEncryptRuleStatementTestCase;
29
30 import java.util.Collection;
31 import java.util.List;
32
33 import static org.hamcrest.CoreMatchers.is;
34 import static org.junit.jupiter.api.Assertions.assertNotNull;
35 import static org.junit.jupiter.api.Assertions.assertNull;
36 import static org.hamcrest.MatcherAssert.assertThat;
37
38
39
40
41 @NoArgsConstructor(access = AccessLevel.PRIVATE)
42 public final class CreateEncryptRuleStatementAssert {
43
44
45
46
47
48
49
50
51 public static void assertIs(final SQLCaseAssertContext assertContext, final CreateEncryptRuleStatement actual, final CreateEncryptRuleStatementTestCase expected) {
52 if (ExistingAssert.assertIs(assertContext, actual, expected)) {
53 assertThat(assertContext.getText("if not exists segment assertion error: "), actual.isIfNotExists(), is(expected.isIfNotExists()));
54 assertEncryptRules(assertContext, actual.getRules(), expected.getRules());
55 }
56 }
57
58 private static void assertEncryptRules(final SQLCaseAssertContext assertContext, final Collection<EncryptRuleSegment> actual, final List<ExpectedEncryptRule> expected) {
59 if (null == expected) {
60 assertNull(actual, assertContext.getText("Actual encrypt rule should not exist."));
61 } else {
62 assertNotNull(actual, assertContext.getText("Actual encrypt rule should exist."));
63 assertThat(assertContext.getText(String.format("Actual encrypt rule size should be %s , but it was %s",
64 expected.size(), actual.size())), actual.size(), is(expected.size()));
65 int count = 0;
66 for (EncryptRuleSegment each : actual) {
67 ExpectedEncryptRule expectedEncryptRule = expected.get(count);
68 assertThat(assertContext.getText("encrypt rule assertion error: "), each.getTableName(), is(expectedEncryptRule.getName()));
69 EncryptRuleAssert.assertIs(assertContext, each, expectedEncryptRule);
70 count++;
71 }
72 }
73 }
74 }