src/test/java/io/reactivex/rxjava4/schedulers/FailOnBlockingTest.java - 4.x - githubmirror/RxJava

src/test/java/io/reactivex/rxjava4/schedulers/FailOnBlockingTest.java
590 строк15 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.schedulers;

import java.util.concurrent.TimeUnit;

import org.junit.jupiter.api.Test;

import io.reactivex.rxjava4.core.*;
import io.reactivex.rxjava4.internal.functions.Functions;
import io.reactivex.rxjava4.plugins.RxJavaPlugins;

public class FailOnBlockingTest extends RxJavaTest {

    @Test
    public void failComputationFlowableBlockingFirst() {

        try {
            RxJavaPlugins.setFailOnNonBlockingScheduler(true);

            Flowable.just(1)
            .subscribeOn(Schedulers.computation())
            .map(v -> {

                Flowable.just(1).delay(10, TimeUnit.SECONDS).blockingFirst();

                return v;
            })
            .test()
            .awaitDone(5, TimeUnit.SECONDS)
            .assertFailure(IllegalStateException.class);
        } finally {
            RxJavaPlugins.reset();
        }

    }

    @Test
    public void failComputationFlowableBlockingLast() {

        try {
            RxJavaPlugins.setFailOnNonBlockingScheduler(true);

            Flowable.just(1)
            .subscribeOn(Schedulers.computation())
            .map(v -> {

                Flowable.just(1).delay(10, TimeUnit.SECONDS).blockingLast();

                return v;
            })
            .test()
            .awaitDone(5, TimeUnit.SECONDS)
            .assertFailure(IllegalStateException.class);
        } finally {
            RxJavaPlugins.reset();
        }

    }

    @Test
    public void failComputationFlowableBlockingIterable() {

        try {
            RxJavaPlugins.setFailOnNonBlockingScheduler(true);

            Flowable.just(1)
            .subscribeOn(Schedulers.computation())
            .map(v -> {

                Flowable.just(1).delay(10, TimeUnit.SECONDS).blockingIterable().iterator().next();

                return v;
            })
            .test()
            .awaitDone(5, TimeUnit.SECONDS)
            .assertFailure(IllegalStateException.class);
        } finally {
            RxJavaPlugins.reset();
        }

    }

    @Test
    public void failComputationFlowableBlockingSubscribe() {

        try {
            RxJavaPlugins.setFailOnNonBlockingScheduler(true);

            Flowable.just(1)
            .subscribeOn(Schedulers.computation())
            .map(v -> {

                Flowable.just(1).delay(10, TimeUnit.SECONDS).blockingSubscribe();

                return v;
            })
            .test()
            .awaitDone(5, TimeUnit.SECONDS)
            .assertFailure(IllegalStateException.class);
        } finally {
            RxJavaPlugins.reset();
        }

    }

    @Test
    public void failComputationFlowableBlockingSingle() {

        try {
            RxJavaPlugins.setFailOnNonBlockingScheduler(true);

            Flowable.just(1)
            .subscribeOn(Schedulers.computation())
            .map(v -> {

                Flowable.just(1).delay(10, TimeUnit.SECONDS).blockingSingle();

                return v;
            })
            .test()
            .awaitDone(5, TimeUnit.SECONDS)
            .assertFailure(IllegalStateException.class);
        } finally {
            RxJavaPlugins.reset();
        }

    }

    @Test
    public void failComputationFlowableBlockingForEach() {

        try {
            RxJavaPlugins.setFailOnNonBlockingScheduler(true);

            Flowable.just(1)
            .subscribeOn(Schedulers.computation())
            .map(v -> {

                Flowable.just(1).delay(10, TimeUnit.SECONDS).blockingForEach(Functions.emptyConsumer());

                return v;
            })
            .test()
            .awaitDone(5, TimeUnit.SECONDS)
            .assertFailure(IllegalStateException.class);
        } finally {
            RxJavaPlugins.reset();
        }

    }

    @Test
    public void failComputationFlowableBlockingLatest() {

        try {
            RxJavaPlugins.setFailOnNonBlockingScheduler(true);

            Flowable.just(1)
            .subscribeOn(Schedulers.computation())
            .map(v -> {

                Flowable.just(1).delay(10, TimeUnit.SECONDS).blockingLatest().iterator().hasNext();

                return v;
            })
            .test()
            .awaitDone(5, TimeUnit.SECONDS)
            .assertFailure(IllegalStateException.class);
        } finally {
            RxJavaPlugins.reset();
        }

    }

    @Test
    public void failComputationFlowableBlockingNext() {

        try {
            RxJavaPlugins.setFailOnNonBlockingScheduler(true);

            Flowable.just(1)
            .subscribeOn(Schedulers.computation())
            .map(v -> {

                Flowable.just(1).delay(10, TimeUnit.SECONDS).blockingNext().iterator().hasNext();

                return v;
            })
            .test()
            .awaitDone(5, TimeUnit.SECONDS)
            .assertFailure(IllegalStateException.class);
        } finally {
            RxJavaPlugins.reset();
        }

    }

