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.segment.assignment;
19
20 import lombok.AccessLevel;
21 import lombok.NoArgsConstructor;
22 import org.apache.shardingsphere.sql.parser.sql.common.segment.dml.assignment.ColumnAssignmentSegment;
23 import org.apache.shardingsphere.sql.parser.sql.common.segment.dml.column.ColumnSegment;
24 import org.apache.shardingsphere.test.it.sql.parser.internal.asserts.SQLCaseAssertContext;
25 import org.apache.shardingsphere.test.it.sql.parser.internal.asserts.segment.column.ColumnAssert;
26 import org.apache.shardingsphere.test.it.sql.parser.internal.cases.parser.jaxb.segment.impl.assignment.ExpectedAssignment;
27
28 import static org.hamcrest.CoreMatchers.is;
29 import static org.hamcrest.MatcherAssert.assertThat;
30
31
32
33
34 @NoArgsConstructor(access = AccessLevel.PRIVATE)
35 public final class AssignmentAssert {
36
37
38
39
40
41
42
43
44 public static void assertIs(final SQLCaseAssertContext assertContext, final ColumnAssignmentSegment actual, final ExpectedAssignment expected) {
45 if (expected.getColumns().isEmpty()) {
46 ColumnAssert.assertIs(assertContext, actual.getColumns().get(0), expected.getColumn());
47 } else {
48 assertThat(assertContext.getText("Assignment columns size assertion error: "), actual.getColumns().size(), is(expected.getColumns().size()));
49 int count = 0;
50 for (ColumnSegment each : actual.getColumns()) {
51 ColumnAssert.assertIs(assertContext, each, expected.getColumns().get(count));
52 count++;
53 }
54 }
55
56 AssignmentValueAssert.assertIs(assertContext, actual.getValue(), expected.getAssignmentValue());
57 }
58 }