doAnswer to check which class called a specific function?

Question

I'm trying to mock a method when called only from a specific class, as the underlying framework calls my mock method N times.

Is there a way to delineate the invoker of a particular mocked method so I can conditionally return data based on the caller?

I'm using Mockito and the doAnswer API

Sorry for late reply. I took a different route, but I am glad you gave me this answer, neat trick. I had an error further up my code that led me down this rabbit hole that I resolved. — John Lippson, Jul 27 '18 at 21:01

score 1 · Accepted Answer · answered Jul 11 '18 at 18:07

If you are looking for call hierarchy of method this would be perfect answer

StackTraceElement[] cause = Thread.currentThread().getStackTrace();

From this array you can get all hierarchy class names, so that you check in this array which class called this method

score 1 · Answer 2 · answered Jul 11 '18 at 20:35

I think you should be able to return a custom Answer. Using this API you have access to the invocation of the mock. You may be able to use Deadpool's answer and should be able to conditionally return something based on the caller. Help that helps!

    Mockito.doAnswer(new Answer<Object>() {

        @Override
        public Object answer(InvocationOnMock invocation) throws Throwable {
            // TODO Auto-generated method stub
            // do something with the stack trace
            StackTraceElement[] cause = Thread.currentThread().getStackTrace();
            return somthing;
        }

    }).when(service).doSomething();

https://static.javadoc.io/org.mockito/mockito-core/1.10.19/org/mockito/invocation/InvocationOnMock.html

this question may help: How do I find the caller of a method using stacktrace or reflection?

doAnswer to check which class called a specific function?

2 Answers2