    @Test
    public void failComputationFlowableToFuture() {

        try {
            RxJavaPlugins.setFailOnNonBlockingScheduler(true);

            Flowable.just(1)
            .subscribeOn(Schedulers.computation())
            .map(v -> {

                Flowable.just(1).delay(10, TimeUnit.SECONDS).toFuture().get();

                return v;
            })
            .test()
            .awaitDone(5, TimeUnit.SECONDS)
            .assertFailure(IllegalStateException.class);
        } finally {
            RxJavaPlugins.reset();
        }

    }

    @Test
    public void failComputationObservableBlockingFirst() {

        try {
            RxJavaPlugins.setFailOnNonBlockingScheduler(true);

            Observable.just(1)
            .subscribeOn(Schedulers.computation())
            .map(v -> {

                Observable.just(1).delay(10, TimeUnit.SECONDS).blockingFirst();

                return v;
            })
            .test()
            .awaitDone(5, TimeUnit.SECONDS)
            .assertFailure(IllegalStateException.class);
        } finally {
            RxJavaPlugins.reset();
        }

    }

    @Test
    public void failComputationObservableBlockingLast() {

        try {
            RxJavaPlugins.setFailOnNonBlockingScheduler(true);

            Observable.just(1)
            .subscribeOn(Schedulers.computation())
            .map(v -> {

                Flowable.just(1).delay(10, TimeUnit.SECONDS).blockingLast();

                return v;
            })
            .test()
            .awaitDone(5, TimeUnit.SECONDS)
            .assertFailure(IllegalStateException.class);
        } finally {
            RxJavaPlugins.reset();
        }

    }

    @Test
    public void failComputationObservableBlockingIterable() {

        try {
            RxJavaPlugins.setFailOnNonBlockingScheduler(true);

            Observable.just(1)
            .subscribeOn(Schedulers.computation())
            .map(v -> {

                Observable.just(1).delay(10, TimeUnit.SECONDS).blockingIterable().iterator().next();

                return v;
            })
            .test()
            .awaitDone(5, TimeUnit.SECONDS)
            .assertFailure(IllegalStateException.class);
        } finally {
            RxJavaPlugins.reset();
        }

    }

    @Test
    public void failComputationObservableBlockingSubscribe() {

        try {
            RxJavaPlugins.setFailOnNonBlockingScheduler(true);

            Observable.just(1)
            .subscribeOn(Schedulers.computation())
            .map(v -> {

                Observable.just(1).delay(10, TimeUnit.SECONDS).blockingSubscribe();

                return v;
            })
            .test()
            .awaitDone(5, TimeUnit.SECONDS)
            .assertFailure(IllegalStateException.class);
        } finally {
            RxJavaPlugins.reset();
        }

    }

    @Test
    public void failComputationObservableBlockingSingle() {

        try {
            RxJavaPlugins.setFailOnNonBlockingScheduler(true);

            Observable.just(1)
            .subscribeOn(Schedulers.computation())
            .map(v -> {

                Observable.just(1).delay(10, TimeUnit.SECONDS).blockingSingle();

                return v;
            })
            .test()
            .awaitDone(5, TimeUnit.SECONDS)
            .assertFailure(IllegalStateException.class);
        } finally {
            RxJavaPlugins.reset();
        }

    }

    @Test
    public void failComputationObservableBlockingForEach() {

        try {
            RxJavaPlugins.setFailOnNonBlockingScheduler(true);

            Observable.just(1)
            .subscribeOn(Schedulers.computation())
            .map(v -> {

                Observable.just(1).delay(10, TimeUnit.SECONDS).blockingForEach(Functions.emptyConsumer());

                return v;
            })
            .test()
            .awaitDone(5, TimeUnit.SECONDS)
            .assertFailure(IllegalStateException.class);
        } finally {
            RxJavaPlugins.reset();
        }

    }

    @Test
    public void failComputationObservableBlockingLatest() {

        try {
            RxJavaPlugins.setFailOnNonBlockingScheduler(true);

            Observable.just(1)
            .subscribeOn(Schedulers.computation())
            .map(v -> {

                Observable.just(1).delay(10, TimeUnit.SECONDS).blockingLatest().iterator().hasNext();

                return v;
            })
            .test()
            .awaitDone(5, TimeUnit.SECONDS)
            .assertFailure(IllegalStateException.class);
        } finally {
            RxJavaPlugins.reset();
        }

    }

