001/** 002 * Licensed to the Apache Software Foundation (ASF) under one 003 * or more contributor license agreements. See the NOTICE file 004 * distributed with this work for additional information 005 * regarding copyright ownership. The ASF licenses this file 006 * to you under the Apache License, Version 2.0 (the 007 * "License"); you may not use this file except in compliance 008 * with the License. You may obtain a copy of the License at 009 * 010 * http://www.apache.org/licenses/LICENSE-2.0 011 * 012 * Unless required by applicable law or agreed to in writing, software 013 * distributed under the License is distributed on an "AS IS" BASIS, 014 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 015 * See the License for the specific language governing permissions and 016 * limitations under the License. 017 */ 018package org.apache.hadoop.hbase.ipc; 019 020import org.apache.hadoop.hbase.HBaseClassTestRule; 021import org.apache.hadoop.hbase.monitoring.MonitoredRPCHandlerImpl; 022import org.apache.hadoop.hbase.testclassification.RPCTests; 023import org.apache.hadoop.hbase.testclassification.SmallTests; 024import org.junit.ClassRule; 025import org.junit.Test; 026import org.junit.experimental.categories.Category; 027import org.mockito.Mockito; 028 029@Category({RPCTests.class, SmallTests.class}) 030public class TestCallRunner { 031 032 @ClassRule 033 public static final HBaseClassTestRule CLASS_RULE = 034 HBaseClassTestRule.forClass(TestCallRunner.class); 035 036 /** 037 * Does nothing but exercise a {@link CallRunner} outside of {@link RpcServer} context. 038 */ 039 @Test 040 public void testSimpleCall() { 041 RpcServerInterface mockRpcServer = Mockito.mock(RpcServerInterface.class); 042 Mockito.when(mockRpcServer.isStarted()).thenReturn(true); 043 ServerCall mockCall = Mockito.mock(ServerCall.class); 044 CallRunner cr = new CallRunner(mockRpcServer, mockCall); 045 cr.setStatus(new MonitoredRPCHandlerImpl()); 046 cr.run(); 047 } 048 049 @Test 050 public void testCallCleanup() { 051 RpcServerInterface mockRpcServer = Mockito.mock(RpcServerInterface.class); 052 Mockito.when(mockRpcServer.isStarted()).thenReturn(true); 053 ServerCall mockCall = Mockito.mock(ServerCall.class); 054 Mockito.when(mockCall.disconnectSince()).thenReturn(1L); 055 056 CallRunner cr = new CallRunner(mockRpcServer, mockCall); 057 cr.setStatus(new MonitoredRPCHandlerImpl()); 058 cr.run(); 059 Mockito.verify(mockCall, Mockito.times(1)).cleanup(); 060 } 061 062 @Test 063 public void testCallRunnerDrop() { 064 RpcServerInterface mockRpcServer = Mockito.mock(RpcServerInterface.class); 065 Mockito.when(mockRpcServer.isStarted()).thenReturn(true); 066 ServerCall mockCall = Mockito.mock(ServerCall.class); 067 Mockito.when(mockCall.disconnectSince()).thenReturn(1L); 068 069 CallRunner cr = new CallRunner(mockRpcServer, mockCall); 070 cr.setStatus(new MonitoredRPCHandlerImpl()); 071 cr.drop(); 072 Mockito.verify(mockCall, Mockito.times(1)).cleanup(); 073 } 074}