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.observable;
  
  import static org.junit.Assert.*;
  import static org.mockito.ArgumentMatchers.any;
  import static org.mockito.Mockito.*;
  
  import java.util.*;
  import java.util.concurrent.TimeUnit;
  import java.util.concurrent.atomic.*;
  
  import org.junit.Test;
  
  import io.reactivex.rxjava3.core.*;
  import io.reactivex.rxjava3.core.Observable;
  import io.reactivex.rxjava3.core.Observer;
  import io.reactivex.rxjava3.disposables.*;
  import io.reactivex.rxjava3.exceptions.*;
  import io.reactivex.rxjava3.functions.*;
  import io.reactivex.rxjava3.internal.functions.Functions;
  import io.reactivex.rxjava3.observers.*;
  import io.reactivex.rxjava3.plugins.RxJavaPlugins;
  import io.reactivex.rxjava3.subjects.*;
  import io.reactivex.rxjava3.testsupport.*;
  
  public class ObservableWindowWithObservableTest extends RxJavaTest {
  
      @Test
      public void windowViaObservableNormal1() {
          PublishSubject<Integer> source = PublishSubject.create();
          PublishSubject<Integer> boundary = PublishSubject.create();
  
          final Observer<Object> o = TestHelper.mockObserver();
  
          final List<Observer<Object>> values = new ArrayList<>();
  
          Observer<Observable<Integer>> wo = new DefaultObserver<Observable<Integer>>() {
              @Override
              public void onNext(Observable<Integer> args) {
                  final Observer<Object> mo = TestHelper.mockObserver();
                  values.add(mo);
  
                  args.subscribe(mo);
              }
  
              @Override
              public void onError(Throwable e) {
                  o.onError(e);
              }
  
              @Override
              public void onComplete() {
                  o.onComplete();
              }
          };
  
          source.window(boundary).subscribe(wo);
  
          int n = 30;
          for (int i = 0; i < n; i++) {
              source.onNext(i);
              if (i % 3 == 2 && i < n - 1) {
                  boundary.onNext(i / 3);
              }
          }
          source.onComplete();
  
          verify(o, never()).onError(any(Throwable.class));
  
          assertEquals(n / 3, values.size());
  
          int j = 0;
          for (Observer<Object> mo : values) {
              verify(mo, never()).onError(any(Throwable.class));
              for (int i = 0; i < 3; i++) {
                  verify(mo).onNext(j + i);
              }
              verify(mo).onComplete();
              j += 3;
          }
  
          verify(o).onComplete();
      }
  
      @Test
      public void windowViaObservableBoundaryCompletes() {
          PublishSubject<Integer> source = PublishSubject.create();
          PublishSubject<Integer> boundary = PublishSubject.create();
  
          final Observer<Object> o = TestHelper.mockObserver();
  
          final List<Observer<Object>> values = new ArrayList<>();
  
          Observer<Observable<Integer>> wo = new DefaultObserver<Observable<Integer>>() {
              @Override
              public void onNext(Observable<Integer> args) {
                  final Observer<Object> mo = TestHelper.mockObserver();
                  values.add(mo);
  
                  args.subscribe(mo);
              }
  
              @Override
              public void onError(Throwable e) {
                  o.onError(e);
              }
  
              @Override
              public void onComplete() {
                  o.onComplete();
              }
          };
  
          source.window(boundary).subscribe(wo);
  
          int n = 30;
          for (int i = 0; i < n; i++) {
              source.onNext(i);
              if (i % 3 == 2 && i < n - 1) {
                  boundary.onNext(i / 3);
              }
          }
          boundary.onComplete();
  
          assertEquals(n / 3, values.size());
  
          int j = 0;
          for (Observer<Object> mo : values) {
              for (int i = 0; i < 3; i++) {
                  verify(mo).onNext(j + i);
              }
              verify(mo).onComplete();
              verify(mo, never()).onError(any(Throwable.class));
              j += 3;
          }
  
          verify(o).onComplete();
          verify(o, never()).onError(any(Throwable.class));
      }
  
      @Test
      public void windowViaObservableBoundaryThrows() {
          PublishSubject<Integer> source = PublishSubject.create();
          PublishSubject<Integer> boundary = PublishSubject.create();
  
          final Observer<Object> o = TestHelper.mockObserver();
  
          final List<Observer<Object>> values = new ArrayList<>();
  
          Observer<Observable<Integer>> wo = new DefaultObserver<Observable<Integer>>() {
              @Override
              public void onNext(Observable<Integer> args) {
                  final Observer<Object> mo = TestHelper.mockObserver();
                  values.add(mo);
  
                  args.subscribe(mo);
              }
  
              @Override
              public void onError(Throwable e) {
                  o.onError(e);
              }
  
              @Override
              public void onComplete() {
                  o.onComplete();
              }
          };
  
          source.window(boundary).subscribe(wo);
  
          source.onNext(0);
          source.onNext(1);
          source.onNext(2);
  
          boundary.onError(new TestException());
  
          assertEquals(1, values.size());
  
          Observer<Object> mo = values.get(0);
  
          verify(mo).onNext(0);
          verify(mo).onNext(1);
          verify(mo).onNext(2);
          verify(mo).onError(any(TestException.class));
  
          verify(o, never()).onComplete();
          verify(o).onError(any(TestException.class));
      }
  
      @Test
      public void windowViaObservableSourceThrows() {
          PublishSubject<Integer> source = PublishSubject.create();
          PublishSubject<Integer> boundary = PublishSubject.create();
  
          final Observer<Object> o = TestHelper.mockObserver();
  
          final List<Observer<Object>> values = new ArrayList<>();
  
          Observer<Observable<Integer>> wo = new DefaultObserver<Observable<Integer>>() {
              @Override
              public void onNext(Observable<Integer> args) {
                  final Observer<Object> mo = TestHelper.mockObserver();
                  values.add(mo);
  
                  args.subscribe(mo);
              }
  
              @Override
              public void onError(Throwable e) {
                  o.onError(e);
              }
  
              @Override
              public void onComplete() {
                  o.onComplete();
              }
          };
  
          source.window(boundary).subscribe(wo);
  
          source.onNext(0);
          source.onNext(1);
          source.onNext(2);
  
          source.onError(new TestException());
  
          assertEquals(1, values.size());
  
          Observer<Object> mo = values.get(0);
  
          verify(mo).onNext(0);
          verify(mo).onNext(1);
          verify(mo).onNext(2);
          verify(mo).onError(any(TestException.class));
  
          verify(o, never()).onComplete();
          verify(o).onError(any(TestException.class));
      }
  
      @Test
      public void boundaryDispose() {
          TestHelper.checkDisposed(Observable.never().window(Observable.never()));
      }
  
      @Test
      public void boundaryOnError() {
          TestObserverEx<Object> to = Observable.error(new TestException())
          .window(Observable.never())
          .flatMap(Functions.<Observable<Object>>identity(), true)
          .to(TestHelper.testConsumer())
          .assertFailure(CompositeException.class);
  
          List<Throwable> errors = TestHelper.compositeList(to.errors().get(0));
  
          TestHelper.assertError(errors, 0, TestException.class);
      }
  
      @Test
      public void innerBadSource() {
          TestHelper.checkBadSourceObservable(new Function<Observable<Integer>, Object>() {
              @Override
              public Object apply(Observable<Integer> o) throws Exception {
                  return Observable.just(1).window(o).flatMap(new Function<Observable<Integer>, ObservableSource<Integer>>() {
                      @Override
                      public ObservableSource<Integer> apply(Observable<Integer> v) throws Exception {
                          return v;
                      }
                  });
              }
          }, false, 1, 1, (Object[])null);
      }
  
      @Test
      public void reentrant() {
          final Subject<Integer> ps = PublishSubject.<Integer>create();
  
          TestObserver<Integer> to = new TestObserver<Integer>() {
              @Override
              public void onNext(Integer t) {
                  super.onNext(t);
                  if (t == 1) {
                      ps.onNext(2);
                      ps.onComplete();
                  }
              }
          };
  
          ps.window(BehaviorSubject.createDefault(1))
          .flatMap(new Function<Observable<Integer>, ObservableSource<Integer>>() {
              @Override
              public ObservableSource<Integer> apply(Observable<Integer> v) throws Exception {
                  return v;
              }
          })
          .subscribe(to);
  
          ps.onNext(1);
  
          to
          .awaitDone(1, TimeUnit.SECONDS)
          .assertResult(1, 2);
      }
  
      @Test
      public void badSource() {
          TestHelper.checkBadSourceObservable(new Function<Observable<Object>, Object>() {
              @Override
              public Object apply(Observable<Object> o) throws Exception {
                  return o.window(Observable.never()).flatMap(new Function<Observable<Object>, ObservableSource<Object>>() {
                      @Override
                      public ObservableSource<Object> apply(Observable<Object> v) throws Exception {
                          return v;
                      }
                  });
              }
          }, false, 1, 1, 1);
      }
  
      @Test
      public void boundaryDirectDoubleOnSubscribe() {
          TestHelper.checkDoubleOnSubscribeObservable(new Function<Observable<Object>, Observable<Observable<Object>>>() {
              @Override
              public Observable<Observable<Object>> apply(Observable<Object> f)
                      throws Exception {
                  return f.window(Observable.never()).takeLast(1);
              }
          });
      }
  
      @Test
      public void upstreamDisposedWhenOutputsDisposed() {
          PublishSubject<Integer> source = PublishSubject.create();
          PublishSubject<Integer> boundary = PublishSubject.create();
  
          TestObserver<Integer> to = source.window(boundary)
          .take(1)
          .flatMap(new Function<Observable<Integer>, ObservableSource<Integer>>() {
              @Override
              public ObservableSource<Integer> apply(
                      Observable<Integer> w) throws Exception {
                  return w.take(1);
              }
          })
          .test();
  
          source.onNext(1);
  
          assertFalse("source not disposed", source.hasObservers());
          assertFalse("boundary not disposed", boundary.hasObservers());
  
          to.assertResult(1);
      }
  
      @Test
      public void mainAndBoundaryBothError() {
          List<Throwable> errors = TestHelper.trackPluginErrors();
          try {
              final AtomicReference<Observer<? super Object>> ref = new AtomicReference<>();
  
              TestObserverEx<Observable<Object>> to = Observable.error(new TestException("main"))
              .window(new Observable<Object>() {
                  @Override
                  protected void subscribeActual(Observer<? super Object> observer) {
                      observer.onSubscribe(Disposable.empty());
                      ref.set(observer);
                  }
              })
              .doOnNext(new Consumer<Observable<Object>>() {
                  @Override
                  public void accept(Observable<Object> w) throws Throwable {
                      w.subscribe(Functions.emptyConsumer(), Functions.emptyConsumer()); // avoid abandonment
                  }
              })
              .to(TestHelper.<Observable<Object>>testConsumer());
  
              to
              .assertValueCount(1)
              .assertError(TestException.class)
              .assertErrorMessage("main")
              .assertNotComplete();
  
              ref.get().onError(new TestException("inner"));
  
              TestHelper.assertUndeliverable(errors, 0, TestException.class, "inner");
          } finally {
              RxJavaPlugins.reset();
          }
      }
  
      @Test
      public void mainCompleteBoundaryErrorRace() {
          final TestException ex = new TestException();
  
          for (int i = 0; i < TestHelper.RACE_LONG_LOOPS; i++) {
              List<Throwable> errors = TestHelper.trackPluginErrors();
              try {
                  final AtomicReference<Observer<? super Object>> refMain = new AtomicReference<>();
                  final AtomicReference<Observer<? super Object>> ref = new AtomicReference<>();
  
                  TestObserverEx<Observable<Object>> to = new Observable<Object>() {
                      @Override
                      protected void subscribeActual(Observer<? super Object> observer) {
                          observer.onSubscribe(Disposable.empty());
                          refMain.set(observer);
                      }
                  }
                  .window(new Observable<Object>() {
                      @Override
                      protected void subscribeActual(Observer<? super Object> observer) {
                          observer.onSubscribe(Disposable.empty());
                          ref.set(observer);
                      }
                  })
                  .to(TestHelper.<Observable<Object>>testConsumer());
  
                  Runnable r1 = new Runnable() {
                      @Override
                      public void run() {
                          refMain.get().onComplete();
                      }
                  };
                  Runnable r2 = new Runnable() {
                      @Override
                      public void run() {
                          ref.get().onError(ex);
                      }
                  };
  
                  TestHelper.race(r1, r2);
  
                  to
                  .assertValueCount(1)
                  .assertTerminated();
  
                  if (!errors.isEmpty()) {
                      TestHelper.assertUndeliverable(errors, 0, TestException.class);
                  }
              } finally {
                  RxJavaPlugins.reset();
              }
          }
      }
  
      @Test
      public void mainNextBoundaryNextRace() {
          for (int i = 0; i < TestHelper.RACE_LONG_LOOPS; i++) {
              final AtomicReference<Observer<? super Object>> refMain = new AtomicReference<>();
              final AtomicReference<Observer<? super Object>> ref = new AtomicReference<>();
  
              TestObserver<Observable<Object>> to = new Observable<Object>() {
                  @Override
                  protected void subscribeActual(Observer<? super Object> observer) {
                      observer.onSubscribe(Disposable.empty());
                      refMain.set(observer);
                  }
              }
              .window(new Observable<Object>() {
                  @Override
                  protected void subscribeActual(Observer<? super Object> observer) {
                      observer.onSubscribe(Disposable.empty());
                      ref.set(observer);
                  }
              })
              .test();
  
              Runnable r1 = new Runnable() {
                  @Override
                  public void run() {
                      refMain.get().onNext(1);
                  }
              };
              Runnable r2 = new Runnable() {
                  @Override
                  public void run() {
                      ref.get().onNext(1);
                  }
              };
  
              TestHelper.race(r1, r2);
  
              to
              .assertValueCount(2)
              .assertNotComplete()
              .assertNoErrors();
          }
      }
  
      @Test
      public void takeOneAnotherBoundary() {
          final AtomicReference<Observer<? super Object>> refMain = new AtomicReference<>();
          final AtomicReference<Observer<? super Object>> ref = new AtomicReference<>();
  
          TestObserverEx<Observable<Object>> to = new Observable<Object>() {
              @Override
              protected void subscribeActual(Observer<? super Object> observer) {
                  observer.onSubscribe(Disposable.empty());
                  refMain.set(observer);
              }
          }
          .window(new Observable<Object>() {
              @Override
              protected void subscribeActual(Observer<? super Object> observer) {
                  observer.onSubscribe(Disposable.empty());
                  ref.set(observer);
              }
          })
          .to(TestHelper.<Observable<Object>>testConsumer());
  
          to.assertValueCount(1)
          .assertNotTerminated()
          .dispose();
  
          ref.get().onNext(1);
  
          to.assertValueCount(1)
          .assertNotTerminated();
      }
  
      @Test
      public void disposeMainBoundaryCompleteRace() {
          for (int i = 0; i < TestHelper.RACE_LONG_LOOPS; i++) {
              final AtomicReference<Observer<? super Object>> refMain = new AtomicReference<>();
              final AtomicReference<Observer<? super Object>> ref = new AtomicReference<>();
  
              final TestObserver<Observable<Object>> to = new Observable<Object>() {
                   @Override
                   protected void subscribeActual(Observer<? super Object> observer) {
                       observer.onSubscribe(Disposable.empty());
                       refMain.set(observer);
                   }
               }
               .window(new Observable<Object>() {
                   @Override
                   protected void subscribeActual(Observer<? super Object> observer) {
                       final AtomicInteger counter = new AtomicInteger();
                       observer.onSubscribe(new Disposable() {
  
                           @Override
                           public void dispose() {
                               // about a microsecond
                               for (int i = 0; i < 100; i++) {
                                   counter.incrementAndGet();
                               }
                           }
  
                           @Override
                           public boolean isDisposed() {
                               return false;
                           }
                        });
                       ref.set(observer);
                   }
               })
               .test();
  
               Runnable r1 = new Runnable() {
                   @Override
                   public void run() {
                       to.dispose();
                   }
               };
               Runnable r2 = new Runnable() {
                   @Override
                   public void run() {
                       Observer<Object> o = ref.get();
                       o.onNext(1);
                       o.onComplete();
                   }
               };
  
               TestHelper.race(r1, r2);
          }
      }
  
      @Test
      @SuppressUndeliverable
      public void disposeMainBoundaryErrorRace() {
          final TestException ex = new TestException();
  
          for (int i = 0; i < TestHelper.RACE_LONG_LOOPS; i++) {
             final AtomicReference<Observer<? super Object>> refMain = new AtomicReference<>();
             final AtomicReference<Observer<? super Object>> ref = new AtomicReference<>();
  
             final TestObserver<Observable<Object>> to = new Observable<Object>() {
                 @Override
                 protected void subscribeActual(Observer<? super Object> observer) {
                     observer.onSubscribe(Disposable.empty());
                     refMain.set(observer);
                 }
             }
             .window(new Observable<Object>() {
                 @Override
                 protected void subscribeActual(Observer<? super Object> observer) {
                     final AtomicInteger counter = new AtomicInteger();
                     observer.onSubscribe(new Disposable() {
  
                         @Override
                         public void dispose() {
                             // about a microsecond
                             for (int i = 0; i < 100; i++) {
                                 counter.incrementAndGet();
                             }
                         }
  
                         @Override
                         public boolean isDisposed() {
                             return false;
                         }
                      });
                     ref.set(observer);
                 }
             })
             .test();
  
              Runnable r1 = new Runnable() {
                  @Override
                  public void run() {
                      to.dispose();
                  }
              };
              Runnable r2 = new Runnable() {
                  @Override
                  public void run() {
                      Observer<Object> o = ref.get();
                      o.onNext(1);
                      o.onError(ex);
                  }
              };
  
              TestHelper.race(r1, r2);
          }
      }
  
      @Test
      public void cancellingWindowCancelsUpstream() {
          PublishSubject<Integer> ps = PublishSubject.create();
  
          TestObserver<Integer> to = ps.window(Observable.<Integer>never())
          .take(1)
          .flatMap(new Function<Observable<Integer>, Observable<Integer>>() {
              @Override
              public Observable<Integer> apply(Observable<Integer> w) throws Throwable {
                  return w.take(1);
              }
          })
          .test();
  
          assertTrue(ps.hasObservers());
  
          ps.onNext(1);
  
          to
          .assertResult(1);
  
          assertFalse("Subject still has observers!", ps.hasObservers());
      }
  
      @Test
      public void windowAbandonmentCancelsUpstream() {
          PublishSubject<Integer> ps = PublishSubject.create();
  
          final AtomicReference<Observable<Integer>> inner = new AtomicReference<>();
  
          TestObserver<Observable<Integer>> to = ps.window(Observable.<Integer>never())
          .doOnNext(new Consumer<Observable<Integer>>() {
              @Override
              public void accept(Observable<Integer> v) throws Throwable {
                  inner.set(v);
              }
          })
          .test();
  
          assertTrue(ps.hasObservers());
  
          to
          .assertValueCount(1)
          ;
  
          ps.onNext(1);
  
          assertTrue(ps.hasObservers());
  
          to.dispose();
  
          to
          .assertValueCount(1)
          .assertNoErrors()
          .assertNotComplete();
  
          assertFalse("Subject still has observers!", ps.hasObservers());
  
          inner.get().test().assertResult();
      }
  }
  