1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20 package org.apache.hadoop.hbase.coprocessor;
21
22 import java.io.IOException;
23
24 import junit.framework.Assert;
25 import org.apache.commons.logging.Log;
26 import org.apache.commons.logging.LogFactory;
27 import org.apache.hadoop.conf.Configuration;
28 import org.apache.hadoop.hbase.*;
29 import org.apache.hadoop.hbase.client.HTable;
30 import org.apache.hadoop.hbase.client.Put;
31 import org.apache.hadoop.hbase.client.Durability;
32 import org.apache.hadoop.hbase.regionserver.HRegionServer;
33 import org.apache.hadoop.hbase.util.Bytes;
34 import org.apache.hadoop.hbase.regionserver.wal.WALEdit;
35 import org.junit.AfterClass;
36 import org.junit.BeforeClass;
37 import org.junit.Test;
38 import org.junit.experimental.categories.Category;
39
40 import static org.junit.Assert.*;
41
42
43
44
45
46
47
48 @Category(MediumTests.class)
49 public class TestRegionServerCoprocessorExceptionWithAbort {
50 static final Log LOG = LogFactory.getLog(TestRegionServerCoprocessorExceptionWithAbort.class);
51 private static final HBaseTestingUtility TEST_UTIL = new HBaseTestingUtility();
52 private static final TableName TABLE_NAME =
53 TableName.valueOf("observed_table");
54
55 @BeforeClass
56 public static void setupBeforeClass() throws Exception {
57
58 Configuration conf = TEST_UTIL.getConfiguration();
59 conf.setInt(HConstants.HBASE_CLIENT_RETRIES_NUMBER, 2);
60 conf.set(CoprocessorHost.REGION_COPROCESSOR_CONF_KEY, BuggyRegionObserver.class.getName());
61 conf.set("hbase.coprocessor.abortonerror", "true");
62 TEST_UTIL.startMiniCluster(3);
63 }
64
65 @AfterClass
66 public static void teardownAfterClass() throws Exception {
67 TEST_UTIL.shutdownMiniCluster();
68 }
69
70 @Test
71 public void testExceptionFromCoprocessorDuringPut()
72 throws IOException, InterruptedException {
73
74
75
76 TableName TEST_TABLE = TABLE_NAME;
77 byte[] TEST_FAMILY = Bytes.toBytes("aaa");
78
79 HTable table = TEST_UTIL.createTable(TEST_TABLE, TEST_FAMILY);
80 TEST_UTIL.createMultiRegions(table, TEST_FAMILY);
81 TEST_UTIL.waitUntilAllRegionsAssigned(TEST_TABLE);
82
83
84 final HRegionServer regionServer = TEST_UTIL.getRSForFirstRegionInTable(TEST_TABLE);
85
86 final byte[] ROW = Bytes.toBytes("aaa");
87 Put put = new Put(ROW);
88 put.add(TEST_FAMILY, ROW, ROW);
89
90 Assert.assertFalse("The region server should be available", regionServer.isAborted());
91 try {
92 table.put(put);
93 fail("The put should have failed, as the coprocessor is buggy");
94 } catch (IOException ignored) {
95
96 }
97 Assert.assertTrue("The region server should have aborted", regionServer.isAborted());
98 table.close();
99 }
100
101 public static class BuggyRegionObserver extends SimpleRegionObserver {
102 @Override
103 public void prePut(final ObserverContext<RegionCoprocessorEnvironment> c,
104 final Put put, final WALEdit edit,
105 final Durability durability) {
106 TableName tableName =
107 c.getEnvironment().getRegion().getRegionInfo().getTableName();
108 if (TABLE_NAME.equals(tableName)) {
109 throw new NullPointerException("Buggy coprocessor");
110 }
111 }
112 }
113
114 }