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.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   * Show storage units statement assert.
36   */
37  @NoArgsConstructor(access = AccessLevel.PRIVATE)
38  public final class ShowStorageUnitsStatementAssert {
39      
40      /**
41       * Assert show storage units statement is correct with expected parser result.
42       *
43       * @param assertContext assert context
44       * @param actual actual show storage units statement
45       * @param expected expected show storage units statement test case
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  }