TestMysqlDumpParser.java

package org.wikibrain.parser;

import com.akiban.sql.StandardException;
import org.apache.commons.io.FileUtils;
import org.junit.Test;
import org.wikibrain.parser.sql.MySqlDumpParser;

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

import static org.junit.Assert.assertEquals;

/**
 */
public class TestMysqlDumpParser {
    public static final File LINK_DUMP = new File("src/test/resources/org/wikibrain/parser/pagelinks.sql");

    @Test
    public void test() {
        MySqlDumpParser parser = new MySqlDumpParser();
        int i = 0;
        for (Object [] column : parser.parse(LINK_DUMP)) {
            assertEquals(column.length, 3);
            assert(column[0] instanceof Integer);
            assert(column[1] instanceof Integer);
            assert(column[2] instanceof String);
            i++;
        }
        assertEquals(i, 1003);
    }

    @Test
    public void testOne() throws IOException, StandardException {
        List<String> lines = FileUtils.readLines(LINK_DUMP, "UTF-8");
        String lastLine = lines.get(lines.size() - 1);
        MySqlDumpParser parser = new MySqlDumpParser();
        List<Object[]> rows = parser.parse(lastLine);
        assertEquals(rows.size(), 437);
    }
}