/**

* Copyright 2014 Netflix, Inc.

*

* 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 rx.observers;

import static org.junit.Assert.assertEquals;

import static org.junit.Assert.assertTrue;

import static org.mockito.Mockito.inOrder;

import static org.mockito.Mockito.mock;

import static org.mockito.Mockito.times;

import java.util.Arrays;

import java.util.concurrent.TimeUnit;

import java.util.concurrent.atomic.AtomicBoolean;

import org.junit.Assert;

import org.junit.Rule;

import org.junit.Test;

import org.junit.rules.ExpectedException;

import org.mockito.InOrder;

import rx.Observable;

import rx.Observer;

import rx.functions.Action0;

import rx.subjects.PublishSubject;

public class TestSubscriberTest {

@Rule

public ExpectedException thrown = ExpectedException.none();

@Test

public void testAssert() {

Observable<Integer> oi = Observable.from(Arrays.asList(1, 2));

TestSubscriber<Integer> o = new TestSubscriber<Integer>();

oi.subscribe(o);

o.assertReceivedOnNext(Arrays.asList(1, 2));

assertEquals(2, o.getOnNextEvents().size());

o.assertTerminalEvent();

}

@Test

public void testAssertNotMatchCount() {

Observable<Integer> oi = Observable.from(Arrays.asList(1, 2));

TestSubscriber<Integer> o = new TestSubscriber<Integer>();

oi.subscribe(o);

thrown.expect(AssertionError.class);

thrown.expectMessage("Number of items does not match. Provided: 1 Actual: 2");

o.assertReceivedOnNext(Arrays.asList(1));

assertEquals(2, o.getOnNextEvents().size());

o.assertTerminalEvent();

}

@Test

public void testAssertNotMatchValue() {

Observable<Integer> oi = Observable.from(Arrays.asList(1, 2));

TestSubscriber<Integer> o = new TestSubscriber<Integer>();

oi.subscribe(o);

thrown.expect(AssertionError.class);

thrown.expectMessage("Value at index: 1 expected to be [3] (Integer) but was: [2] (Integer)");

o.assertReceivedOnNext(Arrays.asList(1, 3));

assertEquals(2, o.getOnNextEvents().size());

o.assertTerminalEvent();

}

@Test

public void testAssertTerminalEventNotReceived() {

PublishSubject<Integer> p = PublishSubject.create();

TestSubscriber<Integer> o = new TestSubscriber<Integer>();

p.subscribe(o);

p.onNext(1);

p.onNext(2);

thrown.expect(AssertionError.class);

thrown.expectMessage("No terminal events received.");

o.assertReceivedOnNext(Arrays.asList(1, 2));

assertEquals(2, o.getOnNextEvents().size());

o.assertTerminalEvent();

}

@Test

public void testWrappingMock() {

Observable<Integer> oi = Observable.from(Arrays.asList(1, 2));

@SuppressWarnings("unchecked")

Observer<Integer> mockObserver = mock(Observer.class);

oi.subscribe(new TestSubscriber<Integer>(mockObserver));

InOrder inOrder = inOrder(mockObserver);

inOrder.verify(mockObserver, times(1)).onNext(1);

inOrder.verify(mockObserver, times(1)).onNext(2);

inOrder.verify(mockObserver, times(1)).onCompleted();

inOrder.verifyNoMoreInteractions();

}

@Test

public void testWrappingMockWhenUnsubscribeInvolved() {

Observable<Integer> oi = Observable.from(Arrays.asList(1, 2, 3, 4, 5, 6, 7, 8, 9)).take(2);

@SuppressWarnings("unchecked")

Observer<Integer> mockObserver = mock(Observer.class);

oi.subscribe(new TestSubscriber<Integer>(mockObserver));

InOrder inOrder = inOrder(mockObserver);

inOrder.verify(mockObserver, times(1)).onNext(1);

inOrder.verify(mockObserver, times(1)).onNext(2);

inOrder.verify(mockObserver, times(1)).onCompleted();

inOrder.verifyNoMoreInteractions();

}

@Test

public void testAssertError() {

RuntimeException e = new RuntimeException("Oops");

TestSubscriber<Object> subscriber = new TestSubscriber<Object>();

Observable.error(e).subscribe(subscriber);

subscriber.assertError(e);

}

@Test

public void testAwaitTerminalEventWithDuration() {

TestSubscriber<Object> ts = new TestSubscriber<Object>();

Observable.just(1).subscribe(ts);

ts.awaitTerminalEvent(1, TimeUnit.SECONDS);

ts.assertTerminalEvent();

}

@Test

public void testAwaitTerminalEventWithDurationAndUnsubscribeOnTimeout() {

TestSubscriber<Object> ts = new TestSubscriber<Object>();

final AtomicBoolean unsub = new AtomicBoolean(false);

Observable.just(1)

//

.doOnUnsubscribe(new Action0() {

@Override

public void call() {

unsub.set(true);

}

})

//

.delay(1000, TimeUnit.MILLISECONDS).subscribe(ts);

ts.awaitTerminalEventAndUnsubscribeOnTimeout(100, TimeUnit.MILLISECONDS);

assertTrue(unsub.get());

}

}