/*
* Copyright 2014 Objectos, Fábrica de Software LTDA.
*
* Licensed under the Apache License, Version 2.0 (the "License"); you may not
* use this file except in compliance with the License. You may obtain a copy of
* the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
* License for the specific language governing permissions and limitations under
* the License.
*/
package br.com.objectos.way.gdrive;
import static br.com.objectos.way.base.testing.WayMatchers.is;
import static br.com.objectos.way.base.testing.WayMatchers.sameInstance;
import static com.google.common.collect.Lists.newArrayList;
import static org.hamcrest.MatcherAssert.assertThat;
import java.util.Iterator;
import java.util.List;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;
import com.google.api.services.drive.model.Change;
import com.google.common.collect.ImmutableList;
/**
* @author mario.marques@objectos.com.br (Mario Marques Junior)
*/
@Test
public class RequestIteratorTest {
private List<List<Change>> changes;
@BeforeMethod
public void setup() {
changes = newArrayList();
}
public void empty_request_iterator_must_work() {
Iterator<Change> iterator = getIterator();
assertThat(iterator.hasNext(), is(false));
}
public void one_element_must_work() {
Change element0 = new Change();
addAll(element0);
Iterator<Change> iterator = getIterator();
assertThat(iterator.hasNext(), is(true));
assertThat(iterator.next(), sameInstance(element0));
assertThat(iterator.hasNext(), is(false));
}
public void more_elements_must_work() {
Change element0 = new Change();
Change element1 = new Change();
Change element2 = new Change();
addAll(element0, element1, element2);
Iterator<Change> iterator = getIterator();
assertThat(iterator.hasNext(), is(true));
assertThat(iterator.next(), sameInstance(element0));
assertThat(iterator.hasNext(), is(true));
assertThat(iterator.next(), sameInstance(element1));
assertThat(iterator.hasNext(), is(true));
assertThat(iterator.next(), sameInstance(element2));
assertThat(iterator.hasNext(), is(false));
}
public void more_pages_must_work() {
Change element0 = new Change();
Change element1 = new Change();
Change element2 = new Change();
Change element3 = new Change();
addAll(element0, element1);
addAll(element2, element3);
Iterator<Change> iterator = getIterator();
assertThat(iterator.hasNext(), is(true));
assertThat(iterator.next(), sameInstance(element0));
assertThat(iterator.hasNext(), is(true));
assertThat(iterator.next(), sameInstance(element1));
assertThat(iterator.hasNext(), is(true));
assertThat(iterator.next(), sameInstance(element2));
assertThat(iterator.hasNext(), is(true));
assertThat(iterator.next(), sameInstance(element3));
assertThat(iterator.hasNext(), is(false));
}
private Iterator<Change> getIterator() {
Request<Change> request = new FakeRequest();
return new RequestIterator<Change>(request);
}
private void addAll(Change... args) {
List<Change> list = ImmutableList.copyOf(args);
changes.add(list);
}
private class FakeRequest implements Request<Change> {
private int counter;
@Override
public List<Change> execute() {
return changes.isEmpty() ? ImmutableList.<Change> of() : changes.get(counter++);
}
@Override
public boolean hasNextPage() {
return counter < changes.size();
}
}
}