DelimitedLineImporterTest.java

/*
 * Copyright (c) 2013-2017 Cinchapi 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.cinchapi.concourse.importer;

import java.util.Set;

import org.junit.Assert;
import org.junit.Test;

import com.cinchapi.common.reflect.Reflection;
import com.cinchapi.concourse.importer.LineBasedImporter;
import com.cinchapi.concourse.test.ConcourseIntegrationTest;
import com.cinchapi.concourse.util.Resources;
import com.google.common.collect.Lists;

/**
 * Base unit test that validates the integrity of importing delimited line based
 * data into Concourse.
 * 
 * @author Jeff Nelson
 */
@SuppressWarnings("deprecation")
public abstract class DelimitedLineImporterTest extends
        ConcourseIntegrationTest {

    /**
     * The importer. A new instance of this is instantiated before each test.
     */
    protected DelimitedLineImporter importer;

    /**
     * A {@link LineBasedImporter legacy} importer that is used for comparisons.
     */
    private LineBasedImporter legacy;

    @Override
    public void beforeEachTest() {
        importer = Reflection.newInstance(getImporterClass(), client);
        legacy = getLegacyImporter();
    }

    @Test
    public void testImport() {
        String file = Resources.get("/" + getImportPath()).getFile();
        Set<Long> records = importer.importFile(file);
        Set<Long> legacyRecords = legacy.importFile(file);
        Assert.assertEquals(
                Lists.newArrayList(client.select(legacyRecords).values()),
                Lists.newArrayList(client.select(records).values()));
    }

    @Test(expected = IllegalStateException.class)
    public void testCannotSetHeaderAfterImport() {
        String file = Resources.get("/" + getImportPath()).getFile();
        importer.importFile(file);
        importer.parseHeader("a,b,c,d,e");
    }

    /**
     * Return the {@link Class} object for the {@link Importer} type used in the
     * test.
     * 
     * @return the Importer class
     */
    protected abstract Class<? extends DelimitedLineImporter> getImporterClass();

    /**
     * Return a {@link LineBasedImporter legacy} importer to use for validation
     * of the new importer.
     * 
     * @return the legacy Importer
     */
    protected abstract LineBasedImporter getLegacyImporter();

    /**
     * Return the path of the file/directory to import.
     * 
     * @return the import path
     */
    protected abstract String getImportPath();

}