/**
* 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.observable;
import static org.junit.Assert.*;
import static org.mockito.Mockito.*;
import java.util.*;
import java.util.concurrent.*;
import java.util.concurrent.atomic.*;
import org.junit.Test;
import org.mockito.*;
import io.reactivex.*;
import io.reactivex.Observable;
import io.reactivex.Observer;
import io.reactivex.disposables.*;
import io.reactivex.exceptions.TestException;
import io.reactivex.functions.*;
import io.reactivex.internal.subscriptions.BooleanSubscription;
import io.reactivex.observables.GroupedObservable;
import io.reactivex.observers.*;
import io.reactivex.schedulers.Schedulers;
import io.reactivex.subjects.PublishSubject;
public class ObservableRetryTest {
@Test
public void iterativeBackoff() {
Observer<String> consumer = TestHelper.mockObserver();
Observable<String> producer = Observable.unsafeCreate(new ObservableSource<String>() {
private AtomicInteger count = new AtomicInteger(4);
long last = System.currentTimeMillis();
@Override
public void subscribe(Observer<? super String> t1) {
t1.onSubscribe(Disposables.empty());
System.out.println(count.get() + " @ " + String.valueOf(last - System.currentTimeMillis()));
last = System.currentTimeMillis();
if (count.getAndDecrement() == 0) {
t1.onNext("hello");
t1.onComplete();
} else {
t1.onError(new RuntimeException());
}
}
});
TestObserver<String> ts = new TestObserver<String>(consumer);
producer.retryWhen(new Function<Observable<? extends Throwable>, Observable<Object>>() {
@Override
public Observable<Object> apply(Observable<? extends Throwable> attempts) {
// Worker w = Schedulers.computation().createWorker();
return attempts
.map(new Function<Throwable, Tuple>() {
@Override
public Tuple apply(Throwable n) {
return new Tuple(new Long(1), n);
}})
.scan(new BiFunction<Tuple, Tuple, Tuple>() {
@Override
public Tuple apply(Tuple t, Tuple n) {
return new Tuple(t.count + n.count, n.n);
}})
.flatMap(new Function<Tuple, Observable<Long>>() {
@Override
public Observable<Long> apply(Tuple t) {
System.out.println("Retry # " + t.count);
return t.count > 20 ?
Observable.<Long>error(t.n) :
Observable.timer(t.count * 1L, TimeUnit.MILLISECONDS);
}}).cast(Object.class);
}
}).subscribe(ts);
ts.awaitTerminalEvent();
ts.assertNoErrors();
InOrder inOrder = inOrder(consumer);
inOrder.verify(consumer, never()).onError(any(Throwable.class));
inOrder.verify(consumer, times(1)).onNext("hello");
inOrder.verify(consumer, times(1)).onComplete();
inOrder.verifyNoMoreInteractions();
}
public static class Tuple {
Long count;
Throwable n;
Tuple(Long c, Throwable n) {
count = c;
this.n = n;
}
}
@Test
public void testRetryIndefinitely() {
Observer<String> observer = TestHelper.mockObserver();
int NUM_RETRIES = 20;
Observable<String> origin = Observable.unsafeCreate(new FuncWithErrors(NUM_RETRIES));
origin.retry().subscribe(new TestObserver<String>(observer));
InOrder inOrder = inOrder(observer);
// should show 3 attempts
inOrder.verify(observer, times(NUM_RETRIES + 1)).onNext("beginningEveryTime");
// should have no errors
inOrder.verify(observer, never()).onError(any(Throwable.class));
// should have a single success
inOrder.verify(observer, times(1)).onNext("onSuccessOnly");
// should have a single successful onComplete
inOrder.verify(observer, times(1)).onComplete();
inOrder.verifyNoMoreInteractions();
}
@Test
public void testSchedulingNotificationHandler() {
Observer<String> observer = TestHelper.mockObserver();
int NUM_RETRIES = 2;
Observable<String> origin = Observable.unsafeCreate(new FuncWithErrors(NUM_RETRIES));
TestObserver<String> to = new TestObserver<String>(observer);
origin.retryWhen(new Function<Observable<? extends Throwable>, Observable<Object>>() {
@Override
public Observable<Object> apply(Observable<? extends Throwable> t1) {
return t1
.observeOn(Schedulers.computation())
.map(new Function<Throwable, Object>() {
@Override
public Object apply(Throwable t1) {
return 1;
}
}).startWith(1);
}
})
.doOnError(new Consumer<Throwable>() {
@Override
public void accept(Throwable e) {
e.printStackTrace();
}
})
.subscribe(to);
to.awaitTerminalEvent();
InOrder inOrder = inOrder(observer);
// should show 3 attempts
inOrder.verify(observer, times(1 + NUM_RETRIES)).onNext("beginningEveryTime");
// should have no errors
inOrder.verify(observer, never()).onError(any(Throwable.class));
// should have a single success
inOrder.verify(observer, times(1)).onNext("onSuccessOnly");
// should have a single successful onComplete
inOrder.verify(observer, times(1)).onComplete();
inOrder.verifyNoMoreInteractions();
}
@Test
public void testOnNextFromNotificationHandler() {
Observer<String> observer = TestHelper.mockObserver();
int NUM_RETRIES = 2;
Observable<String> origin = Observable.unsafeCreate(new FuncWithErrors(NUM_RETRIES));
origin.retryWhen(new Function<Observable<? extends Throwable>, Observable<Object>>() {
@Override
public Observable<Object> apply(Observable<? extends Throwable> t1) {
return t1.map(new Function<Throwable, Integer>() {
@Override
public Integer apply(Throwable t1) {
return 0;
}
}).startWith(0).cast(Object.class);
}
}).subscribe(observer);
InOrder inOrder = inOrder(observer);
// should show 3 attempts
inOrder.verify(observer, times(NUM_RETRIES + 1)).onNext("beginningEveryTime");
// should have no errors
inOrder.verify(observer, never()).onError(any(Throwable.class));
// should have a single success
inOrder.verify(observer, times(1)).onNext("onSuccessOnly");
// should have a single successful onComplete
inOrder.verify(observer, times(1)).onComplete();
inOrder.verifyNoMoreInteractions();
}
@Test
public void testOnCompletedFromNotificationHandler() {
Observer<String> observer = TestHelper.mockObserver();
Observable<String> origin = Observable.unsafeCreate(new FuncWithErrors(1));
TestObserver<String> to = new TestObserver<String>(observer);
origin.retryWhen(new Function<Observable<? extends Throwable>, Observable<?>>() {
@Override
public Observable<?> apply(Observable<? extends Throwable> t1) {
return Observable.empty();
}
}).subscribe(to);
InOrder inOrder = inOrder(observer);
inOrder.verify(observer).onSubscribe((Disposable)notNull());
inOrder.verify(observer, never()).onNext("beginningEveryTime");
inOrder.verify(observer, never()).onNext("onSuccessOnly");
inOrder.verify(observer, times(1)).onComplete();
inOrder.verify(observer, never()).onError(any(Exception.class));
inOrder.verifyNoMoreInteractions();
}
@Test
public void testOnErrorFromNotificationHandler() {
Observer<String> observer = TestHelper.mockObserver();
Observable<String> origin = Observable.unsafeCreate(new FuncWithErrors(2));
origin.retryWhen(new Function<Observable<? extends Throwable>, Observable<?>>() {
@Override
public Observable<?> apply(Observable<? extends Throwable> t1) {
return Observable.error(new RuntimeException());
}
}).subscribe(observer);
InOrder inOrder = inOrder(observer);
inOrder.verify(observer).onSubscribe((Disposable)notNull());
inOrder.verify(observer, never()).onNext("beginningEveryTime");
inOrder.verify(observer, never()).onNext("onSuccessOnly");
inOrder.verify(observer, never()).onComplete();
inOrder.verify(observer, times(1)).onError(any(RuntimeException.class));
inOrder.verifyNoMoreInteractions();
}
@Test
public void testSingleSubscriptionOnFirst() throws Exception {
final AtomicInteger inc = new AtomicInteger(0);
ObservableSource<Integer> onSubscribe = new ObservableSource<Integer>() {
@Override
public void subscribe(Observer<? super Integer> observer) {
observer.onSubscribe(Disposables.empty());
final int emit = inc.incrementAndGet();
observer.onNext(emit);
observer.onComplete();
}
};
int first = Observable.unsafeCreate(onSubscribe)
.retryWhen(new Function<Observable<? extends Throwable>, Observable<?>>() {
@Override
public Observable<?> apply(Observable<? extends Throwable> attempt) {
return attempt.zipWith(Observable.just(1), new BiFunction<Throwable, Integer, Void>() {
@Override
public Void apply(Throwable o, Integer integer) {
return null;
}
});
}
})
.blockingFirst();
assertEquals("Observer did not receive the expected output", 1, first);
assertEquals("Subscribe was not called once", 1, inc.get());
}
@Test
public void testOriginFails() {
Observer<String> observer = TestHelper.mockObserver();
Observable<String> origin = Observable.unsafeCreate(new FuncWithErrors(1));
origin.subscribe(observer);
InOrder inOrder = inOrder(observer);
inOrder.verify(observer, times(1)).onNext("beginningEveryTime");
inOrder.verify(observer, times(1)).onError(any(RuntimeException.class));
inOrder.verify(observer, never()).onNext("onSuccessOnly");
inOrder.verify(observer, never()).onComplete();
}
@Test
public void testRetryFail() {
int NUM_RETRIES = 1;
int NUM_FAILURES = 2;
Observer<String> observer = TestHelper.mockObserver();
Observable<String> origin = Observable.unsafeCreate(new FuncWithErrors(NUM_FAILURES));
origin.retry(NUM_RETRIES).subscribe(observer);
InOrder inOrder = inOrder(observer);
// should show 2 attempts (first time fail, second time (1st retry) fail)
inOrder.verify(observer, times(1 + NUM_RETRIES)).onNext("beginningEveryTime");
// should only retry once, fail again and emit onError
inOrder.verify(observer, times(1)).onError(any(RuntimeException.class));
// no success
inOrder.verify(observer, never()).onNext("onSuccessOnly");
inOrder.verify(observer, never()).onComplete();
inOrder.verifyNoMoreInteractions();
}
@Test
public void testRetrySuccess() {
int NUM_FAILURES = 1;
Observer<String> observer = TestHelper.mockObserver();
Observable<String> origin = Observable.unsafeCreate(new FuncWithErrors(NUM_FAILURES));
origin.retry(3).subscribe(observer);
InOrder inOrder = inOrder(observer);
// should show 3 attempts
inOrder.verify(observer, times(1 + NUM_FAILURES)).onNext("beginningEveryTime");
// should have no errors
inOrder.verify(observer, never()).onError(any(Throwable.class));
// should have a single success
inOrder.verify(observer, times(1)).onNext("onSuccessOnly");
// should have a single successful onComplete
inOrder.verify(observer, times(1)).onComplete();
inOrder.verifyNoMoreInteractions();
}
@Test
public void testInfiniteRetry() {
int NUM_FAILURES = 20;
Observer<String> observer = TestHelper.mockObserver();
Observable<String> origin = Observable.unsafeCreate(new FuncWithErrors(NUM_FAILURES));
origin.retry().subscribe(observer);
InOrder inOrder = inOrder(observer);
// should show 3 attempts
inOrder.verify(observer, times(1 + NUM_FAILURES)).onNext("beginningEveryTime");
// should have no errors
inOrder.verify(observer, never()).onError(any(Throwable.class));
// should have a single success
inOrder.verify(observer, times(1)).onNext("onSuccessOnly");
// should have a single successful onComplete
inOrder.verify(observer, times(1)).onComplete();
inOrder.verifyNoMoreInteractions();
}
/*
* Checks in a simple and synchronous way that retry resubscribes
* after error. This test fails against 0.16.1-0.17.4, hangs on 0.17.5 and
* passes in 0.17.6 thanks to fix for issue #1027.
*/
@SuppressWarnings("unchecked")
@Test
public void testRetrySubscribesAgainAfterError() throws Exception {
// record emitted values with this action
Consumer<Integer> record = mock(Consumer.class);
InOrder inOrder = inOrder(record);
// always throw an exception with this action
Consumer<Integer> throwException = mock(Consumer.class);
doThrow(new RuntimeException()).when(throwException).accept(Mockito.anyInt());
// create a retrying Observable based on a PublishSubject
PublishSubject<Integer> subject = PublishSubject.create();
subject
// record item
.doOnNext(record)
// throw a RuntimeException
.doOnNext(throwException)
// retry on error
.retry()
// subscribe and ignore
.subscribe();
inOrder.verifyNoMoreInteractions();
subject.onNext(1);
inOrder.verify(record).accept(1);
subject.onNext(2);
inOrder.verify(record).accept(2);
subject.onNext(3);
inOrder.verify(record).accept(3);
inOrder.verifyNoMoreInteractions();
}
public static class FuncWithErrors implements ObservableSource<String> {
private final int numFailures;
private final AtomicInteger count = new AtomicInteger(0);
FuncWithErrors(int count) {
this.numFailures = count;
}
@Override
public void subscribe(final Observer<? super String> o) {
o.onSubscribe(Disposables.empty());
o.onNext("beginningEveryTime");
int i = count.getAndIncrement();
if (i < numFailures) {
o.onError(new RuntimeException("forced failure: " + (i + 1)));
} else {
o.onNext("onSuccessOnly");
o.onComplete();
}
}
}
@Test
public void testUnsubscribeFromRetry() {
PublishSubject<Integer> subject = PublishSubject.create();
final AtomicInteger count = new AtomicInteger(0);
Disposable sub = subject.retry().subscribe(new Consumer<Integer>() {
@Override
public void accept(Integer n) {
count.incrementAndGet();
}
});
subject.onNext(1);
sub.dispose();
subject.onNext(2);
assertEquals(1, count.get());
}
@Test
public void testRetryAllowsSubscriptionAfterAllSubscriptionsUnsubscribed() throws InterruptedException {
final AtomicInteger subsCount = new AtomicInteger(0);
ObservableSource<String> onSubscribe = new ObservableSource<String>() {
@Override
public void subscribe(Observer<? super String> s) {
subsCount.incrementAndGet();
s.onSubscribe(Disposables.fromRunnable(new Runnable() {
@Override
public void run() {
subsCount.decrementAndGet();
}
}));
}
};
Observable<String> stream = Observable.unsafeCreate(onSubscribe);
Observable<String> streamWithRetry = stream.retry();
Disposable sub = streamWithRetry.subscribe();
assertEquals(1, subsCount.get());
sub.dispose();
assertEquals(0, subsCount.get());
streamWithRetry.subscribe();
assertEquals(1, subsCount.get());
}
@Test
public void testSourceObservableCallsUnsubscribe() throws InterruptedException {
final AtomicInteger subsCount = new AtomicInteger(0);
final TestObserver<String> ts = new TestObserver<String>();
ObservableSource<String> onSubscribe = new ObservableSource<String>() {
@Override
public void subscribe(Observer<? super String> s) {
BooleanSubscription bs = new BooleanSubscription();
// if isUnsubscribed is true that means we have a bug such as
// https://github.com/ReactiveX/RxJava/issues/1024
if (!bs.isCancelled()) {
subsCount.incrementAndGet();
s.onError(new RuntimeException("failed"));
// it unsubscribes the child directly
// this simulates various error/completion scenarios that could occur
// or just a source that proactively triggers cleanup
// FIXME can't unsubscribe child
// s.unsubscribe();
bs.cancel();
} else {
s.onError(new RuntimeException());
}
}
};
Observable.unsafeCreate(onSubscribe).retry(3).subscribe(ts);
assertEquals(4, subsCount.get()); // 1 + 3 retries
}
@Test
public void testSourceObservableRetry1() throws InterruptedException {
final AtomicInteger subsCount = new AtomicInteger(0);
final TestObserver<String> ts = new TestObserver<String>();
ObservableSource<String> onSubscribe = new ObservableSource<String>() {
@Override
public void subscribe(Observer<? super String> s) {
s.onSubscribe(Disposables.empty());
subsCount.incrementAndGet();
s.onError(new RuntimeException("failed"));
}
};
Observable.unsafeCreate(onSubscribe).retry(1).subscribe(ts);
assertEquals(2, subsCount.get());
}
@Test
public void testSourceObservableRetry0() throws InterruptedException {
final AtomicInteger subsCount = new AtomicInteger(0);
final TestObserver<String> ts = new TestObserver<String>();
ObservableSource<String> onSubscribe = new ObservableSource<String>() {
@Override
public void subscribe(Observer<? super String> s) {
s.onSubscribe(Disposables.empty());
subsCount.incrementAndGet();
s.onError(new RuntimeException("failed"));
}
};
Observable.unsafeCreate(onSubscribe).retry(0).subscribe(ts);
assertEquals(1, subsCount.get());
}
static final class SlowObservable implements ObservableSource<Long> {
final AtomicInteger efforts = new AtomicInteger(0);
final AtomicInteger active = new AtomicInteger(0), maxActive = new AtomicInteger(0);
final AtomicInteger nextBeforeFailure;
private final int emitDelay;
SlowObservable(int emitDelay, int countNext) {
this.emitDelay = emitDelay;
this.nextBeforeFailure = new AtomicInteger(countNext);
}
@Override
public void subscribe(final Observer<? super Long> observer) {
final AtomicBoolean terminate = new AtomicBoolean(false);
observer.onSubscribe(Disposables.fromRunnable(new Runnable() {
@Override
public void run() {
terminate.set(true);
active.decrementAndGet();
}
}));
efforts.getAndIncrement();
active.getAndIncrement();
maxActive.set(Math.max(active.get(), maxActive.get()));
final Thread thread = new Thread() {
@Override
public void run() {
long nr = 0;
try {
while (!terminate.get()) {
Thread.sleep(emitDelay);
if (nextBeforeFailure.getAndDecrement() > 0) {
observer.onNext(nr++);
} else {
observer.onError(new RuntimeException("expected-failed"));
}
}
} catch (InterruptedException t) {
}
}
};
thread.start();
}
}
/** Observer for listener on seperate thread. */
static final class AsyncObserver<T> extends DefaultObserver<T> {
protected CountDownLatch latch = new CountDownLatch(1);
protected Observer<T> target;
/**
* Wrap existing Observer.
* @param target the target nbp subscriber
*/
AsyncObserver(Observer<T> target) {
this.target = target;
}
/** Wait. */
public void await() {
try {
latch.await();
} catch (InterruptedException e) {
fail("Test interrupted");
}
}
// Observer implementation
@Override
public void onComplete() {
target.onComplete();
latch.countDown();
}
@Override
public void onError(Throwable t) {
target.onError(t);
latch.countDown();
}
@Override
public void onNext(T v) {
target.onNext(v);
}
}
@Test(timeout = 10000)
public void testUnsubscribeAfterError() {
Observer<Long> observer = TestHelper.mockObserver();
// Observable that always fails after 100ms
SlowObservable so = new SlowObservable(100, 0);
Observable<Long> o = Observable.unsafeCreate(so).retry(5);
AsyncObserver<Long> async = new AsyncObserver<Long>(observer);
o.subscribe(async);
async.await();
InOrder inOrder = inOrder(observer);
// Should fail once
inOrder.verify(observer, times(1)).onError(any(Throwable.class));
inOrder.verify(observer, never()).onComplete();
assertEquals("Start 6 threads, retry 5 then fail on 6", 6, so.efforts.get());
assertEquals("Only 1 active subscription", 1, so.maxActive.get());
}
@Test(timeout = 10000)
public void testTimeoutWithRetry() {
@SuppressWarnings("unchecked")
DefaultObserver<Long> observer = mock(DefaultObserver.class);
// Observable that sends every 100ms (timeout fails instead)
SlowObservable so = new SlowObservable(100, 10);
Observable<Long> o = Observable.unsafeCreate(so).timeout(80, TimeUnit.MILLISECONDS).retry(5);
AsyncObserver<Long> async = new AsyncObserver<Long>(observer);
o.subscribe(async);
async.await();
InOrder inOrder = inOrder(observer);
// Should fail once
inOrder.verify(observer, times(1)).onError(any(Throwable.class));
inOrder.verify(observer, never()).onComplete();
assertEquals("Start 6 threads, retry 5 then fail on 6", 6, so.efforts.get());
}
@Test//(timeout = 15000)
public void testRetryWithBackpressure() throws InterruptedException {
final int NUM_LOOPS = 1;
for (int j = 0; j < NUM_LOOPS; j++) {
final int NUM_RETRIES = Flowable.bufferSize() * 2;
for (int i = 0; i < 400; i++) {
Observer<String> observer = TestHelper.mockObserver();
Observable<String> origin = Observable.unsafeCreate(new FuncWithErrors(NUM_RETRIES));
TestObserver<String> ts = new TestObserver<String>(observer);
origin.retry().observeOn(Schedulers.computation()).subscribe(ts);
ts.awaitTerminalEvent(5, TimeUnit.SECONDS);
InOrder inOrder = inOrder(observer);
// should have no errors
verify(observer, never()).onError(any(Throwable.class));
// should show NUM_RETRIES attempts
inOrder.verify(observer, times(NUM_RETRIES + 1)).onNext("beginningEveryTime");
// should have a single success
inOrder.verify(observer, times(1)).onNext("onSuccessOnly");
// should have a single successful onComplete
inOrder.verify(observer, times(1)).onComplete();
inOrder.verifyNoMoreInteractions();
}
}
}
@Test//(timeout = 15000)
public void testRetryWithBackpressureParallel() throws InterruptedException {
final int NUM_LOOPS = 1;
final int NUM_RETRIES = Flowable.bufferSize() * 2;
int ncpu = Runtime.getRuntime().availableProcessors();
ExecutorService exec = Executors.newFixedThreadPool(Math.max(ncpu / 2, 2));
try {
for (int r = 0; r < NUM_LOOPS; r++) {
if (r % 10 == 0) {
System.out.println("testRetryWithBackpressureParallelLoop -> " + r);
}
final AtomicInteger timeouts = new AtomicInteger();
final Map<Integer, List<String>> data = new ConcurrentHashMap<Integer, List<String>>();
int m = 5000;
final CountDownLatch cdl = new CountDownLatch(m);
for (int i = 0; i < m; i++) {
final int j = i;
exec.execute(new Runnable() {
@Override
public void run() {
final AtomicInteger nexts = new AtomicInteger();
try {
Observable<String> origin = Observable.unsafeCreate(new FuncWithErrors(NUM_RETRIES));
TestObserver<String> ts = new TestObserver<String>();
origin.retry()
.observeOn(Schedulers.computation()).subscribe(ts);
ts.awaitTerminalEvent(2500, TimeUnit.MILLISECONDS);
List<String> onNextEvents = new ArrayList<String>(ts.values());
if (onNextEvents.size() != NUM_RETRIES + 2) {
for (Throwable t : ts.errors()) {
onNextEvents.add(t.toString());
}
for (long err = ts.completions(); err != 0; err--) {
onNextEvents.add("onComplete");
}
data.put(j, onNextEvents);
}
} catch (Throwable t) {
timeouts.incrementAndGet();
System.out.println(j + " | " + cdl.getCount() + " !!! " + nexts.get());
}
cdl.countDown();
}
});
}
cdl.await();
assertEquals(0, timeouts.get());
if (data.size() > 0) {
fail("Data content mismatch: " + allSequenceFrequency(data));
}
}
} finally {
exec.shutdown();
}
}
static <T> StringBuilder allSequenceFrequency(Map<Integer, List<T>> its) {
StringBuilder b = new StringBuilder();
for (Map.Entry<Integer, List<T>> e : its.entrySet()) {
if (b.length() > 0) {
b.append(", ");
}
b.append(e.getKey()).append("={");
b.append(sequenceFrequency(e.getValue()));
b.append("}");
}
return b;
}
static <T> StringBuilder sequenceFrequency(Iterable<T> it) {
StringBuilder sb = new StringBuilder();
Object prev = null;
int cnt = 0;
for (Object curr : it) {
if (sb.length() > 0) {
if (!curr.equals(prev)) {
if (cnt > 1) {
sb.append(" x ").append(cnt);
cnt = 1;
}
sb.append(", ");
sb.append(curr);
} else {
cnt++;
}
} else {
sb.append(curr);
cnt++;
}
prev = curr;
}
if (cnt > 1) {
sb.append(" x ").append(cnt);
}
return sb;
}
@Test//(timeout = 3000)
public void testIssue1900() throws InterruptedException {
Observer<String> observer = TestHelper.mockObserver();
final int NUM_MSG = 1034;
final AtomicInteger count = new AtomicInteger();
Observable<String> origin = Observable.range(0, NUM_MSG)
.map(new Function<Integer, String>() {
@Override
public String apply(Integer t1) {
return "msg: " + count.incrementAndGet();
}
});
origin.retry()
.groupBy(new Function<String, String>() {
@Override
public String apply(String t1) {
return t1;
}
})
.flatMap(new Function<GroupedObservable<String,String>, Observable<String>>() {
@Override
public Observable<String> apply(GroupedObservable<String, String> t1) {
return t1.take(1);
}
})
.subscribe(new TestObserver<String>(observer));
InOrder inOrder = inOrder(observer);
// should show 3 attempts
inOrder.verify(observer, times(NUM_MSG)).onNext(any(java.lang.String.class));
// // should have no errors
inOrder.verify(observer, never()).onError(any(Throwable.class));
// should have a single success
//inOrder.verify(observer, times(1)).onNext("onSuccessOnly");
// should have a single successful onComplete
inOrder.verify(observer, times(1)).onComplete();
inOrder.verifyNoMoreInteractions();
}
@Test//(timeout = 3000)
public void testIssue1900SourceNotSupportingBackpressure() {
Observer<String> observer = TestHelper.mockObserver();
final int NUM_MSG = 1034;
final AtomicInteger count = new AtomicInteger();
Observable<String> origin = Observable.unsafeCreate(new ObservableSource<String>() {
@Override
public void subscribe(Observer<? super String> o) {
o.onSubscribe(Disposables.empty());
for (int i = 0; i < NUM_MSG; i++) {
o.onNext("msg:" + count.incrementAndGet());
}
o.onComplete();
}
});
origin.retry()
.groupBy(new Function<String, String>() {
@Override
public String apply(String t1) {
return t1;
}
})
.flatMap(new Function<GroupedObservable<String,String>, Observable<String>>() {
@Override
public Observable<String> apply(GroupedObservable<String, String> t1) {
return t1.take(1);
}
})
.subscribe(new TestObserver<String>(observer));
InOrder inOrder = inOrder(observer);
// should show 3 attempts
inOrder.verify(observer, times(NUM_MSG)).onNext(any(java.lang.String.class));
// // should have no errors
inOrder.verify(observer, never()).onError(any(Throwable.class));
// should have a single success
//inOrder.verify(observer, times(1)).onNext("onSuccessOnly");
// should have a single successful onComplete
inOrder.verify(observer, times(1)).onComplete();
inOrder.verifyNoMoreInteractions();
}
@Test
public void retryPredicate() {
Observable.just(1).concatWith(Observable.<Integer>error(new TestException()))
.retry(new Predicate<Throwable>() {
@Override
public boolean test(Throwable v) throws Exception {
return true;
}
})
.take(5)
.test()
.assertResult(1, 1, 1, 1, 1);
}
@Test
public void retryUntil() {
Observable.just(1).concatWith(Observable.<Integer>error(new TestException()))
.retryUntil(new BooleanSupplier() {
@Override
public boolean getAsBoolean() throws Exception {
return false;
}
})
.take(5)
.test()
.assertResult(1, 1, 1, 1, 1);
}
@Test
public void retryLongPredicateInvalid() {
try {
Observable.just(1).retry(-99, new Predicate<Throwable>() {
@Override
public boolean test(Throwable e) throws Exception {
return true;
}
});
fail("Should have thrown");
} catch (IllegalArgumentException ex) {
assertEquals("times >= 0 required but it was -99", ex.getMessage());
}
}
@Test
public void shouldDisposeInnerObservable() {
final PublishSubject<Object> subject = PublishSubject.create();
final Disposable disposable = Observable.error(new RuntimeException("Leak"))
.retryWhen(new Function<Observable<Throwable>, ObservableSource<Object>>() {
@Override
public ObservableSource<Object> apply(Observable<Throwable> errors) throws Exception {
return errors.switchMap(new Function<Throwable, ObservableSource<Object>>() {
@Override
public ObservableSource<Object> apply(Throwable ignore) throws Exception {
return subject;
}
});
}
})
.subscribe();
assertTrue(subject.hasObservers());
disposable.dispose();
assertFalse(subject.hasObservers());
}
}