Метод mock для возврата значения только при передаче какого-либо объекта * не нуль * данного класса

Question

Я использую Mockito, чтобы высмеять метод возврата даты при дате.

when(supplier.calculateDeliveryDate(any(Date.class))) 
    .thenReturn(supplierDeliveryDate); 

Однако, я только хочу, чтобы вернуть supplierDeliveryDate когда он передается ненулевым Дата объект. Когда прошло null, он должен вернуть null.

Возможно ли это? Как мне это сделать?

Answer 1

Вы можете использовать анонимный внутренний класс:

// unit test 
public OrderServiceTest { 

    // instance of class-under-test 
    private OrderService instance; 
    // stub value 
    private Date supplierDeliveryDate = new Date(); 

    // mock as an anonymous inner class 
    private Supplier supplier = new Supplier() { 
     public Date calculateDeliveryDate(Date input) { 
      if (input == null) { 
       return null; 
      } 
      else { 
       return supplierDeliveryDate; 
      } 
     } 
    }; 

    @Before 
    public void setUp() { 
     instance = new OrderService(); 

     // dependency injection 
     instance.setSupplier(supplier); 
    } 

    @Test 
    public void testOrderHappy() { 
     // SETUP 
     Date orderDate = new Date();    

     // CALL 
     Date result = instance.order(orderDate); 

     // VERIFY 
     assertTrue(supplierDeliveryDate == result); 
    } 

    @Test 
    public void testOrderNull() { 
     // SETUP 
     Date orderDate = null; 

     // CALL 
     Date result = instance.order(orderDate); 

     // VERIFY 
     assertNull(result); 
    } 
} 

Но вы действительно должны задаться вопросом, почему вам нужен этот вид поведения.

Если вы пишете четко определенный тестовый пример, вы должны точно знать, как часто и с какими аргументами вызывается ваш макет. Если это так, вы можете просто заглушить ожидаемые вызовы, а не подключать свой макет с условным поведением.

Обратите внимание, что это полезно, если ваш тест как можно более «резкий». Если разное количество вызовов достигает вашего макета, чем ожидалось, или с разными аргументами, тогда тест должен завершиться неудачно.

Answer 2

Использование ArgumentMatchers.isNull() подшивка.

when(supplier.calculateDeliveryDate(any(Date.class))) 
    .thenReturn(supplierDeliveryDate); 
when(supplier.calculateDeliveryDate(isNull())) 
    .thenReturn(null);