0Pricing
Testing Mastery: JUnit, Mockito & Integration Tests · Aula

Respostas e callbacks personalizados

Implemente lógica personalizada para chamadas de métodos simulados usando interfaces `Answer` a fim de simular comportamentos complexos.

Respostas e callbacks personalizados é uma aula grátis de Testing Mastery: JUnit, Mockito & Integration Tests no CoddyKit. Esta é a aula 1 de 4. Você pode ler a aula completa abaixo gratuitamente — depois pratica ao vivo no navegador com um editor de código integrado e um tutor de IA 24/7. Faz parte do caminho de aprendizado de Testing Mastery: JUnit, Mockito & Integration Tests, e seu progresso é sincronizado entre a web e o app CoddyKit. O curso de Testing Mastery: JUnit, Mockito & Integration Tests inclui 4 aulas no total.

Partes desta aula ainda não foram traduzidas e aparecem em inglês.

Beyond Simple Stubbing

Sometimes, simply returning a fixed value with when().thenReturn() isn't enough for your mock objects.

  • What if a mock method needs to modify an argument passed to it?
  • What if it needs to execute a callback function?
  • What if its return value depends on multiple inputs in a complex, dynamic way?

These advanced scenarios call for custom logic within your mocks.

Introducing Mockito's Answer

Mockito provides the org.mockito.stubbing.Answer interface. This powerful tool lets you define custom behavior for a mocked method call.

  • Think of it as writing a small piece of code that runs whenever the mocked method is invoked.
  • The Answer interface has a single method: Object answer(Invocation invocation) throws Throwable;.

It gives you fine-grained control over mock responses.

Anatomy of answer()

The answer() method is where you implement your custom logic. It receives an InvocationOnMock object (often just referred to as invocation).

  • The InvocationOnMock object provides crucial details about the method call:
    • Which method was called?
    • What arguments were passed?
    • Which mock object was involved?

You can use these details to decide what to return, throw an exception, or perform side effects.

Using doAnswer for Flexibility

While you can use when().thenAnswer(), Mockito's doAnswer() method is often more flexible and widely used, especially for void methods or when combining with other stubbings.

  • doAnswer() allows you to specify behavior for a method call before defining when().
  • It's particularly useful when you need to:
    • Capture arguments.
    • Invoke callbacks.
    • Change the state of objects passed as arguments.

Custom Return Logic

Let's see how to use doAnswer to return a value that depends on the input argument. Here, our mock will double the input number.

import org.mockito.Mockito;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;

// A simple interface to mock
interface Calculator {
    int calculate(int input);
}

public class Main {
    public static void main(String[] args) {
        // Create a mock object for our Calculator interface
        Calculator mockCalculator = Mockito.mock(Calculator.class);

        // Define custom answer to double the input argument
        Mockito.doAnswer(new Answer<Integer>() {
            @Override
            public Integer answer(InvocationOnMock invocation) throws Throwable {
                // Get the first argument (index 0) as an Integer
                Integer arg = invocation.getArgument(0);
                return arg * 2; // Return double the argument
            }
        }).when(mockCalculator).calculate(Mockito.anyInt()); // Apply to any int input

        // Test the custom behavior of the mock
        System.out.println("Result for 5: " + mockCalculator.calculate(5));
        System.out.println("Result for 10: " + mockCalculator.calculate(10));
    }
}

Simulating Callbacks

A very common and powerful use case for doAnswer is simulating asynchronous operations or invoking callbacks.

  • Imagine a service that takes a Callback interface (e.g., a Consumer) as an argument.
  • You can use doAnswer to manually invoke the onSuccess or onFailure method of that callback.

This allows you to test how your application code reacts to different callback outcomes without actual asynchronous execution.

Mocking a Callback

Here, we simulate a DataService calling a Consumer callback. doAnswer lets us trigger the callback with a specific value when fetchData is called.

import org.mockito.Mockito;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;
import java.util.function.Consumer;

// An interface representing a service that fetches data asynchronously
interface DataService {
    void fetchData(String query, Consumer<String> callback);
}

public class Main {
    public static void main(String[] args) {
        DataService mockService = Mockito.mock(DataService.class);

        // Configure the mock to simulate fetching data and calling the callback
        Mockito.doAnswer(new Answer<Void>() {
            @Override
            public Void answer(InvocationOnMock invocation) throws Throwable {
                // Get the arguments passed to fetchData
                String query = invocation.getArgument(0); // The first argument is the query
                Consumer<String> callback = invocation.getArgument(1); // The second is the callback
                
                // Simulate a successful data fetch by invoking the callback
                callback.accept("Data for '" + query + "' fetched successfully!");
                return null; // Void methods in doAnswer return null
            }
        }).when(mockService).fetchData(Mockito.anyString(), Mockito.any(Consumer.class));

        // Our application code would typically use this service
        System.out.println("Application requesting user data...");
        mockService.fetchData("user:456", result -> {
            System.out.println("Application received data: " + result);
        });
        System.out.println("Application finished request setup.");
    }
}

