diff --git a/hbase-client/src/test/java/org/apache/hadoop/hbase/client/TestAsyncProcess.java b/hbase-client/src/test/java/org/apache/hadoop/hbase/client/TestAsyncProcess.java index 067f2ad..087bf62 100644 --- a/hbase-client/src/test/java/org/apache/hadoop/hbase/client/TestAsyncProcess.java +++ b/hbase-client/src/test/java/org/apache/hadoop/hbase/client/TestAsyncProcess.java @@ -71,7 +71,8 @@ import org.mockito.Mockito; @Category({ClientTests.class, MediumTests.class}) public class TestAsyncProcess { - @Rule public final TestRule timeout = CategoryBasedTimeout.builder().withTimeout(this.getClass()). + @Rule public final TestRule timeout = CategoryBasedTimeout.builder(). + withTimeout(50, TimeUnit.SECONDS). withLookingForStuckThread(true).build(); private final static Log LOG = LogFactory.getLog(TestAsyncProcess.class); private static final TableName DUMMY_TABLE = diff --git a/hbase-common/src/test/java/org/apache/hadoop/hbase/CategoryBasedTimeout.java b/hbase-common/src/test/java/org/apache/hadoop/hbase/CategoryBasedTimeout.java index 7fd286f..dd2c26c 100644 --- a/hbase-common/src/test/java/org/apache/hadoop/hbase/CategoryBasedTimeout.java +++ b/hbase-common/src/test/java/org/apache/hadoop/hbase/CategoryBasedTimeout.java @@ -65,6 +65,8 @@ public class CategoryBasedTimeout extends Timeout { } else if (c == LargeTests.class) { // Let large tests have a ten minute timeout. return withTimeout(10, TimeUnit.MINUTES); + } else { + throw new RuntimeException("Unknown category: " + c); } } }