com.moz.fiji.mapreduce.kvstore.TestKeyValueStoreConfiguration.java Source code

Introduction

Here is the source code for com.moz.fiji.mapreduce.kvstore.TestKeyValueStoreConfiguration.java

Source

/**
 * (c) Copyright 2012 WibiData, Inc.
 *
 * See the NOTICE file distributed with this work for additional
 * information regarding copyright ownership.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.moz.fiji.mapreduce.kvstore;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;

import java.util.Arrays;

import org.apache.hadoop.conf.Configuration;
import org.junit.Test;

import com.moz.fiji.mapreduce.kvstore.framework.KeyValueStoreConfiguration;

public class TestKeyValueStoreConfiguration {
    @Test
    public void testCopyFromConfiguration() {
        Configuration conf = new Configuration(false);
        conf.set("foo", "foo-value");
        conf.setInt("bar", 123);
        conf.setClass("qaz", String.class, Object.class);

        KeyValueStoreConfiguration kvStoreConf = KeyValueStoreConfiguration.fromConf(conf);
        assertEquals("foo-value", kvStoreConf.get("foo"));
        assertEquals(123, kvStoreConf.getInt("bar", 0));
        assertEquals(String.class, kvStoreConf.getClass("qaz", null));
    }

    @Test
    public void testStoreString() {
        Configuration parent = new Configuration(false);
        KeyValueStoreConfiguration isolated = KeyValueStoreConfiguration.createInConfiguration(parent, 0);
        isolated.set("foo-key", "foo-value");
        assertEquals("foo-value", isolated.get("foo-key"));

        // Check that this value is stored in the namespace on the parent:
        Configuration delegate = isolated.getDelegate();
        assertEquals("foo-value", delegate.get(KeyValueStoreConfiguration.confKeyAtIndex("foo-key", 0)));
    }

    @Test
    public void testStoreBoolean() {
        final boolean expected = true;
        Configuration parent = new Configuration(false);
        KeyValueStoreConfiguration isolated = KeyValueStoreConfiguration.createInConfiguration(parent, 0);
        isolated.setBoolean("foo-key", expected);
        assertEquals(expected, isolated.getBoolean("foo-key", !expected));

        // Check that this value is stored in the namespace on the parent:
        Configuration delegate = isolated.getDelegate();
        assertEquals(expected,
                delegate.getBoolean(KeyValueStoreConfiguration.confKeyAtIndex("foo-key", 0), !expected));
    }

    @Test
    public void testStoreClass() {
        Configuration parent = new Configuration(false);
        KeyValueStoreConfiguration isolated = KeyValueStoreConfiguration.createInConfiguration(parent, 0);
        isolated.setClass("foo-key", String.class, Object.class);
        assertEquals(String.class, isolated.getClass("foo-key", null));
        assertEquals(String.class, isolated.getClass("foo-key", null, Object.class));

        // Check that this value is stored in the namespace on the parent:
        Configuration delegate = isolated.getDelegate();
        assertEquals(String.class,
                delegate.getClass(KeyValueStoreConfiguration.confKeyAtIndex("foo-key", 0), null));
    }

    @Test
    public void testStoreFloat() {
        Configuration parent = new Configuration(false);
        KeyValueStoreConfiguration isolated = KeyValueStoreConfiguration.createInConfiguration(parent, 0);
        isolated.setFloat("foo-key", 3.14F);
        assertEquals(3.14F, isolated.getFloat("foo-key", 0.0F), 0.0F);

        // Check that this value is stored in the namespace on the parent:
        Configuration delegate = isolated.getDelegate();
        assertEquals(3.14F, delegate.getFloat(KeyValueStoreConfiguration.confKeyAtIndex("foo-key", 0), 0.0F), 0.0F);
    }

    @Test
    public void testStoreInt() {
        Configuration parent = new Configuration(false);
        KeyValueStoreConfiguration isolated = KeyValueStoreConfiguration.createInConfiguration(parent, 0);
        isolated.setInt("foo-key", 123);
        assertEquals(123, isolated.getInt("foo-key", 0));

        // Check that this value is stored in the namespace on the parent:
        Configuration delegate = isolated.getDelegate();
        assertEquals(123, delegate.getInt(KeyValueStoreConfiguration.confKeyAtIndex("foo-key", 0), 0));
    }

    @Test
    public void testStoreLong() {
        Configuration parent = new Configuration(false);
        KeyValueStoreConfiguration isolated = KeyValueStoreConfiguration.createInConfiguration(parent, 0);
        isolated.setLong("foo-key", 12345L);
        assertEquals(12345L, isolated.getLong("foo-key", 0L));

        // Check that this value is stored in the namespace on the parent:
        Configuration delegate = isolated.getDelegate();
        assertEquals(12345L, delegate.getLong(KeyValueStoreConfiguration.confKeyAtIndex("foo-key", 0), 0L));
    }

    @Test
    public void testStoreStringArray() {
        Configuration parent = new Configuration(false);
        KeyValueStoreConfiguration isolated = KeyValueStoreConfiguration.createInConfiguration(parent, 0);
        isolated.setStrings("foo-key", "first", "second", "third");
        assertTrue(Arrays.equals(new String[] { "first", "second", "third" }, isolated.getStrings("foo-key")));
        // Also test the specifying a defaultValue works.
        assertTrue(Arrays.equals(new String[] { "first", "second", "third" },
                isolated.getStrings("foo-key", new String[0])));

        // Check that this value is stored in the namespace on the parent:
        Configuration delegate = isolated.getDelegate();
        assertTrue(Arrays.equals(new String[] { "first", "second", "third" },
                delegate.getStrings(KeyValueStoreConfiguration.confKeyAtIndex("foo-key", 0))));
    }
}