    @Test
    public void failComputationObservableBlockingNext() {

        try {
            RxJavaPlugins.setFailOnNonBlockingScheduler(true);

            Observable.just(1)
            .subscribeOn(Schedulers.computation())
            .map(v -> {

                Observable.just(1).delay(10, TimeUnit.SECONDS).blockingNext().iterator().hasNext();

                return v;
            })
            .test()
            .awaitDone(5, TimeUnit.SECONDS)
            .assertFailure(IllegalStateException.class);
        } finally {
            RxJavaPlugins.reset();
        }

    }

    @Test
    public void failComputationObservableToFuture() {

        try {
            RxJavaPlugins.setFailOnNonBlockingScheduler(true);

            Observable.just(1)
            .subscribeOn(Schedulers.computation())
            .map(v -> {

                Observable.just(1).delay(10, TimeUnit.SECONDS).toFuture().get();

                return v;
            })
            .test()
            .awaitDone(5, TimeUnit.SECONDS)
            .assertFailure(IllegalStateException.class);
        } finally {
            RxJavaPlugins.reset();
        }

    }

    @Test
    public void failSingleObservableBlockingFirst() {

        try {
            RxJavaPlugins.setFailOnNonBlockingScheduler(true);

            Observable.just(1)
            .subscribeOn(Schedulers.single())
            .map(v -> {

                Observable.just(1).delay(10, TimeUnit.SECONDS).blockingFirst();

                return v;
            })
            .test()
            .awaitDone(5, TimeUnit.SECONDS)
            .assertFailure(IllegalStateException.class);
        } finally {
            RxJavaPlugins.reset();
        }

    }

    @Test
    public void failSingleSingleBlockingGet() {

        try {
            RxJavaPlugins.setFailOnNonBlockingScheduler(true);

            Single.just(1)
            .subscribeOn(Schedulers.single())
            .map(v -> {

                Single.just(1).delay(10, TimeUnit.SECONDS).blockingGet();

                return v;
            })
            .test()
            .awaitDone(5, TimeUnit.SECONDS)
            .assertFailure(IllegalStateException.class);
        } finally {
            RxJavaPlugins.reset();
        }

    }

    @Test
    public void failSingleMaybeBlockingGet() {

        try {
            RxJavaPlugins.setFailOnNonBlockingScheduler(true);

            Maybe.just(1)
            .subscribeOn(Schedulers.single())
            .map(v -> {

                Maybe.just(1).delay(10, TimeUnit.SECONDS).blockingGet();

                return v;
            })
            .test()
            .awaitDone(5, TimeUnit.SECONDS)
            .assertFailure(IllegalStateException.class);
        } finally {
            RxJavaPlugins.reset();
        }

    }

    @Test
    public void failSingleCompletableBlockingGet() {

        try {
            RxJavaPlugins.setFailOnNonBlockingScheduler(true);

            Completable.complete()
            .subscribeOn(Schedulers.single())
            .doOnComplete(() -> Completable.complete().delay(10, TimeUnit.SECONDS).blockingAwait())
            .test()
            .awaitDone(5, TimeUnit.SECONDS)
            .assertFailure(IllegalStateException.class);
        } finally {
            RxJavaPlugins.reset();
        }

    }

    @Test
    public void failSingleCompletableBlockingAwait() {

        try {
            RxJavaPlugins.setFailOnNonBlockingScheduler(true);

            Completable.complete()
            .subscribeOn(Schedulers.single())
            .doOnComplete(() -> Completable.complete().delay(10, TimeUnit.SECONDS).blockingAwait())
            .test()
            .awaitDone(5, TimeUnit.SECONDS)
            .assertFailure(IllegalStateException.class);
        } finally {
            RxJavaPlugins.reset();
        }

    }

    @Test
    public void dontfailIOObservableBlockingFirst() {

        try {
            RxJavaPlugins.setFailOnNonBlockingScheduler(true);

            Observable.just(1)
            .subscribeOn(Schedulers.cached())
            .map(_ -> Observable.just(2).delay(100, TimeUnit.MILLISECONDS).blockingFirst())
            .test()
            .awaitDone(5, TimeUnit.SECONDS)
            .assertResult(2);
        } finally {
            RxJavaPlugins.reset();
        }

    }

    @Test
    public void failWithCustomHandler() {
        try {
            RxJavaPlugins.setOnBeforeBlocking(() -> true);
            RxJavaPlugins.setFailOnNonBlockingScheduler(true);

            Flowable.just(1)
            .map(v -> {

                Flowable.just(1).delay(10, TimeUnit.SECONDS).blockingLast();

                return v;
            })
            .test()
            .awaitDone(5, TimeUnit.SECONDS)
            .assertFailure(IllegalStateException.class);

        } finally {
            RxJavaPlugins.reset();
        }

        Flowable.just(1)
        .map(_ -> Flowable.just(2).delay(100, TimeUnit.MILLISECONDS).blockingLast())
        .test()
        .awaitDone(5, TimeUnit.SECONDS)
        .assertResult(2);
    }
}