Deeper with Invocation

The InvocationOnMock object (or simply invocation) passed to your answer() method is a powerful context object.

  • invocation.getArguments(): Returns an array of all arguments passed to the mocked method.
  • invocation.getArgument(index): Returns a specific argument by its index.
  • invocation.getMethod(): Gives you the Method object that was called, allowing reflection.
  • invocation.getMock(): Returns the mock object itself.
  • invocation.callRealMethod(): For spies, this allows invoking the actual, real method.

Tips for Answer Use

Custom answers are powerful, but it's important to use them thoughtfully to keep your tests maintainable and clear:

  • Keep them simple: If your answer logic becomes too complex, it might indicate a design issue in your production code or an overly complicated test setup.
  • Prefer simpler stubbing: Always choose simpler stubbing methods like thenReturn(), thenThrow(), or thenCallRealMethod() if they can achieve the desired behavior.
  • Test the Answer itself: For very complex Answer implementations, consider extracting them into a separate class and writing a unit test for that class.

When to Use Answer

Here are typical situations where the Answer interface (usually via doAnswer()) truly shines:

  • Dynamic return values: When the return value depends on input arguments in a non-trivial or calculated way.
  • Side effects: Modifying arguments that are passed by reference (e.g., updating an object).
  • Callbacks: Triggering success or failure callbacks to simulate asynchronous operations.
  • Chaining operations: Simulating a sequence of interactions or state changes within a single mock call.

Custom Mock Behavior

You are testing a UserService that depends on a UserRepository. The UserRepository has a method save(User user) which typically updates the user's ID after saving it to a database.

You want to mock UserRepository to simulate this behavior without hitting a real database. Assume a User class with setId(Long id).

Recap: Dynamic Mocks

You've learned how to bring dynamic and complex behavior to your mocks using Mockito's Answer interface and the doAnswer method.

  • The Answer interface lets you define custom logic that executes when a mocked method is called.
  • The InvocationOnMock object provides essential context, including method arguments and the mock itself.
  • doAnswer is ideal for dynamic return values, simulating callbacks, or modifying arguments passed by reference.

Remember to use `Answer` judiciously, preferring simpler stubbing when possible, to keep your tests clear and focused.

Perguntas Frequentes

A aula “Respostas e callbacks personalizados” é grátis?

Sim — o texto completo de “Respostas e callbacks personalizados” é grátis para ler aqui na web. Para praticá-la interativamente (um editor de código integrado e um tutor de IA 24/7) e desbloquear o restante do curso de Testing Mastery: JUnit, Mockito & Integration Tests, atualize para CoddyKit PRO. O curso de Testing Mastery: JUnit, Mockito & Integration Tests inclui 4 aulas no total.

O que vou aprender em “Respostas e callbacks personalizados”?

Implemente lógica personalizada para chamadas de métodos simulados usando interfaces `Answer` a fim de simular comportamentos complexos. Você pratica Testing Mastery: JUnit, Mockito & Integration Tests com código prático que executa diretamente no navegador, e um tutor de IA 24/7 responde suas dúvidas enquanto trabalha na aula.

Preciso ter experiência prévia para começar Testing Mastery: JUnit, Mockito & Integration Tests?

Nenhuma experiência prévia é necessária. Testing Mastery: JUnit, Mockito & Integration Tests no CoddyKit é estruturado para alunos iniciantes até avançados, então você pode começar aqui ou desde o início e aprender no seu ritmo. Esta é a aula 1 de 4.

Quanto tempo leva a aula “Respostas e callbacks personalizados”?

A maioria das aulas CoddyKit leva cerca de 5–10 minutos. Cada uma é compacta e interativa, então você faz progresso constante e retoma exatamente de onde parou entre web e app.

Posso escrever e executar código nesta aula de Testing Mastery: JUnit, Mockito & Integration Tests?

Sim. Cada aula de Testing Mastery: JUnit, Mockito & Integration Tests inclui um editor de código integrado, então você escreve e executa código real direto no navegador e recebe feedback de IA instantaneamente — nenhuma configuração local necessária.

Todas as aulas deste curso

  1. Respostas e callbacks personalizados
  2. Simulação de métodos estáticos e construtores
  3. Boas práticas do Mockito
  4. Capturando argumentos com ArgumentCaptor
← Voltar para Testing Mastery: JUnit, Mockito & Integration Tests