src/test/java/io/reactivex/rxjava4/completable/CompletableRetryTest.java - 4.x - githubmirror/RxJava

src/test/java/io/reactivex/rxjava4/completable/CompletableRetryTest.java
142 строки4 KB
David Karnok
4.x: Convert from JUnit 4 to JUnit 6 (#8202)
30 июн 2026, 12:01
30 июн 2026, 12:01531388f
/*
 * 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.rxjava4.completable;

import static org.junit.jupiter.api.Assertions.assertEquals;

import java.util.concurrent.atomic.AtomicInteger;

import org.junit.jupiter.api.Test;

import io.reactivex.rxjava4.core.*;
import io.reactivex.rxjava4.exceptions.TestException;
import io.reactivex.rxjava4.internal.functions.Functions;

public class CompletableRetryTest extends RxJavaTest {
    @Test
    public void retryTimesPredicateWithMatchingPredicate() {
        final AtomicInteger atomicInteger = new AtomicInteger(3);
        final AtomicInteger numberOfSubscribeCalls = new AtomicInteger(0);

        Completable.fromAction(() -> {
            numberOfSubscribeCalls.incrementAndGet();

            if (atomicInteger.decrementAndGet() != 0) {
                throw new RuntimeException();
            }

            throw new IllegalArgumentException();
        })
        .retry(Integer.MAX_VALUE, throwable -> !(throwable instanceof IllegalArgumentException))
        .test()
        .assertFailure(IllegalArgumentException.class);

        assertEquals(3, numberOfSubscribeCalls.get());
    }

    @Test
    public void retryTimesPredicateWithMatchingRetryAmount() {
        final AtomicInteger atomicInteger = new AtomicInteger(3);
        final AtomicInteger numberOfSubscribeCalls = new AtomicInteger(0);

        Completable.fromAction(() -> {
            numberOfSubscribeCalls.incrementAndGet();

            if (atomicInteger.decrementAndGet() != 0) {
                throw new RuntimeException();
            }
        })
        .retry(2, Functions.alwaysTrue())
        .test()
        .assertResult();

        assertEquals(3, numberOfSubscribeCalls.get());
    }

    @Test
    public void retryTimesPredicateWithNotMatchingRetryAmount() {
        final AtomicInteger atomicInteger = new AtomicInteger(3);
        final AtomicInteger numberOfSubscribeCalls = new AtomicInteger(0);

        Completable.fromAction(() -> {
            numberOfSubscribeCalls.incrementAndGet();

            if (atomicInteger.decrementAndGet() != 0) {
                throw new RuntimeException();
            }
        })
        .retry(1, Functions.alwaysTrue())
        .test()
        .assertFailure(RuntimeException.class);

        assertEquals(2, numberOfSubscribeCalls.get());
    }

    @Test
    public void retryTimesPredicateWithZeroRetries() {
        final AtomicInteger atomicInteger = new AtomicInteger(2);
        final AtomicInteger numberOfSubscribeCalls = new AtomicInteger(0);

        Completable.fromAction(() -> {
            numberOfSubscribeCalls.incrementAndGet();

            if (atomicInteger.decrementAndGet() != 0) {
                throw new RuntimeException();
            }
        })
        .retry(0, Functions.alwaysTrue())
        .test()
        .assertFailure(RuntimeException.class);

        assertEquals(1, numberOfSubscribeCalls.get());
    }

    @Test
    public void untilTrueEmpty() {
        Completable.complete()
        .retryUntil(() -> true)
        .test()
        .assertResult();
    }

    @Test
    public void untilFalseEmpty() {
        Completable.complete()
        .retryUntil(() -> false)
        .test()
        .assertResult();
    }

    @Test
    public void untilTrueError() {
        Completable.error(new TestException())
        .retryUntil(() -> true)
        .test()
        .assertFailure(TestException.class);
    }

    @Test
    public void untilFalseError() {
        AtomicInteger counter = new AtomicInteger();
        Completable.defer(() -> {
            if (counter.getAndIncrement() == 0) {
                return Completable.error(new TestException());
            }
            return Completable.complete();
        })
        .retryUntil(() -> false)
        .test()
        .assertResult();
    }
}