View Javadoc
1   /*
2    * Licensed to the Apache Software Foundation (ASF) under one or more
3    * contributor license agreements.  See the NOTICE file distributed with
4    * this work for additional information regarding copyright ownership.
5    * The ASF licenses this file to You under the Apache License, Version 2.0
6    * (the "License"); you may not use this file except in compliance with
7    * the License.  You may obtain a copy of the License at
8    *
9    *     http://www.apache.org/licenses/LICENSE-2.0
10   *
11   * Unless required by applicable law or agreed to in writing, software
12   * distributed under the License is distributed on an "AS IS" BASIS,
13   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14   * See the License for the specific language governing permissions and
15   * limitations under the License.
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.ShowTablesStatement;
23  import org.apache.shardingsphere.single.distsql.statement.rql.ShowSingleTableStatement;
24  import org.apache.shardingsphere.test.it.sql.parser.internal.asserts.SQLCaseAssertContext;
25  import org.apache.shardingsphere.test.it.sql.parser.internal.asserts.segment.database.DatabaseAssert;
26  import org.apache.shardingsphere.test.it.sql.parser.internal.asserts.statement.rql.impl.table.ShowSingleTablesStatementAssert;
27  import org.apache.shardingsphere.test.it.sql.parser.internal.cases.parser.jaxb.SQLParserTestCase;
28  import org.apache.shardingsphere.test.it.sql.parser.internal.cases.parser.jaxb.statement.rql.table.ShowTableStatementTestCase;
29  
30  import static org.hamcrest.CoreMatchers.instanceOf;
31  import static org.hamcrest.CoreMatchers.is;
32  import static org.hamcrest.MatcherAssert.assertThat;
33  import static org.junit.jupiter.api.Assertions.assertFalse;
34  import static org.junit.jupiter.api.Assertions.assertTrue;
35  
36  /**
37   * Show tables statement assert.
38   */
39  @NoArgsConstructor(access = AccessLevel.PRIVATE)
40  public final class ShowTablesStatementAssert {
41      
42      /**
43       * Assert show tables statement is correct with expected parser result.
44       *
45       * @param assertContext assert context
46       * @param actual actual show tables statement
47       * @param expected expected show tables statement test case
48       */
49      public static void assertIs(final SQLCaseAssertContext assertContext, final ShowTablesStatement actual, final SQLParserTestCase expected) {
50          assertThat("Expected value should be ShowTableStatementTestCase", expected, instanceOf(ShowTableStatementTestCase.class));
51          assertIs(assertContext, actual, (ShowTableStatementTestCase) expected);
52          if (actual instanceof ShowSingleTableStatement) {
53              ShowSingleTablesStatementAssert.assertIs(assertContext, (ShowSingleTableStatement) actual, expected);
54          }
55      }
56      
57      private static void assertIs(final SQLCaseAssertContext assertContext, final ShowTablesStatement actual, final ShowTableStatementTestCase expected) {
58          if (null == expected.getDatabase()) {
59              assertFalse(actual.getDatabase().isPresent(), assertContext.getText("Actual database should not exist."));
60          } else {
61              assertTrue(actual.getDatabase().isPresent(), assertContext.getText("Actual database should exist."));
62              DatabaseAssert.assertIs(assertContext, actual.getDatabase().get(), expected.getDatabase());
63          }
64          if (null == expected.getLikePattern()) {
65              assertFalse(actual.getLikePattern().isPresent(), assertContext.getText("Actual like pattern should not exist."));
66          } else {
67              assertTrue(actual.getLikePattern().isPresent(), assertContext.getText("Actual like pattern should exist."));
68              assertThat(assertContext.getText("Like pattern assertion error"), actual.getLikePattern().get(), is(expected.getLikePattern()));
69          }
70      }
71  }