uk.camsw.rxjava.test.dsl.subscriber.BaseSubscriber Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of rxjava-test-dsl Show documentation
rxjava-test-dsl
The newest version!
package uk.camsw.rxjava.test.dsl.subscriber;

import org.assertj.core.api.AbstractThrowableAssert;
import rx.exceptions.OnErrorNotImplementedException;
import rx.subscriptions.SerialSubscription;
import uk.camsw.rxjava.test.dsl.scenario.ExecutionContext;
import uk.camsw.rxjava.test.dsl.when.IWhen;

import java.util.List;

import static org.assertj.core.api.Assertions.assertThat;

public class BaseSubscriber implements ISubscriber, rx.Observer {
    private final SerialSubscription subscription = new SerialSubscription();

    private final String id;
    private final ExecutionContext context;
    private final rx.observers.TestSubscriber inner;

    public BaseSubscriber(String id, ExecutionContext context) {
        this.id = id;
        this.context= context;
        this.inner = new rx.observers.TestSubscriber<>();
    }

    public String getId() {
        return id;
    }

    @Override
    public WHEN subscribes() {
        context.addCommand(c -> subscription.set(c.getStreamUnderTest().subscribe(this)));
        context.addFinally(c -> {
            if (!subscription.isUnsubscribed()) subscription.unsubscribe();
        });
        return context.getWhen();
    }

    @Override
    public WHEN unsubscribes() {
        context.addCommand(c -> subscription.unsubscribe());
        return context.getWhen();
    }

    @Override
    public WHEN waitsForEvents(int eventCount) {
        context.addCommand(context -> context.await().until(() -> inner.getOnNextEvents().size() >= eventCount));
        return context.getWhen();
    }

    @Override
    public int eventCount() {
        return inner.getOnNextEvents().size();
    }

    public boolean isErrored() {
        return inner.getOnErrorEvents().size() > 0;
    }

    @Override
    public U event(int index) {
        return getEvent(index);
    }

    public List events() {
        return inner.getOnNextEvents();
    }

    private U getEvent(int index) {
        return inner.getOnNextEvents().get(index);
    }

    @Override
    public int completedCount() {
        return inner.getOnCompletedEvents().size();
    }

    @Override
    public Class errorClass() {
        //noinspection ThrowableResultOfMethodCallIgnored
        return inner.getOnErrorEvents().get(0).getClass();
    }

    @Override
    public String errorMessage() {
        return inner.getOnErrorEvents().get(0).getMessage();
    }

    @Override
    public AbstractThrowableAssert error(int index) {
        return assertThat(inner.getOnErrorEvents().get(index));
    }

    @Override
    public void onCompleted() {
        inner.onCompleted();
    }

    @Override
    public void onError(Throwable e) {
        if (context.handleErrors()) {
            inner.onError(e);
        } else {
            throw new OnErrorNotImplementedException("Unhandled Error", e);
        }
    }

    @Override
    public void onNext(U next) {
        inner.onNext(next);
    }
}