DownloadIT.java

package org.wikibrain.integration;

import org.apache.commons.io.FileUtils;
import org.junit.Ignore;
import org.junit.Test;
import org.wikibrain.conf.ConfigurationException;
import org.wikibrain.core.WikiBrainException;
import org.wikibrain.core.lang.Language;
import org.wikibrain.core.lang.LanguageSet;
import org.wikibrain.download.RequestedLinkGetter;

import java.io.File;
import java.io.IOException;
import java.text.ParseException;
import java.util.List;

import static org.junit.Assert.*;

/**
 * @author Shilad Sen
 */
public class DownloadIT {

    @Ignore
    @Test
    public void testLinkGetter() throws ConfigurationException, WikiBrainException, ParseException, IOException {
        RequestedLinkGetter.main(TestUtils.getArgs());
        File f = new File("../integration-tests/download/list.tsv");
        assertTrue(f.isFile());

        List<String> lines = FileUtils.readLines(f);
        assertEquals(5, lines.size());

        for (Language lang : new LanguageSet("simple,la")) {
            String match = null;
            for (String line : lines) {
                if (line.startsWith(lang.getLangCode())) {
                    match = line;
                    break;
                }
            }
            assertNotNull(match);
            assertTrue(match.contains("http://dumps.wikimedia.org/" + lang.getLangCode() + "wiki/"));
        }
    }
}