io.quarkus.panache.mock.PanacheMock Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of quarkus-panache-mock Show documentation
Mocking with Panache
There is a newer version: 3.17.0.CR1
package io.quarkus.panache.mock;

import java.lang.reflect.Method;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

import org.mockito.Mockito;
import org.mockito.internal.debugging.LocationFactory;
import org.mockito.internal.invocation.DefaultInvocationFactory;
import org.mockito.internal.invocation.InterceptedInvocation;
import org.mockito.internal.invocation.RealMethod;
import org.mockito.internal.util.MockUtil;
import org.mockito.invocation.MockHandler;
import org.mockito.mock.MockCreationSettings;
import org.mockito.stubbing.Answer;
import org.mockito.verification.VerificationMode;

public class PanacheMock {

    public static volatile boolean IsMockEnabled = false;

    private final static Map, Object> mocks = new ConcurrentHashMap<>();

    @SuppressWarnings("unchecked")
    public static synchronized  T getMock(Class klass) {
        return (T) mocks.get(klass);
    }

    public static synchronized Object[] getMocks(Class... classes) {
        Object[] mocks = new Object[classes.length];
        for (int i = 0; i < classes.length; i++) {
            mocks[i] = getMock(classes[i]);
        }
        return mocks;
    }

    public static synchronized void mock(Class... classes) {
        for (Class klass : classes) {
            mocks.computeIfAbsent(klass, v -> Mockito.mock(klass));
        }
        IsMockEnabled = !mocks.isEmpty();
    }

    public static synchronized void reset() {
        mocks.clear();
        IsMockEnabled = false;
    }

    public static synchronized boolean isMocked(Class klass) {
        return mocks.containsKey(klass);
    }

    public static Object mockMethod(Class klass, String methodName, Class[] parameterTypes, Object[] args)
            throws InvokeRealMethodException {
        try {
            Method invokedMethod = klass.getDeclaredMethod(methodName, parameterTypes);
            Object mock = getMock(klass);
            MockCreationSettings settings = MockUtil.getMockSettings(mock);
            MyRealMethod myRealMethod = new MyRealMethod();
            InterceptedInvocation invocation = DefaultInvocationFactory.createInvocation(mock, invokedMethod, args,
                    myRealMethod, settings, LocationFactory.create(true));
            MockHandler handler = MockUtil.getMockHandler(mock);
            return handler.handle(invocation);
        } catch (InvokeRealMethodException e) {
            throw e;
        } catch (RuntimeException e) {
            throw e;
        } catch (Throwable e) {
            throw new RuntimeException(e);
        }
    }

    //
    // Delegating

    public static  T verify(Class klass) {
        return Mockito.verify(getMock(klass));
    }

    public static  T verify(Class klass, VerificationMode verificationMode) {
        return Mockito.verify(getMock(klass), verificationMode);
    }

    public static void verifyNoMoreInteractions(Class... classes) {
        Mockito.verifyNoMoreInteractions(getMocks(classes));
    }

    public static void verifyNoInteractions(Class... classes) {
        Mockito.verifyNoInteractions(getMocks(classes));
    }

    @SuppressWarnings("serial")
    public static class InvokeRealMethodException extends Exception {
    }

    @SuppressWarnings("serial")
    public static class MyRealMethod implements RealMethod {

        @Override
        public boolean isInvokable() {
            return true;
        }

        @Override
        public Object invoke() throws Throwable {
            throw new InvokeRealMethodException();
        }

    }

    public static PanacheStubber doAnswer(Answer answer) {
        return new PanacheStubber(Mockito.doAnswer(answer));
    }

    public static PanacheStubber doCallRealMethod() {
        return new PanacheStubber(Mockito.doCallRealMethod());
    }

    public static PanacheStubber doNothing() {
        return new PanacheStubber(Mockito.doNothing());
    }

    public static PanacheStubber doReturn(Object objectToBeReturned) {
        return new PanacheStubber(Mockito.doReturn(objectToBeReturned));
    }

    public static PanacheStubber doReturn(Object objectToBeReturned, Object... toBeReturnedNext) {
        return new PanacheStubber(Mockito.doReturn(objectToBeReturned, toBeReturnedNext));
    }

    public static PanacheStubber doThrow(Class toBeThrown) {
        return new PanacheStubber(Mockito.doThrow(toBeThrown));
    }

    public static PanacheStubber doThrow(Class toBeThrown, Class... toBeThrownNext) {
        return new PanacheStubber(Mockito.doThrow(toBeThrown, toBeThrownNext));
    }

    public static PanacheStubber doThrow(Throwable... toBeThrown) {
        return new PanacheStubber(Mockito.doThrow(toBeThrown));
    }
}