Argument aggregation in JUnit - Test Execution Trace

Test Overview

This test demonstrates how to use argument aggregation in JUnit 5 to combine multiple arguments into a single object for parameterized tests. It verifies that the aggregated object correctly holds the combined values.

Test Code - JUnit 5

JUnit

import static org.junit.jupiter.api.Assertions.assertEquals;
import org.junit.jupiter.params.ParameterizedTest;
import org.junit.jupiter.params.aggregator.AggregateWith;
import org.junit.jupiter.params.aggregator.ArgumentsAccessor;
import org.junit.jupiter.params.provider.CsvSource;

class User {
    String name;
    int age;

    User(String name, int age) {
        this.name = name;
        this.age = age;
    }
}

class UserAggregator implements org.junit.jupiter.params.aggregator.ArgumentsAggregator {
    @Override
    public User aggregateArguments(ArgumentsAccessor accessor, java.lang.reflect.Parameter parameter) {
        return new User(accessor.getString(0), accessor.getInteger(1));
    }
}

public class ArgumentAggregationTest {

    @ParameterizedTest
    @CsvSource({"Alice,30", "Bob,25"})
    void testUserAggregation(@AggregateWith(UserAggregator.class) User user) {
        assertEquals("Alice", user.name, "First user's name should be Alice");
        assertEquals(30, user.age, "First user's age should be 30");
    }

    @ParameterizedTest
    @CsvSource({"Alice,30", "Bob,25"})
    void testUserAggregationDynamic(@AggregateWith(UserAggregator.class) User user) {
        if (user.name.equals("Alice")) {
            assertEquals(30, user.age);
        } else if (user.name.equals("Bob")) {
            assertEquals(25, user.age);
        } else {
            throw new AssertionError("Unexpected user name: " + user.name);
        }
    }
}

Execution Trace - 5 Steps

Step	Action	System State	Assertion	Result
1	Test runner starts and loads ArgumentAggregationTest class	JUnit test environment initialized	—	PASS
2	JUnit invokes testUserAggregation with first CSV input: "Alice,30"	Test method called with aggregated User object (name=Alice, age=30)	assertEquals("Alice", user.name) and assertEquals(30, user.age)	PASS
3	JUnit invokes testUserAggregation with second CSV input: "Bob,25"	Test method called with aggregated User object (name=Bob, age=25)	assertEquals("Alice", user.name) fails because user.name is "Bob"	FAIL
4	JUnit invokes testUserAggregationDynamic with first CSV input: "Alice,30"	Test method called with aggregated User object (name=Alice, age=30)	assertEquals(30, user.age) for user.name == "Alice"	PASS
5	JUnit invokes testUserAggregationDynamic with second CSV input: "Bob,25"	Test method called with aggregated User object (name=Bob, age=25)	assertEquals(25, user.age) for user.name == "Bob"	PASS

Failure Scenario

Failing Condition: Using fixed assertions for specific input in testUserAggregation causes failure on second input

Execution Trace Quiz - 3 Questions

What does the @AggregateWith annotation do in this test?

A It runs the test multiple times with different CSV inputs

B It asserts that the User object is not null

C It combines multiple CSV values into a single User object

D It skips the test if the User object is invalid

Key Result

When using argument aggregation in parameterized tests, write assertions that adapt to each input value instead of fixed expected values to avoid false failures.

Practice

1. What is the main purpose of using @ArgumentsAggregator in JUnit parameterized tests?

easy

A. To combine multiple method arguments into a single object for easier testing

B. To run tests in parallel threads

C. To generate random test data automatically

D. To skip certain test cases during execution

Solution

1. Step 1: Understand the role of @ArgumentsAggregator

   This annotation is used to create a custom aggregator that combines multiple input arguments into one object.

2. Step 2: Recognize the benefit in parameterized tests

   Aggregating arguments simplifies test method signatures and improves readability by passing one object instead of many parameters.

3. Final Answer:

   To combine multiple method arguments into a single object for easier testing -> Option A

4. Quick Check:

   @ArgumentsAggregator = combine inputs [OK]

Hint: Remember: Aggregator groups inputs into one object [OK]

Common Mistakes:

• Confusing @ArgumentsAggregator with parallel test execution
• Thinking it generates test data automatically
• Assuming it skips tests

2. Which of the following is the correct way to apply a custom aggregator in a JUnit 5 parameterized test method parameter?

easy

A. @AggregateWith(CustomAggregator.class) MyObject obj

B. @ArgumentsSource(CustomAggregator.class) MyObject obj

C. @AggregateArguments(CustomAggregator.class) MyObject obj

