JUnit - 如何以不同的参数以相同的方法测试重复指令？

Question

我有一个更新方法，但我在进行单元测试时遇到了问题。在该方法中，我需要验证要更改的对象是否存在于数据库中，并且我还需要检查用户插入的新数据是否会导致记录重复。为此，我在两个不同的时间使用findById() 在数据库中进行搜索：

    public void update(Form form, Long storeCode, Long productCode, Long purchaseQuantity) {

    var id = new PrimaryKeyBuilder(DiscountPK.builder().build()).build(storeCode,productCode, purchaseQuantity);
    var targetToUpdate = repository.findById(id).orElseThrow(NotFoundException::new);

    var dataToInsert = SerializationUtils.clone(id);
    dataToInsert.setPurchaseQuantity(form.getPurchaseQuantity());

    var newPk = repository.findById(dataToInsert);
    throwExceptionIf(newPk.isPresent(), new DuplicatedPkException());

    targetToUpdate.getId().setPurchaseQuantity(form.getPurchaseQuantity());
    targetToUpdate.setDiscountPercentage(form.getDiscountPercentage());

    repository.save(targetToUpdate);

}

问题是：我无法在单元测试中区分这两个findById() 指令。不是成功通过它，而是在我的第一次验证中抛出NotFoundException。就好像第一个 given() 语句被忽略，只考虑第二个语句

    @Test
    public void update_successfully() {

        var targetToUpdate = ObjectFactory.createMain();
        var form = FormFactory.createUpdateForm();
        var id = ObjectFactory.createFirstAux();
        var dataToInsert = ObjectFactory.createSecondAux();

        given(repository.findById(id)).willReturn(Optional.of(targetToUpdate));
        given(repository.findById(dataToInsert)).willReturn(Optional.empty());

        given(repository.save(any())).willReturn(targetToUpdate);

        service.update(form, STORE_CODE, PRODUCT_CODE, PURCHASE_QUANTITY);
        verify(repository).save(targetToUpdate);
    }

奖励代码：构建传入findById()的对象的类

 public class PrimaryKeyBuilder {

    private final DiscountPK id;

    public PrimaryKeyBuilder(DiscountPK id) {
        this.id = id;
    }

    public DiscountPK build(Long storeCode, Long productCode, Long purchaseQuantity) {

        id.setPurchaseBoxQuantity(purchaseBoxQuantity);
        id.setProduct(Product.builder().id(ProductPK
                .builder().productCode(productCode).store(Store.builder().code(storeCode).build()).build()).build());

        return id;
    }

}

Answer 1

使用given 或when 定义模拟行为时，您需要提供与被测试代码将提供的参数相匹配的参数。 Mockito documentation 解释说：“Mockito 以自然 java 风格验证参数值：通过使用 equals() 方法。”

如果DiscountPK 类没有覆盖equals 方法，则将使用java.lang.Object 中的默认实现。这会比较对象身份，并且仅当两个对象都是同一个实例时才返回 true。

如果您可以修改DiscountPK 类，最直接的解决方案是覆盖该类中的equals 和hashCode 方法，以根据其包含的值定义相等性。假设DiscountPK 类仅包含两个字段，分别名为purchaseBoxQuantity 和product，您可以使用java.util.Objects 类来定义这些字段：

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;
        DiscountPK that = (DiscountPK) o;
        return Objects.equals(purchaseBoxQuantity, that.purchaseBoxQuantity) && product.equals(that.product);
    }

    @Override
    public int hashCode() {
        return Objects.hash(purchaseBoxQuantity, product);
    }

这还假设Product 类定义了equals 和hashCode。

请注意，尽管 Mockito 仅使用 equals 方法，而不是 hashCode，但始终建议在覆盖 equals 时覆盖 hashCode，以确保相等的对象也具有相等的哈希码。

我还建议使主键类不可变。这使得它们更容易推理，因为实体主键的内容永远不会改变。它还确保hashCode 的返回值在对象的生命周期内不会发生变化，这可能会导致错误。

Answer 2

如果您在模拟存储库，那么在使用 myRepo.save(myObj); 时它们实际上不会保存更改

不过，您可以使用 Mockito 来模拟它：when(X).thenReturn(Y)。

例如，

import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;

import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;

@ExtendWith(MockitoExtension.class)
class MyServiceTest {
    
    @Mock private MyRepo myRepo;
    @InjectMocks private MyService myService;

    ...

    @Test
    void myTest() {
        MyClass myObj = new MyClass();
        Mockito.when(myRepo.findById(myObj.getId())
          .thenReturn(Optional.of(myObj));
        // perform desired test
    }
}