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.single;
  
  import static org.junit.Assert.*;
  
  import java.util.*;
  import java.util.concurrent.atomic.AtomicReference;
  
  import org.junit.Test;
  
  import io.reactivex.rxjava3.core.*;
  import io.reactivex.rxjava3.disposables.Disposable;
  import io.reactivex.rxjava3.exceptions.TestException;
  import io.reactivex.rxjava3.functions.*;
  import io.reactivex.rxjava3.internal.functions.Functions;
  import io.reactivex.rxjava3.observers.TestObserver;
  import io.reactivex.rxjava3.plugins.RxJavaPlugins;
  import io.reactivex.rxjava3.processors.PublishProcessor;
  import io.reactivex.rxjava3.subjects.SingleSubject;
  import io.reactivex.rxjava3.testsupport.TestHelper;
  
  public class SingleZipArrayTest extends RxJavaTest {
  
      final BiFunction<Object, Object, Object> addString = new BiFunction<Object, Object, Object>() {
          @Override
          public Object apply(Object a, Object b) throws Exception {
              return "" + a + b;
          }
      };
  
      final Function3<Object, Object, Object, Object> addString3 = new Function3<Object, Object, Object, Object>() {
          @Override
          public Object apply(Object a, Object b, Object c) throws Exception {
              return "" + a + b + c;
          }
      };
  
      @Test
      public void firstError() {
          Single.zip(Single.error(new TestException()), Single.just(1), addString)
          .test()
          .assertFailure(TestException.class);
      }
  
      @Test
      public void secondError() {
          Single.zip(Single.just(1), Single.<Integer>error(new TestException()), addString)
          .test()
          .assertFailure(TestException.class);
      }
  
      @Test
      public void dispose() {
          PublishProcessor<Integer> pp = PublishProcessor.create();
  
          TestObserver<Object> to = Single.zip(pp.single(0), pp.single(0), addString)
          .test();
  
          assertTrue(pp.hasSubscribers());
  
          to.dispose();
  
          assertFalse(pp.hasSubscribers());
      }
  
      @Test
      public void zipperThrows() {
          Single.zip(Single.just(1), Single.just(2), new BiFunction<Integer, Integer, Object>() {
              @Override
              public Object apply(Integer a, Integer b) throws Exception {
                  throw new TestException();
              }
          })
          .test()
          .assertFailure(TestException.class);
      }
  
      @Test
      public void zipperReturnsNull() {
          Single.zip(Single.just(1), Single.just(2), new BiFunction<Integer, Integer, Object>() {
              @Override
              public Object apply(Integer a, Integer b) throws Exception {
                  return null;
              }
          })
          .test()
          .assertFailure(NullPointerException.class);
      }
  
      @Test
      public void middleError() {
          PublishProcessor<Integer> pp0 = PublishProcessor.create();
          PublishProcessor<Integer> pp1 = PublishProcessor.create();
  
          TestObserver<Object> to = Single.zip(pp0.single(0), pp1.single(0), pp0.single(0), addString3)
          .test();
  
          pp1.onError(new TestException());
  
          assertFalse(pp0.hasSubscribers());
  
          to.assertFailure(TestException.class);
      }
  
      @Test
      public void innerErrorRace() {
          for (int i = 0; i < TestHelper.RACE_DEFAULT_LOOPS; i++) {
              List<Throwable> errors = TestHelper.trackPluginErrors();
              try {
                  final PublishProcessor<Integer> pp0 = PublishProcessor.create();
                  final PublishProcessor<Integer> pp1 = PublishProcessor.create();
  
                  final TestObserver<Object> to = Single.zip(pp0.single(0), pp1.single(0), addString)
                  .test();
  
                  final TestException ex = new TestException();
  
                  Runnable r1 = new Runnable() {
                      @Override
                      public void run() {
                          pp0.onError(ex);
                      }
                  };
  
                  Runnable r2 = new Runnable() {
                      @Override
                      public void run() {
                          pp1.onError(ex);
                      }
                  };
  
                  TestHelper.race(r1, r2);
  
                  to.assertFailure(TestException.class);
  
                  if (!errors.isEmpty()) {
                      TestHelper.assertUndeliverable(errors, 0, TestException.class);
                  }
              } finally {
                  RxJavaPlugins.reset();
              }
          }
      }
  
      @Test(expected = NullPointerException.class)
      public void zipArrayOneIsNull() {
          Single.zipArray(new Function<Object[], Object>() {
              @Override
              public Object apply(Object[] v) {
                  return 1;
              }
          }, Single.just(1), null)
          .blockingGet();
      }
  
      @SuppressWarnings("unchecked")
      @Test
      public void emptyArray() {
          Single.zipArray(new Function<Object[], Object[]>() {
              @Override
              public Object[] apply(Object[] a) throws Exception {
                  return a;
              }
          }, new SingleSource[0])
          .test()
          .assertFailure(NoSuchElementException.class);
      }
  
      @Test
      public void oneArray() {
          Single.zipArray(new Function<Object[], Object>() {
              @Override
              public Object apply(Object[] a) throws Exception {
                  return (Integer)a[0] + 1;
              }
          }, Single.just(1))
          .test()
          .assertResult(2);
      }
  
      @Test
      public void singleSourceZipperReturnsNull() {
          Single.zipArray(Functions.justFunction(null), Single.just(1))
          .to(TestHelper.<Object>testConsumer())
          .assertFailureAndMessage(NullPointerException.class, "The zipper returned a null value");
      }
  
      @Test
      public void singleSourceZipperReturnsNull2() {
          Single.zipArray(Functions.justFunction(null), Single.just(1), Single.just(2))
          .to(TestHelper.<Object>testConsumer())
          .assertFailureAndMessage(NullPointerException.class, "The zipper returned a null value");
      }
  
      @Test
      public void dispose2() {
          TestHelper.checkDisposed(Single.zipArray(Functions.justFunction(1), SingleSubject.create(), SingleSubject.create()));
      }
  
      @Test
      public void bothSucceed() {
          Single.zipArray(a -> Arrays.asList(a), Single.just(1), Single.just(2))
          .test()
          .assertResult(Arrays.asList(1, 2));
      }
  
      @Test
      public void onSuccessAfterDispose() {
          AtomicReference<SingleObserver<? super Integer>> emitter = new AtomicReference<>();
  
          TestObserver<List<Object>> to = Single.zipArray(Arrays::asList,
                  (SingleSource<Integer>)o -> emitter.set(o), Single.<Integer>never())
          .test();
  
          emitter.get().onSubscribe(Disposable.empty());
  
          to.dispose();
  
          emitter.get().onSuccess(1);
  
          to.assertEmpty();
      }
  }
  