/** * Copyright 2014 Netflix, Inc. * * 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 rx.internal.operators; import static org.mockito.Matchers.any; import static org.mockito.Mockito.inOrder; import static org.mockito.Mockito.mock; import static org.mockito.Mockito.never; import static org.mockito.Mockito.times; import static org.mockito.Mockito.verify; import java.util.concurrent.TimeUnit; import org.junit.Test; import org.mockito.InOrder; import rx.Observable; import rx.Observer; import rx.exceptions.TestException; import rx.schedulers.TestScheduler; import rx.subjects.PublishSubject; public class OperatorTakeLastTimedTest { @Test(expected = IndexOutOfBoundsException.class) public void testTakeLastTimedWithNegativeCount() { Observable.just("one").takeLast(-1, 1, TimeUnit.SECONDS); } @Test public void takeLastTimed() { TestScheduler scheduler = new TestScheduler(); PublishSubject<Object> source = PublishSubject.create(); Observable<Object> result = source.takeLast(1, TimeUnit.SECONDS, scheduler); @SuppressWarnings("unchecked") Observer<Object> o = mock(Observer.class); InOrder inOrder = inOrder(o); result.subscribe(o); source.onNext(1); // T: 0ms scheduler.advanceTimeBy(250, TimeUnit.MILLISECONDS); source.onNext(2); // T: 250ms scheduler.advanceTimeBy(250, TimeUnit.MILLISECONDS); source.onNext(3); // T: 500ms scheduler.advanceTimeBy(250, TimeUnit.MILLISECONDS); source.onNext(4); // T: 750ms scheduler.advanceTimeBy(250, TimeUnit.MILLISECONDS); source.onNext(5); // T: 1000ms scheduler.advanceTimeBy(250, TimeUnit.MILLISECONDS); source.onCompleted(); // T: 1250ms inOrder.verify(o, times(1)).onNext(2); inOrder.verify(o, times(1)).onNext(3); inOrder.verify(o, times(1)).onNext(4); inOrder.verify(o, times(1)).onNext(5); inOrder.verify(o, times(1)).onCompleted(); verify(o, never()).onError(any(Throwable.class)); } @Test public void takeLastTimedDelayCompletion() { TestScheduler scheduler = new TestScheduler(); PublishSubject<Object> source = PublishSubject.create(); Observable<Object> result = source.takeLast(1, TimeUnit.SECONDS, scheduler); @SuppressWarnings("unchecked") Observer<Object> o = mock(Observer.class); InOrder inOrder = inOrder(o); result.subscribe(o); source.onNext(1); // T: 0ms scheduler.advanceTimeBy(250, TimeUnit.MILLISECONDS); source.onNext(2); // T: 250ms scheduler.advanceTimeBy(250, TimeUnit.MILLISECONDS); source.onNext(3); // T: 500ms scheduler.advanceTimeBy(250, TimeUnit.MILLISECONDS); source.onNext(4); // T: 750ms scheduler.advanceTimeBy(250, TimeUnit.MILLISECONDS); source.onNext(5); // T: 1000ms scheduler.advanceTimeBy(1250, TimeUnit.MILLISECONDS); source.onCompleted(); // T: 2250ms inOrder.verify(o, times(1)).onCompleted(); verify(o, never()).onNext(any()); verify(o, never()).onError(any(Throwable.class)); } @Test public void takeLastTimedWithCapacity() { TestScheduler scheduler = new TestScheduler(); PublishSubject<Object> source = PublishSubject.create(); Observable<Object> result = source.takeLast(2, 1, TimeUnit.SECONDS, scheduler); @SuppressWarnings("unchecked") Observer<Object> o = mock(Observer.class); InOrder inOrder = inOrder(o); result.subscribe(o); source.onNext(1); // T: 0ms scheduler.advanceTimeBy(250, TimeUnit.MILLISECONDS); source.onNext(2); // T: 250ms scheduler.advanceTimeBy(250, TimeUnit.MILLISECONDS); source.onNext(3); // T: 500ms scheduler.advanceTimeBy(250, TimeUnit.MILLISECONDS); source.onNext(4); // T: 750ms scheduler.advanceTimeBy(250, TimeUnit.MILLISECONDS); source.onNext(5); // T: 1000ms scheduler.advanceTimeBy(250, TimeUnit.MILLISECONDS); source.onCompleted(); // T: 1250ms inOrder.verify(o, times(1)).onNext(4); inOrder.verify(o, times(1)).onNext(5); inOrder.verify(o, times(1)).onCompleted(); verify(o, never()).onError(any(Throwable.class)); } @Test public void takeLastTimedThrowingSource() { TestScheduler scheduler = new TestScheduler(); PublishSubject<Object> source = PublishSubject.create(); Observable<Object> result = source.takeLast(1, TimeUnit.SECONDS, scheduler); @SuppressWarnings("unchecked") Observer<Object> o = mock(Observer.class); InOrder inOrder = inOrder(o); result.subscribe(o); source.onNext(1); // T: 0ms scheduler.advanceTimeBy(250, TimeUnit.MILLISECONDS); source.onNext(2); // T: 250ms scheduler.advanceTimeBy(250, TimeUnit.MILLISECONDS); source.onNext(3); // T: 500ms scheduler.advanceTimeBy(250, TimeUnit.MILLISECONDS); source.onNext(4); // T: 750ms scheduler.advanceTimeBy(250, TimeUnit.MILLISECONDS); source.onNext(5); // T: 1000ms scheduler.advanceTimeBy(250, TimeUnit.MILLISECONDS); source.onError(new TestException()); // T: 1250ms inOrder.verify(o, times(1)).onError(any(TestException.class)); verify(o, never()).onNext(any()); verify(o, never()).onCompleted(); } @Test public void takeLastTimedWithZeroCapacity() { TestScheduler scheduler = new TestScheduler(); PublishSubject<Object> source = PublishSubject.create(); Observable<Object> result = source.takeLast(0, 1, TimeUnit.SECONDS, scheduler); @SuppressWarnings("unchecked") Observer<Object> o = mock(Observer.class); InOrder inOrder = inOrder(o); result.subscribe(o); source.onNext(1); // T: 0ms scheduler.advanceTimeBy(250, TimeUnit.MILLISECONDS); source.onNext(2); // T: 250ms scheduler.advanceTimeBy(250, TimeUnit.MILLISECONDS); source.onNext(3); // T: 500ms scheduler.advanceTimeBy(250, TimeUnit.MILLISECONDS); source.onNext(4); // T: 750ms scheduler.advanceTimeBy(250, TimeUnit.MILLISECONDS); source.onNext(5); // T: 1000ms scheduler.advanceTimeBy(250, TimeUnit.MILLISECONDS); source.onCompleted(); // T: 1250ms inOrder.verify(o, times(1)).onCompleted(); verify(o, never()).onNext(any()); verify(o, never()).onError(any(Throwable.class)); } }