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;
19
20 import lombok.AccessLevel;
21 import lombok.NoArgsConstructor;
22 import org.apache.shardingsphere.distsql.segment.DataSourceSegment;
23 import org.apache.shardingsphere.distsql.statement.rdl.resource.unit.type.AlterStorageUnitStatement;
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.DataSourceAssert;
26 import org.apache.shardingsphere.test.it.sql.parser.internal.cases.parser.jaxb.segment.impl.distsql.ExpectedDataSource;
27 import org.apache.shardingsphere.test.it.sql.parser.internal.cases.parser.jaxb.statement.rdl.resource.AlterStorageUnitStatementTestCase;
28
29 import java.util.Collection;
30 import java.util.List;
31
32 import static org.hamcrest.CoreMatchers.is;
33 import static org.hamcrest.MatcherAssert.assertThat;
34 import static org.junit.jupiter.api.Assertions.assertNotNull;
35 import static org.junit.jupiter.api.Assertions.assertNull;
36
37
38
39
40 @NoArgsConstructor(access = AccessLevel.PRIVATE)
41 public final class AlterStorageUnitStatementAssert {
42
43
44
45
46
47
48
49
50 public static void assertIs(final SQLCaseAssertContext assertContext, final AlterStorageUnitStatement actual, final AlterStorageUnitStatementTestCase expected) {
51 if (null == expected) {
52 assertNull(actual, assertContext.getText("Actual statement should not exist."));
53 } else {
54 assertNotNull(actual, assertContext.getText("Actual statement should exist."));
55 assertDataSources(assertContext, actual.getStorageUnits(), expected.getDataSources());
56 }
57 }
58
59 private static void assertDataSources(final SQLCaseAssertContext assertContext, final Collection<DataSourceSegment> actual, final List<ExpectedDataSource> expected) {
60 if (null == expected) {
61 assertNull(actual, assertContext.getText("Actual storage unit should not exist."));
62 } else {
63 assertNotNull(actual, assertContext.getText("Actual storage unit should exist."));
64 assertThat(assertContext.getText(String.format("Actual storage unit size should be %s , but it was %s", expected.size(), actual.size())), actual.size(), is(expected.size()));
65 int count = 0;
66 for (DataSourceSegment each : actual) {
67 DataSourceAssert.assertIs(assertContext, each, expected.get(count));
68 count++;
69 }
70 }
71 }
72 }