D. @ArgumentAggregator(CustomAggregator.class) MyObject obj

Solution

1. Step 1: Recall the correct annotation for applying aggregators

   The annotation to apply a custom aggregator on a parameter is @AggregateWith.

2. Step 2: Match the syntax

   The correct syntax is @AggregateWith(CustomAggregator.class) followed by the parameter type and name.

3. Final Answer:

   @AggregateWith(CustomAggregator.class) MyObject obj -> Option A

4. Quick Check:

   @AggregateWith = apply aggregator [OK]

Hint: Use @AggregateWith to link aggregator class to parameter [OK]

Common Mistakes:

• Using @ArgumentsSource instead of @AggregateWith
• Misspelling the annotation name
• Confusing @AggregateWith with @ArgumentAggregator

3. Given the following aggregator implementation and test method, what will be the output of the assertion?

public class PersonAggregator implements ArgumentsAggregator {
  @Override
  public Person aggregateArguments(ArgumentsAccessor accessor, ParameterContext context) {
    return new Person(accessor.getString(0), accessor.getInteger(1));
  }
}

@ParameterizedTest
@CsvSource({"Alice,30", "Bob,25"})
void testPerson(@AggregateWith(PersonAggregator.class) Person person) {
  assertTrue(person.getAge() > 20);
}

medium

A. Test fails for Alice because age is 30

B. Test fails for Bob because age is 25

C. Test passes for both Alice and Bob

D. Test fails for both because age is not checked correctly

Solution

1. Step 1: Understand the aggregator behavior

   The aggregator creates a Person object with name and age from CSV input.

2. Step 2: Evaluate the assertion for each input

   Both Alice (30) and Bob (25) have ages greater than 20, so assertTrue(person.getAge() > 20) passes for both.

3. Final Answer:

   Test passes for both Alice and Bob -> Option C

4. Quick Check:

   Age > 20 for both inputs = pass [OK]

Hint: Check each input value against assertion condition [OK]

Common Mistakes:

• Assuming 25 is less than or equal to 20
• Confusing assertion logic
• Ignoring aggregator mapping

4. Identify the error in this aggregator implementation:

public class ProductAggregator implements ArgumentsAggregator {
  @Override
  public Product aggregateArguments(ArgumentsAccessor accessor, ParameterContext context) {
    String name = accessor.getString(0);
    int price = accessor.getInteger(1);
    return new Product(name, price);
  }
}

@ParameterizedTest
@CsvSource({"Laptop,1000", "Phone,abc"})
void testProduct(@AggregateWith(ProductAggregator.class) Product product) {
  assertNotNull(product);
}

medium

A. The aggregator method signature is incorrect

B. The Product constructor parameters are in wrong order

C. The @AggregateWith annotation is missing

D. The second CSV input 'Phone,abc' causes a runtime error due to invalid integer

Solution

1. Step 1: Analyze CSV input and aggregator code

   The second CSV line has 'abc' where an integer price is expected.

2. Step 2: Understand the impact on aggregation

   Calling accessor.getInteger(1) on 'abc' will throw a runtime exception (NumberFormatException).

3. Final Answer:

   The second CSV input 'Phone,abc' causes a runtime error due to invalid integer -> Option D

4. Quick Check:

   Invalid integer input = runtime error [OK]

Hint: Check CSV data types match aggregator expectations [OK]

Common Mistakes:

• Ignoring invalid CSV data causing exceptions
• Assuming aggregator method signature is wrong
• Thinking annotation is missing

5. You want to test a method that takes a Person object with fields name and age. You have CSV data with columns name, age, and city. How can you write a custom aggregator to ignore the city column and only aggregate name and age into Person?

hard

A. Use @AggregateWith but do not implement ArgumentsAggregator interface

B. Use ArgumentsAccessor to get only the first two columns and create Person ignoring city

C. Use @CsvSource with only two columns to avoid city

D. Use ArgumentsAccessor to get all three columns and pass city to Person constructor

Solution

1. Step 1: Understand the CSV data and Person constructor

   CSV has three columns but Person only needs name and age.

2. Step 2: Implement aggregator to pick only needed columns

   Use ArgumentsAccessor.getString(0) and getInteger(1) to get name and age, ignoring the third city column.

3. Final Answer:

   Use ArgumentsAccessor to get only the first two columns and create Person ignoring city -> Option B

4. Quick Check:

   Pick needed columns only in aggregator [OK]

Hint: Pick only needed columns with ArgumentsAccessor [OK]

Common Mistakes:

• Trying to pass extra columns to Person constructor
• Changing CSV source instead of aggregator logic
• Not implementing ArgumentsAggregator interface