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.assertEquals;
21 import static org.junit.Assert.assertTrue;
22
23 import org.apache.hadoop.conf.Configuration;
24 import org.apache.hadoop.hbase.HConstants;
25 import org.apache.hadoop.hbase.SmallTests;
26 import org.junit.Test;
27 import org.junit.experimental.categories.Category;
28
29
30
31
32 @Category(SmallTests.class)
33 public class TestCustomWALCellCodec {
34
35 public static class CustomWALCellCodec extends WALCellCodec {
36 public Configuration conf;
37 public CompressionContext context;
38
39 public CustomWALCellCodec(Configuration conf, CompressionContext compression) {
40 super(conf, compression);
41 this.conf = conf;
42 this.context = compression;
43 }
44 }
45
46
47
48
49
50
51 @Test
52 public void testCreatePreparesCodec() throws Exception {
53 Configuration conf = new Configuration(false);
54 conf.setClass(WALCellCodec.WAL_CELL_CODEC_CLASS_KEY, CustomWALCellCodec.class,
55 WALCellCodec.class);
56 CustomWALCellCodec codec = (CustomWALCellCodec) WALCellCodec.create(conf, null);
57 assertEquals("Custom codec didn't get initialized with the right configuration!", conf,
58 codec.conf);
59 assertEquals("Custom codec didn't get initialized with the right compression context!", null,
60 codec.context);
61 }
62 }