com.fullcontact.sstable.index.SSTableIndexIndexerTest.java Source code

Introduction

Here is the source code for com.fullcontact.sstable.index.SSTableIndexIndexerTest.java

Source

/*
 * Copyright 2014 FullContact, Inc.
 *
 * 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.fullcontact.sstable.index;

import com.google.common.io.Files;
import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.fs.FileSystem;
import org.apache.hadoop.fs.LocalFileSystem;
import org.apache.hadoop.fs.Path;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

import java.io.File;

import static junit.framework.Assert.assertEquals;
import static junit.framework.Assert.assertNotNull;

public class SSTableIndexIndexerTest {
    private static final String INDEX_FILE = "/data/Keyspace1-Standard1-ic-0-Index.db";
    private LocalFileSystem fileSystem;

    @Before
    public void setUp() throws Exception {
        fileSystem = FileSystem.getLocal(new Configuration());
    }

    @After
    public void tearDown() throws Exception {
        fileSystem.close();
    }

    @Test
    public void testIndex() throws Exception {

        // Create a temp dir...
        final File tmpDir = Files.createTempDir();
        tmpDir.deleteOnExit();

        // Copy the .Index.db file to the temp dir...
        File sampleFile = new File(getClass().getResource(INDEX_FILE).getFile());
        File tmpSampleFile = new File(tmpDir, sampleFile.getName());
        Files.copy(sampleFile, tmpSampleFile);

        fileSystem.getConf().set("hadoop.sstable.split.mb", "1");

        SSTableIndexIndexer indexer = new SSTableIndexIndexer(fileSystem.getConf());

        indexer.index(new Path(tmpDir.getAbsolutePath(), sampleFile.getName()));

        indexer.complete();

        SSTableIndexIndex index = SSTableIndexIndex.readIndex(fileSystem,
                new Path(tmpDir.getAbsolutePath(), sampleFile.getName()));
        assertNotNull(index);
        assertEquals(1, index.getOffsets().size());
        assertEquals(0, index.getOffsets().get(0).getStart());
        assertEquals(53, index.getOffsets().get(0).getEnd());
    }
}