View Javadoc

1   /**
2    * Licensed to the Apache Software Foundation (ASF) under one
3    * or more contributor license agreements.  See the NOTICE file
4    * distributed with this work for additional information
5    * regarding copyright ownership.  The ASF licenses this file
6    * to you under the Apache License, Version 2.0 (the
7    * "License"); you may not use this file except in compliance
8    * with the License.  You may obtain a copy of the License at
9    *
10   *     http://www.apache.org/licenses/LICENSE-2.0
11   *
12   * Unless required by applicable law or agreed to in writing, software
13   * distributed under the License is distributed on an "AS IS" BASIS,
14   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15   * See the License for the specific language governing permissions and
16   * limitations under the License.
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.testclassification.SmallTests;
25  import org.junit.Test;
26  import org.junit.experimental.categories.Category;
27  
28  /**
29   * Test that we can create, load, setup our own custom codec
30   */
31  @Category(SmallTests.class)
32  public class TestCustomWALCellCodec {
33  
34    public static class CustomWALCellCodec extends WALCellCodec {
35      public Configuration conf;
36      public CompressionContext context;
37  
38      public CustomWALCellCodec(Configuration conf, CompressionContext compression) {
39        super(conf, compression);
40        this.conf = conf;
41        this.context = compression;
42      }
43    }
44  
45    /**
46     * Test that a custom {@link WALCellCodec} will be completely setup when it is instantiated via
47     * {@link WALCellCodec}
48     * @throws Exception on failure
49     */
50    @Test
51    public void testCreatePreparesCodec() throws Exception {
52      Configuration conf = new Configuration(false);
53      conf.setClass(WALCellCodec.WAL_CELL_CODEC_CLASS_KEY, CustomWALCellCodec.class,
54        WALCellCodec.class);
55      CustomWALCellCodec codec = (CustomWALCellCodec) WALCellCodec.create(conf, null, null);
56      assertEquals("Custom codec didn't get initialized with the right configuration!", conf,
57        codec.conf);
58      assertEquals("Custom codec didn't get initialized with the right compression context!", null,
59        codec.context);
60    }
61  }