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.util; 019 020import static org.junit.Assert.assertTrue; 021 022import java.util.concurrent.atomic.AtomicBoolean; 023import org.apache.hadoop.hbase.util.PoolMap.PoolType; 024import org.junit.After; 025import org.junit.Before; 026 027public abstract class PoolMapTestBase { 028 029 protected PoolMap<String, String> poolMap; 030 031 protected static final int POOL_SIZE = 3; 032 033 @Before 034 public void setUp() throws Exception { 035 this.poolMap = new PoolMap<>(getPoolType(), POOL_SIZE); 036 } 037 038 @After 039 public void tearDown() throws Exception { 040 this.poolMap.clear(); 041 } 042 043 protected abstract PoolType getPoolType(); 044 045 protected void runThread(final String randomKey, final String randomValue, 046 final String expectedValue) throws InterruptedException { 047 final AtomicBoolean matchFound = new AtomicBoolean(false); 048 Thread thread = new Thread(new Runnable() { 049 @Override 050 public void run() { 051 poolMap.put(randomKey, randomValue); 052 String actualValue = poolMap.get(randomKey); 053 matchFound 054 .set(expectedValue == null ? actualValue == null : expectedValue.equals(actualValue)); 055 } 056 }); 057 thread.start(); 058 thread.join(); 059 assertTrue(matchFound.get()); 060 } 061}