private void mockUpdateProduct(final Product product, Owner owner) { when(productManager.updateProduct(any(ProductData.class), eq(owner), anyBoolean())) .thenAnswer( new Answer<Product>() { @Override public Product answer(InvocationOnMock invocation) throws Throwable { Object[] args = invocation.getArguments(); ProductData pdata = (ProductData) args[0]; return (product.getId().equals(pdata.getId())) ? product : null; } }); }