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.rql.impl;
19
20 import lombok.AccessLevel;
21 import lombok.NoArgsConstructor;
22 import org.apache.shardingsphere.distsql.statement.rql.resource.ShowStorageUnitsStatement;
23 import org.apache.shardingsphere.test.it.sql.parser.internal.asserts.SQLCaseAssertContext;
24 import org.apache.shardingsphere.test.it.sql.parser.internal.asserts.segment.database.DatabaseAssert;
25 import org.apache.shardingsphere.test.it.sql.parser.internal.cases.parser.jaxb.SQLParserTestCase;
26 import org.apache.shardingsphere.test.it.sql.parser.internal.cases.parser.jaxb.statement.rql.storage.unit.ShowStorageUnitsStatementTestCase;
27
28 import static org.hamcrest.CoreMatchers.instanceOf;
29 import static org.hamcrest.CoreMatchers.is;
30 import static org.hamcrest.MatcherAssert.assertThat;
31 import static org.junit.jupiter.api.Assertions.assertFalse;
32 import static org.junit.jupiter.api.Assertions.assertTrue;
33
34
35
36
37 @NoArgsConstructor(access = AccessLevel.PRIVATE)
38 public final class ShowStorageUnitsStatementAssert {
39
40
41
42
43
44
45
46
47 public static void assertIs(final SQLCaseAssertContext assertContext, final ShowStorageUnitsStatement actual, final SQLParserTestCase expected) {
48 assertThat("Expected value should be ShowStorageUnitsStatementTestCase", expected, instanceOf(ShowStorageUnitsStatementTestCase.class));
49 assertIs(assertContext, actual, (ShowStorageUnitsStatementTestCase) expected);
50 }
51
52 private static void assertIs(final SQLCaseAssertContext assertContext, final ShowStorageUnitsStatement actual, final ShowStorageUnitsStatementTestCase expected) {
53 if (null == expected.getDatabase()) {
54 assertFalse(actual.getDatabase().isPresent(), assertContext.getText("Actual database should not exist."));
55 } else {
56 assertTrue(actual.getDatabase().isPresent(), assertContext.getText("Actual database should exist."));
57 DatabaseAssert.assertIs(assertContext, actual.getDatabase().get(), expected.getDatabase());
58 }
59 if (null == expected.getUsageCount()) {
60 assertFalse(actual.getUsageCount().isPresent(), assertContext.getText("Actual usage count should not exist."));
61 } else {
62 assertTrue(actual.getUsageCount().isPresent(), assertContext.getText("Actual usage count should exist."));
63 assertThat(assertContext.getText("Usage count assertion error"), actual.getUsageCount().get(), is(expected.getUsageCount()));
64 }
65 }
66 }