package org.grouplens.lenskit.data.text;

import org.grouplens.lenskit.cursors.Cursors;
import org.grouplens.lenskit.data.event.Rating;
import org.grouplens.lenskit.data.event.Ratings;
import org.junit.Rule;
import org.junit.Test;
import org.junit.rules.TemporaryFolder;

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

import static org.hamcrest.Matchers.*;
import static org.junit.Assert.*;

public class TextEventDAOTest {
    @Rule
    public TemporaryFolder folder = new TemporaryFolder();

    private File createBasicCSV() throws IOException {
        File file = folder.newFile("ratings.csv");
        PrintWriter pw = new PrintWriter(file);
        try {
            pw.println("1,10,3.5");
            pw.println("1,4,2.5");
            pw.println("2,42,5");
        } finally {
            pw.close();
        }
        return file;
    }

    @Test
    public void testBasicCSV() throws IOException {
        File file = createBasicCSV();
        TextEventDAO dao = TextEventDAO.create(file, Formats.csvRatings());
        List<Rating> ratings = Cursors.makeList(dao.streamEvents(Rating.class));
        assertThat(ratings, containsInAnyOrder(Ratings.make(1, 10, 3.5),
                                               Ratings.make(1, 4, 2.5),
                                               Ratings.make(2, 42, 5)));
    }

    private File createHeaderCSV() throws IOException {
        File file = folder.newFile("ratings.csv");
        PrintWriter pw = new PrintWriter(file);
        try {
            pw.println("user,item,rating");
            pw.println("1,10,3.5");
            pw.println("1,4,2.5");
            pw.println("2,42,5");
        } finally {
            pw.close();
        }
        return file;
    }

    @Test
    public void testHeaderCSV() throws IOException {
        File file = createHeaderCSV();
        TextEventDAO dao = TextEventDAO.create(file, Formats.csvRatings().setHeaderLines(1));
        List<Rating> ratings = Cursors.makeList(dao.streamEvents(Rating.class));
        assertThat(ratings, containsInAnyOrder(Ratings.make(1, 10, 3.5),
                                               Ratings.make(1, 4, 2.5),
                                               Ratings.make(2, 42, 5)));
    }
}

@@ -1,8 +1,1001 @@
+/*%0A * LensKit, an open source recommender systems toolkit.%0A * Copyright 2010-2014 LensKit Contributors.  See CONTRIBUTORS.md.%0A * Work on LensKit has been funded by the National Science Foundation under%0A * grants IIS 05-34939, 08-08692, 08-12148, and 10-17697.%0A *%0A * This program is free software; you can redistribute it and/or modify%0A * it under the terms of the GNU Lesser General Public License as%0A * published by the Free Software Foundation; either version 2.1 of the%0A * License, or (at your option) any later version.%0A *%0A * This program is distributed in the hope that it will be useful, but WITHOUT%0A * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS%0A * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more%0A * details.%0A *%0A * You should have received a copy of the GNU General Public License along with%0A * this program; if not, write to the Free Software Foundation, Inc., 51%0A * Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.%0A */%0A
 package 

Add license header to event test