1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18 package org.apache.hadoop.hbase.regionserver.wal;
19
20 import static org.junit.Assert.assertTrue;
21
22 import org.apache.hadoop.conf.Configuration;
23 import org.apache.hadoop.hbase.SmallTests;
24 import org.junit.Test;
25 import org.junit.experimental.categories.Category;
26
27
28
29
30 @Category(SmallTests.class)
31 public class TestCustomWALEditCodec {
32
33 public static class CustomWALEditCodec extends WALEditCodec {
34 public boolean initialized = false;
35 public boolean compressionSet = false;
36
37 @Override
38 public void init(Configuration conf) {
39 this.initialized = true;
40 }
41
42 @Override
43 public void setCompression(CompressionContext compression) {
44 this.compressionSet = true;
45 }
46 }
47
48
49
50
51
52
53 @Test
54 public void testCreatePreparesCodec() throws Exception {
55 Configuration conf = new Configuration(false);
56 conf.setClass(WALEditCodec.WAL_EDIT_CODEC_CLASS_KEY, CustomWALEditCodec.class, WALEditCodec.class);
57 CustomWALEditCodec codec = (CustomWALEditCodec) WALEditCodec.create(conf, null);
58 assertTrue("Custom codec didn't get initialized", codec.initialized);
59 assertTrue("Custom codec didn't have compression set", codec.compressionSet);
60 }
61 }