Method before

test/BaseTestHBaseClient.java:130–188

()

  protected FakeTimer timer;
  
  @Before
  public void before() throws Exception {
    region_clients.clear();
    rootclient = mock(RegionClient.class);
    when(rootclient.toString()).thenReturn(MOCK_ROOT_CLIENT_NAME);
    metaclient = mock(RegionClient.class);
    when(metaclient.toString()).thenReturn(MOCK_META_CLIENT_NAME);
    regionclient = mock(RegionClient.class);
    when(regionclient.toString()).thenReturn(MOCK_RS_CLIENT_NAME);
    zkclient = mock(ZKClient.class);
    channel_factory = mock(NioClientSocketChannelFactory.class);
    chan = mock(SocketChannel.class);
    timer = new FakeTimer();
    
    when(zkclient.getDeferredRoot()).thenReturn(new Deferred<Object>());
    PowerMockito.mockStatic(Executors.class);
    PowerMockito.when(Executors.defaultThreadFactory())
      .thenReturn(mock(ThreadFactory.class));
    PowerMockito.when(Executors.newCachedThreadPool())
      .thenReturn(mock(ExecutorService.class));
    PowerMockito.whenNew(NioClientSocketChannelFactory.class).withAnyArguments()
      .thenReturn(channel_factory);
    
    PowerMockito.whenNew(HashedWheelTimer.class).withAnyArguments()
      .thenReturn(timer);
    PowerMockito.whenNew(NioClientBossPool.class).withAnyArguments()
      .thenReturn(mock(NioClientBossPool.class));
    PowerMockito.whenNew(NioWorkerPool.class).withAnyArguments()
      .thenReturn(mock(NioWorkerPool.class));
    
    client = PowerMockito.spy(new HBaseClient("test-quorum-spec"));
    Whitebox.setInternalState(client, "zkclient", zkclient);
    Whitebox.setInternalState(client, "rootregion", rootclient);
    regions_cache = Whitebox.getInternalState(client, "regions_cache");
    region2client = Whitebox.getInternalState(client, "region2client");
    client2regions = Whitebox.getInternalState(client, "client2regions");
    got_nsre = Whitebox.getInternalState(client, "got_nsre");
    ip2client = Whitebox.getInternalState(client, "ip2client");
    injectRegionInCache(meta, metaclient, META_IP + ":" + RS_PORT);
    injectRegionInCache(region, regionclient, REGION_CLIENT_IP + ":" + RS_PORT);
    
    when(channel_factory.newChannel(any(ChannelPipeline.class)))
      .thenReturn(chan);
    when(chan.getConfig()).thenReturn(mock(SocketChannelConfig.class));
    when(rootclient.toString()).thenReturn("Mock RootClient");
    
    PowerMockito.doAnswer(new Answer<RegionClient>(){
      @Override
      public RegionClient answer(InvocationOnMock invocation) throws Throwable {
        final Object[] args = invocation.getArguments();
        final String endpoint = (String)args[0] + ":" + (Integer)args[1];
        final RegionClient rc = mock(RegionClient.class);
        when(rc.getRemoteAddress()).thenReturn(endpoint);
        client2regions.put(rc, new ArrayList<RegionInfo>());
        region_clients.add(rc);
        return rc;
      }
    }).when(client, "newClient", anyString(), anyInt());