63 lines
1.5 KiB
Java
63 lines
1.5 KiB
Java
package eu.dnetlib.enabling.resultset;
|
|
|
|
import static org.junit.Assert.*;
|
|
|
|
import java.util.List;
|
|
|
|
import org.junit.Before;
|
|
import org.junit.Test;
|
|
|
|
import com.google.common.collect.Lists;
|
|
|
|
public class StreamingResultSetListenerTest {
|
|
|
|
private StreamingResultSetListener listener;
|
|
private List<String> data;
|
|
|
|
@Before
|
|
public void setUp() throws Exception {
|
|
data = Lists.newArrayList("uno", "due", "tre", "quattro");
|
|
listener = new StreamingResultSetListener(data.iterator(), data.size());
|
|
}
|
|
|
|
@Test
|
|
public void testGetResult() {
|
|
assertEquals(data.size(), listener.getSize());
|
|
|
|
final List<String> res = listener.getResult(1, 2);
|
|
assertEquals(2, res.size());
|
|
assertEquals("uno", res.get(0));
|
|
assertEquals("due", res.get(1));
|
|
}
|
|
|
|
@Test
|
|
public void testGetResult2() {
|
|
listener.getResult(1, 2);
|
|
final List<String> res = listener.getResult(3, 4);
|
|
assertEquals(2, res.size());
|
|
assertEquals("tre", res.get(0));
|
|
assertEquals("quattro", res.get(1));
|
|
}
|
|
|
|
@Test
|
|
public void testGetResult_retry() {
|
|
listener.getResult(1, 2);
|
|
final List<String> res = listener.getResult(1, 2);
|
|
assertEquals(2, res.size());
|
|
assertEquals("uno", res.get(0));
|
|
assertEquals("due", res.get(1));
|
|
}
|
|
|
|
@Test(expected = IllegalArgumentException.class)
|
|
public void testGetResult_random() {
|
|
listener.getResult(1, 2);
|
|
listener.getResult(2, 3);
|
|
}
|
|
|
|
@Test(expected = IllegalArgumentException.class)
|
|
public void testGetResult_skip() {
|
|
listener.getResult(1, 2);
|
|
listener.getResult(4, 4);
|
|
}
|
|
}
|