PromiseMocker.java

/*******************************************************************************
 * Copyright (c) 2017 RedHat, Inc.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *   RedHat, Inc. - initial commit
 *******************************************************************************/package org.eclipse.che.plugin.testing.ide;

import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

import org.eclipse.che.api.promises.client.Function;
import org.eclipse.che.api.promises.client.Operation;
import org.eclipse.che.api.promises.client.OperationException;
import org.eclipse.che.api.promises.client.Promise;
import org.eclipse.che.api.promises.client.PromiseError;
import org.mockito.Matchers;

/**
 * Mockito mock provider for Che Promise objects.
 *
 * @author David Festal
 */
public class PromiseMocker<T> {
    final private Promise<T> promise;
    
    public PromiseMocker(Promise<T> promise) {
        this.promise = promise;
    }

    @SuppressWarnings("unchecked")
    public PromiseMocker() {
        this.promise = (Promise<T>) mock(Promise.class);
    }

    public Promise<T> getPromise() {
        return promise;
    }
    public <B> PromiseMocker<T> applyOnThenPromise(T value) {
        when(promise.thenPromise(Matchers.<Function<T, Promise<B>>> any())).then(new FunctionAnswer<Function<T, Promise<B>>, Promise<B>>(function -> {
            try {
                return function.apply(value);
            } catch(Exception e) {
                e.printStackTrace();
            }
            return null;
        }));
        return this;
    }
    
    public PromiseMocker<T> applyOnThenOperation(T value) {
        when(promise.then(Matchers.<Operation<T>> any())).then(new FunctionAnswer<Operation<T>, Promise<T>>( op -> {
            try {
                op.apply(value);
            } catch(Exception e) {
                e.printStackTrace();
            }
            return promise;
        }));
        return this;
    }
    
    public PromiseMocker<T> applyOnCatchErrorOperation(PromiseError error) {
        when(promise.catchError(Matchers.<Operation<PromiseError>> any())).then(new FunctionAnswer<Operation<PromiseError>, Promise<T>>(op -> {
            try {
                op.apply(error);
            } catch (OperationException e) {
                e.printStackTrace();
            }
            return promise;
        }));
        return this;
    }
}