View Javadoc

1   /**
2    * Licensed to the Apache Software Foundation (ASF) under one
3    * or more contributor license agreements.  See the NOTICE file
4    * distributed with this work for additional information
5    * regarding copyright ownership.  The ASF licenses this file
6    * to you under the Apache License, Version 2.0 (the
7    * "License"); you may not use this file except in compliance
8    * with the License.  You may obtain a copy of the License at
9    *
10   *     http://www.apache.org/licenses/LICENSE-2.0
11   *
12   * Unless required by applicable law or agreed to in writing, software
13   * distributed under the License is distributed on an "AS IS" BASIS,
14   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15   * See the License for the specific language governing permissions and
16   * limitations under the License.
17   */
18  package org.apache.hadoop.hbase.ipc;
19  
20  import static org.apache.hadoop.hbase.ipc.IPCUtil.wrapException;
21  import static org.junit.Assert.assertTrue;
22  
23  import java.net.ConnectException;
24  import java.net.SocketTimeoutException;
25  
26  import org.apache.hadoop.hbase.exceptions.ConnectionClosingException;
27  import org.apache.hadoop.hbase.net.Address;
28  import org.apache.hadoop.hbase.testclassification.ClientTests;
29  import org.apache.hadoop.hbase.testclassification.SmallTests;
30  import org.junit.Test;
31  import org.junit.experimental.categories.Category;
32  
33  @Category({ ClientTests.class, SmallTests.class })
34  public class TestIPCUtil {
35  
36    @Test
37    public void testWrapException() throws Exception {
38      final Address address = Address.fromParts("localhost", 0);
39      assertTrue(wrapException(address, new ConnectException()) instanceof ConnectException);
40      assertTrue(
41        wrapException(address, new SocketTimeoutException()) instanceof SocketTimeoutException);
42      assertTrue(wrapException(address, new ConnectionClosingException(
43          "Test AbstractRpcClient#wrapException")) instanceof ConnectionClosingException);
44      assertTrue(
45        wrapException(address, new CallTimeoutException("Test AbstractRpcClient#wrapException"))
46            .getCause() instanceof CallTimeoutException);
47    }
48  }