code: /*
   * 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.rxjava3.internal.operators.flowable;
  
  import static org.junit.Assert.assertEquals;
  
  import java.util.*;
  import java.util.concurrent.TimeUnit;
  import java.util.concurrent.atomic.AtomicInteger;
  
  import org.junit.Test;
  import org.reactivestreams.*;
  
  import io.reactivex.rxjava3.core.*;
  import io.reactivex.rxjava3.exceptions.TestException;
  import io.reactivex.rxjava3.functions.*;
  import io.reactivex.rxjava3.internal.functions.Functions;
  import io.reactivex.rxjava3.internal.subscriptions.BooleanSubscription;
  import io.reactivex.rxjava3.plugins.RxJavaPlugins;
  import io.reactivex.rxjava3.processors.PublishProcessor;
  import io.reactivex.rxjava3.schedulers.Schedulers;
  import io.reactivex.rxjava3.subscribers.TestSubscriber;
  import io.reactivex.rxjava3.testsupport.TestHelper;
  
  public class FlowableBlockingTest extends RxJavaTest {
  
      @Test
      public void blockingFirst() {
          assertEquals(1, Flowable.range(1, 10)
                  .subscribeOn(Schedulers.computation()).blockingFirst().intValue());
      }
  
      @Test
      public void blockingFirstDefault() {
          assertEquals(1, Flowable.<Integer>empty()
                  .subscribeOn(Schedulers.computation()).blockingFirst(1).intValue());
      }
  
      @Test
      public void blockingSubscribeConsumer() {
          final List<Integer> list = new ArrayList<>();
  
          Flowable.range(1, 5)
          .subscribeOn(Schedulers.computation())
          .blockingSubscribe(new Consumer<Integer>() {
              @Override
              public void accept(Integer v) throws Exception {
                  list.add(v);
              }
          });
  
          assertEquals(Arrays.asList(1, 2, 3, 4, 5), list);
      }
  
      @Test
      public void boundedBlockingSubscribeConsumer() {
          final List<Integer> list = new ArrayList<>();
  
          Flowable.range(1, 5)
                  .subscribeOn(Schedulers.computation())
                  .blockingSubscribe(new Consumer<Integer>() {
                      @Override
                      public void accept(Integer v) throws Exception {
                          list.add(v);
                      }
                  }, 128);
  
          assertEquals(Arrays.asList(1, 2, 3, 4, 5), list);
      }
  
      @Test
      public void boundedBlockingSubscribeConsumerBufferExceed() {
          final List<Integer> list = new ArrayList<>();
  
          Flowable.range(1, 5)
                  .subscribeOn(Schedulers.computation())
                  .blockingSubscribe(new Consumer<Integer>() {
                      @Override
                      public void accept(Integer v) throws Exception {
                          list.add(v);
                      }
                  }, 3);
  
          assertEquals(Arrays.asList(1, 2, 3, 4, 5), list);
      }
  
      @Test
      public void blockingSubscribeConsumerConsumer() {
          final List<Object> list = new ArrayList<>();
  
          Flowable.range(1, 5)
          .subscribeOn(Schedulers.computation())
          .blockingSubscribe(new Consumer<Integer>() {
              @Override
              public void accept(Integer v) throws Exception {
                  list.add(v);
              }
          }, Functions.emptyConsumer());
  
          assertEquals(Arrays.asList(1, 2, 3, 4, 5), list);
      }
  
      @Test
      public void boundedBlockingSubscribeConsumerConsumer() {
          final List<Object> list = new ArrayList<>();
  
          Flowable.range(1, 5)
                  .subscribeOn(Schedulers.computation())
                  .blockingSubscribe(new Consumer<Integer>() {
                      @Override
                      public void accept(Integer v) throws Exception {
                          list.add(v);
                      }
                  }, Functions.emptyConsumer(), 128);
  
          assertEquals(Arrays.asList(1, 2, 3, 4, 5), list);
      }
  
      @Test
      public void boundedBlockingSubscribeConsumerConsumerBufferExceed() {
          final List<Object> list = new ArrayList<>();
  
          Flowable.range(1, 5)
                  .subscribeOn(Schedulers.computation())
                  .blockingSubscribe(new Consumer<Integer>() {
                      @Override
                      public void accept(Integer v) throws Exception {
                          list.add(v);
                      }
                  }, Functions.emptyConsumer(), 3);
  
          assertEquals(Arrays.asList(1, 2, 3, 4, 5), list);
      }
  
      @Test
      public void blockingSubscribeConsumerConsumerError() {
          final List<Object> list = new ArrayList<>();
  
          TestException ex = new TestException();
  
          Consumer<Object> cons = new Consumer<Object>() {
              @Override
              public void accept(Object v) throws Exception {
                  list.add(v);
              }
          };
  
          Flowable.range(1, 5).concatWith(Flowable.<Integer>error(ex))
          .subscribeOn(Schedulers.computation())
          .blockingSubscribe(cons, cons);
  
          assertEquals(Arrays.asList(1, 2, 3, 4, 5, ex), list);
      }
  
      @Test
      public void boundedBlockingSubscribeConsumerConsumerError() {
          final List<Object> list = new ArrayList<>();
  
          TestException ex = new TestException();
  
          Consumer<Object> cons = new Consumer<Object>() {
              @Override
              public void accept(Object v) throws Exception {
                  list.add(v);
              }
          };
  
          Flowable.range(1, 5).concatWith(Flowable.<Integer>error(ex))
                  .subscribeOn(Schedulers.computation())
                  .blockingSubscribe(cons, cons, 128);
  
          assertEquals(Arrays.asList(1, 2, 3, 4, 5, ex), list);
      }
  
      @Test
      public void blockingSubscribeConsumerConsumerAction() {
          final List<Object> list = new ArrayList<>();
  
          Consumer<Object> cons = new Consumer<Object>() {
              @Override
              public void accept(Object v) throws Exception {
                  list.add(v);
              }
          };
  
          Flowable.range(1, 5)
          .subscribeOn(Schedulers.computation())
          .blockingSubscribe(cons, cons, new Action() {
              @Override
              public void run() throws Exception {
                  list.add(100);
              }
          });
  
          assertEquals(Arrays.asList(1, 2, 3, 4, 5, 100), list);
      }
  
      @Test
      public void boundedBlockingSubscribeConsumerConsumerAction() {
          final List<Object> list = new ArrayList<>();
  
          Consumer<Object> cons = new Consumer<Object>() {
              @Override
              public void accept(Object v) throws Exception {
                  list.add(v);
              }
          };
  
          Action action = new Action() {
              @Override
              public void run() throws Exception {
                  list.add(100);
              }
          };
  
          Flowable.range(1, 5)
                  .subscribeOn(Schedulers.computation())
                  .blockingSubscribe(cons, cons, action, 128);
  
          assertEquals(Arrays.asList(1, 2, 3, 4, 5, 100), list);
      }
  
      @Test
      public void boundedBlockingSubscribeConsumerConsumerActionBufferExceed() {
          final List<Object> list = new ArrayList<>();
  
          Consumer<Object> cons = new Consumer<Object>() {
              @Override
              public void accept(Object v) throws Exception {
                  list.add(v);
              }
          };
  
          Action action = new Action() {
              @Override
              public void run() throws Exception {
                  list.add(100);
              }
          };
  
          Flowable.range(1, 5)
                  .subscribeOn(Schedulers.computation())
                  .blockingSubscribe(cons, cons, action, 3);
  
          assertEquals(Arrays.asList(1, 2, 3, 4, 5, 100), list);
      }
  
      @Test
      public void boundedBlockingSubscribeConsumerConsumerActionBufferExceedMillionItem() {
          final List<Object> list = new ArrayList<>();
  
          Consumer<Object> cons = new Consumer<Object>() {
              @Override
              public void accept(Object v) throws Exception {
                  list.add(v);
              }
          };
  
          Action action = new Action() {
              @Override
              public void run() throws Exception {
                  list.add(1000001);
              }
          };
  
          Flowable.range(1, 1000000)
                  .subscribeOn(Schedulers.computation())
                  .blockingSubscribe(cons, cons, action, 128);
  
          assertEquals(1000000 + 1, list.size());
      }
  
      @Test
      public void blockingSubscribeObserver() {
          final List<Object> list = new ArrayList<>();
  
          Flowable.range(1, 5)
          .subscribeOn(Schedulers.computation())
          .blockingSubscribe(new FlowableSubscriber<Object>() {
  
              @Override
              public void onSubscribe(Subscription s) {
                  s.request(Long.MAX_VALUE);
              }
  
              @Override
              public void onNext(Object value) {
                  list.add(value);
              }
  
              @Override
              public void onError(Throwable e) {
                  list.add(e);
              }
  
              @Override
              public void onComplete() {
                  list.add(100);
              }
  
          });
  
          assertEquals(Arrays.asList(1, 2, 3, 4, 5, 100), list);
      }
  
      @Test
      public void blockingSubscribeObserverError() {
          final List<Object> list = new ArrayList<>();
  
          final TestException ex = new TestException();
  
          Flowable.range(1, 5).concatWith(Flowable.<Integer>error(ex))
          .subscribeOn(Schedulers.computation())
          .blockingSubscribe(new FlowableSubscriber<Object>() {
  
              @Override
              public void onSubscribe(Subscription s) {
                  s.request(Long.MAX_VALUE);
              }
  
              @Override
              public void onNext(Object value) {
                  list.add(value);
              }
  
              @Override
              public void onError(Throwable e) {
                  list.add(e);
              }
  
              @Override
              public void onComplete() {
                  list.add(100);
              }
  
          });
  
          assertEquals(Arrays.asList(1, 2, 3, 4, 5, ex), list);
      }
  
      @Test(expected = TestException.class)
      public void blockingForEachThrows() {
          Flowable.just(1)
          .blockingForEach(new Consumer<Integer>() {
              @Override
              public void accept(Integer e) throws Exception {
                  throw new TestException();
              }
          });
      }
  
      @Test(expected = NoSuchElementException.class)
      public void blockingFirstEmpty() {
          Flowable.empty().blockingFirst();
      }
  
      @Test(expected = NoSuchElementException.class)
      public void blockingLastEmpty() {
          Flowable.empty().blockingLast();
      }
  
      @Test
      public void blockingFirstNormal() {
          assertEquals(1, Flowable.just(1, 2).blockingFirst(3).intValue());
      }
  
      @Test
      public void blockingLastNormal() {
          assertEquals(2, Flowable.just(1, 2).blockingLast(3).intValue());
      }
  
      @Test
      public void firstFgnoredCancelAndOnNext() {
          Flowable<Integer> source = Flowable.fromPublisher(new Publisher<Integer>() {
              @Override
              public void subscribe(Subscriber<? super Integer> s) {
                  s.onSubscribe(new BooleanSubscription());
                  s.onNext(1);
                  s.onNext(2);
              }
          });
  
          assertEquals(1, source.blockingFirst().intValue());
      }
  
      @Test
      public void firstIgnoredCancelAndOnError() {
          List<Throwable> list = TestHelper.trackPluginErrors();
          try {
              Flowable<Integer> source = Flowable.fromPublisher(new Publisher<Integer>() {
                  @Override
                  public void subscribe(Subscriber<? super Integer> s) {
                      s.onSubscribe(new BooleanSubscription());
                      s.onNext(1);
                      s.onError(new TestException());
                  }
              });
  
              assertEquals(1, source.blockingFirst().intValue());
  
              TestHelper.assertUndeliverable(list, 0, TestException.class);
          } finally {
              RxJavaPlugins.reset();
          }
      }
  
      @Test(expected = TestException.class)
      public void firstOnError() {
          Flowable<Integer> source = Flowable.fromPublisher(new Publisher<Integer>() {
              @Override
              public void subscribe(Subscriber<? super Integer> s) {
                  s.onSubscribe(new BooleanSubscription());
                  s.onError(new TestException());
              }
          });
  
          source.blockingFirst();
      }
  
      @Test
      public void interrupt() {
          TestSubscriber<Integer> ts = new TestSubscriber<>(0L);
  
          Thread.currentThread().interrupt();
  
          try {
              Flowable.just(1)
              .blockingSubscribe(ts);
  
              ts.assertFailure(InterruptedException.class);
          } finally {
              Thread.interrupted(); // clear interrupted status just in case
          }
      }
  
      @Test(expected = NoSuchElementException.class)
      public void blockingSingleEmpty() {
          Flowable.empty().blockingSingle();
      }
  
      @Test
      public void onCompleteDelayed() {
          TestSubscriber<Object> ts = new TestSubscriber<>();
  
          Flowable.empty().delay(100, TimeUnit.MILLISECONDS)
          .blockingSubscribe(ts);
  
          ts.assertResult();
      }
  
      @Test
      public void utilityClass() {
          TestHelper.checkUtilityClass(FlowableBlockingSubscribe.class);
      }
  
      @Test
      public void disposeUpFront() {
          TestSubscriber<Object> ts = new TestSubscriber<>();
          ts.cancel();
          Flowable.just(1).blockingSubscribe(ts);
  
          ts.assertEmpty();
      }
  
      @SuppressWarnings("rawtypes")
      @Test
      public void delayed() throws Exception {
          final TestSubscriber<Object> ts = new TestSubscriber<>();
          final Subscriber[] s = { null };
  
          Schedulers.single().scheduleDirect(new Runnable() {
              @SuppressWarnings("unchecked")
              @Override
              public void run() {
                  ts.cancel();
                  s[0].onNext(1);
              }
          }, 200, TimeUnit.MILLISECONDS);
  
          new Flowable<Integer>() {
              @Override
              protected void subscribeActual(Subscriber<? super Integer> subscriber) {
                  subscriber.onSubscribe(new BooleanSubscription());
                  s[0] = subscriber;
              }
          }.blockingSubscribe(ts);
  
          while (!ts.isCancelled()) {
              Thread.sleep(100);
          }
  
          ts.assertEmpty();
      }
  
      @Test
      public void blockinsSubscribeCancelAsync() {
          for (int i = 0; i < TestHelper.RACE_DEFAULT_LOOPS; i++) {
              final TestSubscriber<Integer> ts = new TestSubscriber<>();
  
              final PublishProcessor<Integer> pp = PublishProcessor.create();
  
              final Runnable r1 = new Runnable() {
                  @Override
                  public void run() {
                      ts.cancel();
                  }
              };
  
              final Runnable r2 = new Runnable() {
                  @Override
                  public void run() {
                      pp.onNext(1);
                  }
              };
  
              final AtomicInteger c = new AtomicInteger(2);
  
              Schedulers.computation().scheduleDirect(new Runnable() {
                  @Override
                  public void run() {
                      c.decrementAndGet();
                      while (c.get() != 0 && !pp.hasSubscribers()) { }
  
                      TestHelper.race(r1, r2);
                  }
              });
  
              c.decrementAndGet();
              while (c.get() != 0) { }
  
              pp
              .blockingSubscribe(ts);
          }
      }
  }
  