/** * Copyright (c) 2016-present, RxJava Contributors. * * 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 io.reactivex.internal.operators.flowable; import static org.mockito.Mockito.*; import org.junit.*; import org.mockito.InOrder; import org.reactivestreams.Subscriber; import java.util.NoSuchElementException; import io.reactivex.*; import io.reactivex.functions.Predicate; public class FlowableFirstTest { Subscriber<String> w; SingleObserver<Object> wo; MaybeObserver<Object> wm; private static final Predicate<String> IS_D = new Predicate<String>() { @Override public boolean test(String value) { return "d".equals(value); } }; @Before public void before() { w = TestHelper.mockSubscriber(); wo = TestHelper.mockSingleObserver(); wm = TestHelper.mockMaybeObserver(); } @Test public void testFirstOrElseOfNoneFlowable() { Flowable<String> src = Flowable.empty(); src.first("default").toFlowable().subscribe(w); verify(w, times(1)).onNext(anyString()); verify(w, times(1)).onNext("default"); verify(w, never()).onError(any(Throwable.class)); verify(w, times(1)).onComplete(); } @Test public void testFirstOrElseOfSomeFlowable() { Flowable<String> src = Flowable.just("a", "b", "c"); src.first("default").toFlowable().subscribe(w); verify(w, times(1)).onNext(anyString()); verify(w, times(1)).onNext("a"); verify(w, never()).onError(any(Throwable.class)); verify(w, times(1)).onComplete(); } @Test public void testFirstOrElseWithPredicateOfNoneMatchingThePredicateFlowable() { Flowable<String> src = Flowable.just("a", "b", "c"); src.filter(IS_D).first("default").toFlowable().subscribe(w); verify(w, times(1)).onNext(anyString()); verify(w, times(1)).onNext("default"); verify(w, never()).onError(any(Throwable.class)); verify(w, times(1)).onComplete(); } @Test public void testFirstOrElseWithPredicateOfSomeFlowable() { Flowable<String> src = Flowable.just("a", "b", "c", "d", "e", "f"); src.filter(IS_D).first("default").toFlowable().subscribe(w); verify(w, times(1)).onNext(anyString()); verify(w, times(1)).onNext("d"); verify(w, never()).onError(any(Throwable.class)); verify(w, times(1)).onComplete(); } @Test public void testFirstFlowable() { Flowable<Integer> observable = Flowable.just(1, 2, 3).firstElement().toFlowable(); Subscriber<Integer> observer = TestHelper.mockSubscriber(); observable.subscribe(observer); InOrder inOrder = inOrder(observer); inOrder.verify(observer, times(1)).onNext(1); inOrder.verify(observer, times(1)).onComplete(); inOrder.verifyNoMoreInteractions(); } @Test public void testFirstWithOneElementFlowable() { Flowable<Integer> observable = Flowable.just(1).firstElement().toFlowable(); Subscriber<Integer> observer = TestHelper.mockSubscriber(); observable.subscribe(observer); InOrder inOrder = inOrder(observer); inOrder.verify(observer, times(1)).onNext(1); inOrder.verify(observer, times(1)).onComplete(); inOrder.verifyNoMoreInteractions(); } @Test public void testFirstWithEmptyFlowable() { Flowable<Integer> observable = Flowable.<Integer> empty().firstElement().toFlowable(); Subscriber<Integer> observer = TestHelper.mockSubscriber(); observable.subscribe(observer); InOrder inOrder = inOrder(observer); inOrder.verify(observer).onComplete(); inOrder.verify(observer, never()).onError(any(Throwable.class)); inOrder.verifyNoMoreInteractions(); } @Test public void testFirstWithPredicateFlowable() { Flowable<Integer> observable = Flowable.just(1, 2, 3, 4, 5, 6) .filter(new Predicate<Integer>() { @Override public boolean test(Integer t1) { return t1 % 2 == 0; } }) .firstElement().toFlowable(); Subscriber<Integer> observer = TestHelper.mockSubscriber(); observable.subscribe(observer); InOrder inOrder = inOrder(observer); inOrder.verify(observer, times(1)).onNext(2); inOrder.verify(observer, times(1)).onComplete(); inOrder.verifyNoMoreInteractions(); } @Test public void testFirstWithPredicateAndOneElementFlowable() { Flowable<Integer> observable = Flowable.just(1, 2) .filter(new Predicate<Integer>() { @Override public boolean test(Integer t1) { return t1 % 2 == 0; } }) .firstElement().toFlowable(); Subscriber<Integer> observer = TestHelper.mockSubscriber(); observable.subscribe(observer); InOrder inOrder = inOrder(observer); inOrder.verify(observer, times(1)).onNext(2); inOrder.verify(observer, times(1)).onComplete(); inOrder.verifyNoMoreInteractions(); } @Test public void testFirstWithPredicateAndEmptyFlowable() { Flowable<Integer> observable = Flowable.just(1) .filter(new Predicate<Integer>() { @Override public boolean test(Integer t1) { return t1 % 2 == 0; } }) .firstElement().toFlowable(); Subscriber<Integer> observer = TestHelper.mockSubscriber(); observable.subscribe(observer); InOrder inOrder = inOrder(observer); inOrder.verify(observer).onComplete(); inOrder.verify(observer, never()).onError(any(Throwable.class)); inOrder.verifyNoMoreInteractions(); } @Test public void testFirstOrDefaultFlowable() { Flowable<Integer> observable = Flowable.just(1, 2, 3) .first(4).toFlowable(); Subscriber<Integer> observer = TestHelper.mockSubscriber(); observable.subscribe(observer); InOrder inOrder = inOrder(observer); inOrder.verify(observer, times(1)).onNext(1); inOrder.verify(observer, times(1)).onComplete(); inOrder.verifyNoMoreInteractions(); } @Test public void testFirstOrDefaultWithOneElementFlowable() { Flowable<Integer> observable = Flowable.just(1).first(2).toFlowable(); Subscriber<Integer> observer = TestHelper.mockSubscriber(); observable.subscribe(observer); InOrder inOrder = inOrder(observer); inOrder.verify(observer, times(1)).onNext(1); inOrder.verify(observer, times(1)).onComplete(); inOrder.verifyNoMoreInteractions(); } @Test public void testFirstOrDefaultWithEmptyFlowable() { Flowable<Integer> observable = Flowable.<Integer> empty() .first(1).toFlowable(); Subscriber<Integer> observer = TestHelper.mockSubscriber(); observable.subscribe(observer); InOrder inOrder = inOrder(observer); inOrder.verify(observer, times(1)).onNext(1); inOrder.verify(observer, times(1)).onComplete(); inOrder.verifyNoMoreInteractions(); } @Test public void testFirstOrDefaultWithPredicateFlowable() { Flowable<Integer> observable = Flowable.just(1, 2, 3, 4, 5, 6) .filter(new Predicate<Integer>() { @Override public boolean test(Integer t1) { return t1 % 2 == 0; } }) .first(8).toFlowable(); Subscriber<Integer> observer = TestHelper.mockSubscriber(); observable.subscribe(observer); InOrder inOrder = inOrder(observer); inOrder.verify(observer, times(1)).onNext(2); inOrder.verify(observer, times(1)).onComplete(); inOrder.verifyNoMoreInteractions(); } @Test public void testFirstOrDefaultWithPredicateAndOneElementFlowable() { Flowable<Integer> observable = Flowable.just(1, 2) .filter(new Predicate<Integer>() { @Override public boolean test(Integer t1) { return t1 % 2 == 0; } }) .first(4).toFlowable(); Subscriber<Integer> observer = TestHelper.mockSubscriber(); observable.subscribe(observer); InOrder inOrder = inOrder(observer); inOrder.verify(observer, times(1)).onNext(2); inOrder.verify(observer, times(1)).onComplete(); inOrder.verifyNoMoreInteractions(); } @Test public void testFirstOrDefaultWithPredicateAndEmptyFlowable() { Flowable<Integer> observable = Flowable.just(1) .filter(new Predicate<Integer>() { @Override public boolean test(Integer t1) { return t1 % 2 == 0; } }) .first(2).toFlowable(); Subscriber<Integer> observer = TestHelper.mockSubscriber(); observable.subscribe(observer); InOrder inOrder = inOrder(observer); inOrder.verify(observer, times(1)).onNext(2); inOrder.verify(observer, times(1)).onComplete(); inOrder.verifyNoMoreInteractions(); } @Test public void testFirstOrElseOfNone() { Flowable<String> src = Flowable.empty(); src.first("default").subscribe(wo); verify(wo, times(1)).onSuccess(anyString()); verify(wo, times(1)).onSuccess("default"); verify(wo, never()).onError(any(Throwable.class)); } @Test public void testFirstOrElseOfSome() { Flowable<String> src = Flowable.just("a", "b", "c"); src.first("default").subscribe(wo); verify(wo, times(1)).onSuccess(anyString()); verify(wo, times(1)).onSuccess("a"); verify(wo, never()).onError(any(Throwable.class)); } @Test public void testFirstOrElseWithPredicateOfNoneMatchingThePredicate() { Flowable<String> src = Flowable.just("a", "b", "c"); src.filter(IS_D).first("default").subscribe(wo); verify(wo, times(1)).onSuccess(anyString()); verify(wo, times(1)).onSuccess("default"); verify(wo, never()).onError(any(Throwable.class)); } @Test public void testFirstOrElseWithPredicateOfSome() { Flowable<String> src = Flowable.just("a", "b", "c", "d", "e", "f"); src.filter(IS_D).first("default").subscribe(wo); verify(wo, times(1)).onSuccess(anyString()); verify(wo, times(1)).onSuccess("d"); verify(wo, never()).onError(any(Throwable.class)); } @Test public void testFirst() { Maybe<Integer> observable = Flowable.just(1, 2, 3).firstElement(); observable.subscribe(wm); InOrder inOrder = inOrder(wm); inOrder.verify(wm, times(1)).onSuccess(1); inOrder.verifyNoMoreInteractions(); } @Test public void testFirstWithOneElement() { Maybe<Integer> observable = Flowable.just(1).firstElement(); observable.subscribe(wm); InOrder inOrder = inOrder(wm); inOrder.verify(wm, times(1)).onSuccess(1); inOrder.verifyNoMoreInteractions(); } @Test public void testFirstWithEmpty() { Maybe<Integer> observable = Flowable.<Integer> empty().firstElement(); observable.subscribe(wm); InOrder inOrder = inOrder(wm); inOrder.verify(wm).onComplete(); inOrder.verify(wm, never()).onError(any(Throwable.class)); inOrder.verifyNoMoreInteractions(); } @Test public void testFirstWithPredicate() { Maybe<Integer> observable = Flowable.just(1, 2, 3, 4, 5, 6) .filter(new Predicate<Integer>() { @Override public boolean test(Integer t1) { return t1 % 2 == 0; } }) .firstElement(); observable.subscribe(wm); InOrder inOrder = inOrder(wm); inOrder.verify(wm, times(1)).onSuccess(2); inOrder.verifyNoMoreInteractions(); } @Test public void testFirstWithPredicateAndOneElement() { Maybe<Integer> observable = Flowable.just(1, 2) .filter(new Predicate<Integer>() { @Override public boolean test(Integer t1) { return t1 % 2 == 0; } }) .firstElement(); observable.subscribe(wm); InOrder inOrder = inOrder(wm); inOrder.verify(wm, times(1)).onSuccess(2); inOrder.verifyNoMoreInteractions(); } @Test public void testFirstWithPredicateAndEmpty() { Maybe<Integer> observable = Flowable.just(1) .filter(new Predicate<Integer>() { @Override public boolean test(Integer t1) { return t1 % 2 == 0; } }) .firstElement(); observable.subscribe(wm); InOrder inOrder = inOrder(wm); inOrder.verify(wm).onComplete(); inOrder.verify(wm, never()).onError(any(Throwable.class)); inOrder.verifyNoMoreInteractions(); } @Test public void testFirstOrDefault() { Single<Integer> observable = Flowable.just(1, 2, 3) .first(4); observable.subscribe(wo); InOrder inOrder = inOrder(wo); inOrder.verify(wo, times(1)).onSuccess(1); inOrder.verifyNoMoreInteractions(); } @Test public void testFirstOrDefaultWithOneElement() { Single<Integer> observable = Flowable.just(1).first(2); observable.subscribe(wo); InOrder inOrder = inOrder(wo); inOrder.verify(wo, times(1)).onSuccess(1); inOrder.verifyNoMoreInteractions(); } @Test public void testFirstOrDefaultWithEmpty() { Single<Integer> observable = Flowable.<Integer> empty() .first(1); observable.subscribe(wo); InOrder inOrder = inOrder(wo); inOrder.verify(wo, times(1)).onSuccess(1); inOrder.verifyNoMoreInteractions(); } @Test public void testFirstOrDefaultWithPredicate() { Single<Integer> observable = Flowable.just(1, 2, 3, 4, 5, 6) .filter(new Predicate<Integer>() { @Override public boolean test(Integer t1) { return t1 % 2 == 0; } }) .first(8); observable.subscribe(wo); InOrder inOrder = inOrder(wo); inOrder.verify(wo, times(1)).onSuccess(2); inOrder.verifyNoMoreInteractions(); } @Test public void testFirstOrDefaultWithPredicateAndOneElement() { Single<Integer> observable = Flowable.just(1, 2) .filter(new Predicate<Integer>() { @Override public boolean test(Integer t1) { return t1 % 2 == 0; } }) .first(4); observable.subscribe(wo); InOrder inOrder = inOrder(wo); inOrder.verify(wo, times(1)).onSuccess(2); inOrder.verifyNoMoreInteractions(); } @Test public void testFirstOrDefaultWithPredicateAndEmpty() { Single<Integer> observable = Flowable.just(1) .filter(new Predicate<Integer>() { @Override public boolean test(Integer t1) { return t1 % 2 == 0; } }) .first(2); observable.subscribe(wo); InOrder inOrder = inOrder(wo); inOrder.verify(wo, times(1)).onSuccess(2); inOrder.verifyNoMoreInteractions(); } @Test public void firstOrErrorNoElement() { Flowable.empty() .firstOrError() .test() .assertNoValues() .assertError(NoSuchElementException.class); } @Test public void firstOrErrorOneElement() { Flowable.just(1) .firstOrError() .test() .assertNoErrors() .assertValue(1); } @Test public void firstOrErrorMultipleElements() { Flowable.just(1, 2, 3) .firstOrError() .test() .assertNoErrors() .assertValue(1); } @Test public void firstOrErrorError() { Flowable.error(new RuntimeException("error")) .firstOrError() .test() .assertNoValues() .assertErrorMessage("error") .assertError(RuntimeException.class); } @Test public void firstOrErrorNoElementFlowable() { Flowable.empty() .firstOrError() .toFlowable() .test() .assertNoValues() .assertError(NoSuchElementException.class); } @Test public void firstOrErrorOneElementFlowable() { Flowable.just(1) .firstOrError() .toFlowable() .test() .assertNoErrors() .assertValue(1); } @Test public void firstOrErrorMultipleElementsFlowable() { Flowable.just(1, 2, 3) .firstOrError() .toFlowable() .test() .assertNoErrors() .assertValue(1); } @Test public void firstOrErrorErrorFlowable() { Flowable.error(new RuntimeException("error")) .firstOrError() .toFlowable() .test() .assertNoValues() .assertErrorMessage("error") .assertError(RuntimeException.class); } }