content: /**
   * 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.flowable.internal.operators;
  
  import static org.junit.Assert.*;
  import static org.mockito.Mockito.*;
  
  import org.junit.Test;
  import org.reactivestreams.*;
  
  import io.reactivex.flowable.*;
  import io.reactivex.flowable.processors.PublishProcessor;
  import io.reactivex.flowable.subscribers.TestSubscriber;
  
  public class FlowableTakeUntilTest {
  
      @Test
      public void testTakeUntil() {
          Subscription sSource = mock(Subscription.class);
          Subscription sOther = mock(Subscription.class);
          TestObservable source = new TestObservable(sSource);
          TestObservable other = new TestObservable(sOther);
  
          Subscriber<String> result = TestHelper.mockSubscriber();
          Flowable<String> stringObservable = Flowable.unsafeCreate(source)
                  .takeUntil(Flowable.unsafeCreate(other));
          stringObservable.subscribe(result);
          source.sendOnNext("one");
          source.sendOnNext("two");
          other.sendOnNext("three");
          source.sendOnNext("four");
          source.sendOnCompleted();
          other.sendOnCompleted();
  
          verify(result, times(1)).onNext("one");
          verify(result, times(1)).onNext("two");
          verify(result, times(0)).onNext("three");
          verify(result, times(0)).onNext("four");
          verify(sSource, times(1)).cancel();
          verify(sOther, times(1)).cancel();
  
      }
  
      @Test
      public void testTakeUntilSourceCompleted() {
          Subscription sSource = mock(Subscription.class);
          Subscription sOther = mock(Subscription.class);
          TestObservable source = new TestObservable(sSource);
          TestObservable other = new TestObservable(sOther);
  
          Subscriber<String> result = TestHelper.mockSubscriber();
          Flowable<String> stringObservable = Flowable.unsafeCreate(source).takeUntil(Flowable.unsafeCreate(other));
          stringObservable.subscribe(result);
          source.sendOnNext("one");
          source.sendOnNext("two");
          source.sendOnCompleted();
  
          verify(result, times(1)).onNext("one");
          verify(result, times(1)).onNext("two");
          verify(sSource, never()).cancel();
          verify(sOther, times(1)).cancel();
  
      }
  
      @Test
      public void testTakeUntilSourceError() {
          Subscription sSource = mock(Subscription.class);
          Subscription sOther = mock(Subscription.class);
          TestObservable source = new TestObservable(sSource);
          TestObservable other = new TestObservable(sOther);
          Throwable error = new Throwable();
  
          Subscriber<String> result = TestHelper.mockSubscriber();
          Flowable<String> stringObservable = Flowable.unsafeCreate(source).takeUntil(Flowable.unsafeCreate(other));
          stringObservable.subscribe(result);
          source.sendOnNext("one");
          source.sendOnNext("two");
          source.sendOnError(error);
          source.sendOnNext("three");
  
          verify(result, times(1)).onNext("one");
          verify(result, times(1)).onNext("two");
          verify(result, times(0)).onNext("three");
          verify(result, times(1)).onError(error);
          verify(sSource, never()).cancel();
          verify(sOther, times(1)).cancel();
  
      }
  
      @Test
      public void testTakeUntilOtherError() {
          Subscription sSource = mock(Subscription.class);
          Subscription sOther = mock(Subscription.class);
          TestObservable source = new TestObservable(sSource);
          TestObservable other = new TestObservable(sOther);
          Throwable error = new Throwable();
  
          Subscriber<String> result = TestHelper.mockSubscriber();
          Flowable<String> stringObservable = Flowable.unsafeCreate(source).takeUntil(Flowable.unsafeCreate(other));
          stringObservable.subscribe(result);
          source.sendOnNext("one");
          source.sendOnNext("two");
          other.sendOnError(error);
          source.sendOnNext("three");
  
          verify(result, times(1)).onNext("one");
          verify(result, times(1)).onNext("two");
          verify(result, times(0)).onNext("three");
          verify(result, times(1)).onError(error);
          verify(result, times(0)).onComplete();
          verify(sSource, times(1)).cancel();
          verify(sOther, never()).cancel();
  
      }
  
      /**
       * If the 'other' onCompletes then we unsubscribe from the source and onComplete.
       */
      @Test
      public void testTakeUntilOtherCompleted() {
          Subscription sSource = mock(Subscription.class);
          Subscription sOther = mock(Subscription.class);
          TestObservable source = new TestObservable(sSource);
          TestObservable other = new TestObservable(sOther);
  
          Subscriber<String> result = TestHelper.mockSubscriber();
          Flowable<String> stringObservable = Flowable.unsafeCreate(source).takeUntil(Flowable.unsafeCreate(other));
          stringObservable.subscribe(result);
          source.sendOnNext("one");
          source.sendOnNext("two");
          other.sendOnCompleted();
          source.sendOnNext("three");
  
          verify(result, times(1)).onNext("one");
          verify(result, times(1)).onNext("two");
          verify(result, times(0)).onNext("three");
          verify(result, times(1)).onComplete();
          verify(sSource, times(1)).cancel();
          verify(sOther, never()).cancel(); // unsubscribed since SafeSubscriber unsubscribes after onComplete
  
      }
  
      private static class TestObservable implements Publisher<String> {
  
          Subscriber<? super String> observer;
          Subscription s;
  
          TestObservable(Subscription s) {
              this.s = s;
          }
  
          /* used to simulate subscription */
          public void sendOnCompleted() {
              observer.onComplete();
          }
  
          /* used to simulate subscription */
          public void sendOnNext(String value) {
              observer.onNext(value);
          }
  
          /* used to simulate subscription */
          public void sendOnError(Throwable e) {
              observer.onError(e);
          }
  
          @Override
          public void subscribe(Subscriber<? super String> observer) {
              this.observer = observer;
              observer.onSubscribe(s);
          }
      }
  
      @Test
      public void testUntilFires() {
          PublishProcessor<Integer> source = PublishProcessor.create();
          PublishProcessor<Integer> until = PublishProcessor.create();
  
          TestSubscriber<Integer> ts = new TestSubscriber<Integer>();
  
          source.takeUntil(until).subscribe(ts);
  
          assertTrue(source.hasSubscribers());
          assertTrue(until.hasSubscribers());
  
          source.onNext(1);
  
          ts.assertValue(1);
          until.onNext(1);
  
          ts.assertValue(1);
          ts.assertNoErrors();
          ts.assertTerminated();
  
          assertFalse("Source still has observers", source.hasSubscribers());
          assertFalse("Until still has observers", until.hasSubscribers());
          assertFalse("TestSubscriber is unsubscribed", ts.isCancelled());
      }
      @Test
      public void testMainCompletes() {
          PublishProcessor<Integer> source = PublishProcessor.create();
          PublishProcessor<Integer> until = PublishProcessor.create();
  
          TestSubscriber<Integer> ts = new TestSubscriber<Integer>();
  
          source.takeUntil(until).subscribe(ts);
  
          assertTrue(source.hasSubscribers());
          assertTrue(until.hasSubscribers());
  
          source.onNext(1);
          source.onComplete();
  
          ts.assertValue(1);
          ts.assertNoErrors();
          ts.assertTerminated();
  
          assertFalse("Source still has observers", source.hasSubscribers());
          assertFalse("Until still has observers", until.hasSubscribers());
          assertFalse("TestSubscriber is unsubscribed", ts.isCancelled());
      }
      @Test
      public void testDownstreamUnsubscribes() {
          PublishProcessor<Integer> source = PublishProcessor.create();
          PublishProcessor<Integer> until = PublishProcessor.create();
  
          TestSubscriber<Integer> ts = new TestSubscriber<Integer>();
  
          source.takeUntil(until).take(1).subscribe(ts);
  
          assertTrue(source.hasSubscribers());
          assertTrue(until.hasSubscribers());
  
          source.onNext(1);
  
          ts.assertValue(1);
          ts.assertNoErrors();
          ts.assertTerminated();
  
          assertFalse("Source still has observers", source.hasSubscribers());
          assertFalse("Until still has observers", until.hasSubscribers());
          assertFalse("TestSubscriber is unsubscribed", ts.isCancelled());
      }
  
      @Test
      public void testBackpressure() {
          PublishProcessor<Integer> until = PublishProcessor.create();
  
          TestSubscriber<Integer> ts = new TestSubscriber<Integer>(0L);
  
          Flowable.range(1, 10).takeUntil(until).subscribe(ts);
  
          assertTrue(until.hasSubscribers());
  
          ts.request(1);
  
          ts.assertValue(1);
          ts.assertNoErrors();
          ts.assertNotComplete();
  
          until.onNext(5);
  
          ts.assertComplete();
          ts.assertNoErrors();
  
          assertFalse("Until still has observers", until.hasSubscribers());
          assertFalse("TestSubscriber is unsubscribed", ts.isCancelled());
      }
  
      @Test
      public void dispose() {
          TestHelper.checkDisposed(PublishProcessor.create().takeUntil(Flowable.never()));
      }
  }
  