InMemoryCachingEntityCheckerManagerImplTest.java

/**
 * This file is part of General Entity Annotator Benchmark.
 *
 * General Entity Annotator Benchmark is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * General Entity Annotator Benchmark is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with General Entity Annotator Benchmark.  If not, see <http://www.gnu.org/licenses/>.
 */
package org.aksw.gerbil.dataset.check;

import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.HashSet;
import java.util.List;

import org.aksw.gerbil.dataset.check.impl.InMemoryCachingEntityCheckerManager;
import org.aksw.gerbil.transfer.nif.data.Annotation;
import org.junit.Assert;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.Parameterized;
import org.junit.runners.Parameterized.Parameters;

@RunWith(Parameterized.class)
public class InMemoryCachingEntityCheckerManagerImplTest implements EntityChecker {

    private static final String CORRECT_URI = "http://aksw.org/correct";
    private static final String WRONG_URI = "http://aksw.org/wrong";
    private static final long CACHE_DURATION = 2000;

    @Parameters
    public static Collection<Object[]> data() {
        List<Object[]> testConfigs = new ArrayList<Object[]>();
        testConfigs.add(new Object[] { new String[] { CORRECT_URI }, new boolean[] { true } });
        testConfigs.add(new Object[] { new String[] { WRONG_URI }, new boolean[] { false } });
        testConfigs.add(new Object[] { new String[] { CORRECT_URI, WRONG_URI }, new boolean[] { true, false } });
        return testConfigs;
    }

    private String uris[];
    private boolean expectingSameUri[];
    private boolean expectCall;

    public InMemoryCachingEntityCheckerManagerImplTest(String uris[], boolean expectingSameUri[]) {
        this.uris = uris;
        this.expectingSameUri = expectingSameUri;
    }

    @Test
    public void test() throws IOException, InterruptedException {
        InMemoryCachingEntityCheckerManager manager = new InMemoryCachingEntityCheckerManager(10000, CACHE_DURATION);
        manager.registerEntityChecker("http://aksw.org/", this);

        expectCall = true;
        runSingleTest(manager);

        expectCall = false;
        runSingleTest(manager);

        Thread.sleep(CACHE_DURATION);

        expectCall = true;
        runSingleTest(manager);
    }

    private void runSingleTest(EntityCheckerManager manager) {
        Annotation annotation = new Annotation(new HashSet<String>(Arrays.asList(uris)));
        manager.checkMeanings(Arrays.asList(annotation));
        for (int i = 0; i < uris.length; ++i) {
            if (expectingSameUri[i]) {
                Assert.assertTrue("Expected the URI to be there after the check.", annotation.containsUri(uris[i]));
            } else {
                Assert.assertFalse("Expected the URI to be changed during the check.", annotation.containsUri(uris[i]));
            }
        }
    }

    @Override
    public boolean entityExists(String uri) {
        Assert.assertTrue("Didn't expected to be called.", expectCall);
        return uri.equals(CORRECT_URI);
    }
}