Có thể thêm máy đánh chặn vào tất cả các câu trả lời khi sử dụng Mockito không?

Question

Giả sử tôi có chú thích xác thực trên phương thức Giao diện của mình để xác thực đối số đầu vào và giá trị trả về. Có thể tại thời điểm này (V 1.9.5) để yêu cầu Mockito gọi trình xác thực này trong quá trình triệu gọi không?

Nền tảng sẽ ngăn không cho các nhà phát triển viết các bài kiểm tra không thực tế bằng cách chế nhạo giao diện đã cho theo cách vi phạm trình xác thực được chỉ định.

Vì vậy, những gì tôi muốn là để đăng ký một cái gì đó giống như

class MyAnswerInterceptor<T> implements AnswerInterceptor<T> { 
    @Override 
    public Answer<T> intercept(final Answer<T> answer) { 
     return new Answer<T>() { 
      @Override 
      public T answer(InvocationOnMock invocation) throws Throwable { 
       validateArguments(invocation); 
       T result = answer.answer(invocation); 
       validateReturnValue(result); 
       return result; 
      } 
     } 
    } 
} 

được kêu gọi tất cả các câu trả lời của một mô hình nhất định. Điều này có thể thực hiện được không? Tôi đã xem xét mã, cũng để kiểm tra xem tôi có thể hack vào một lúc nào đó (thậm chí sử dụng sự phản chiếu hay tương tự), nhưng có vẻ như do sự vướng víu về tạo và tạo logic, khó có thể đạt được những gì tôi muốn (ví dụ công cụ như MockHandler mockHandler = new MockHandlerFactory().create(settings); làm cho nó không thể treo vào và đặt công cụ tùy chỉnh trên đầu mà không cần vá và triển khai toàn bộ sự việc ...)

Bất kỳ cái nhìn sâu sắc sẽ được đánh giá cao :-)

Answer 1

Bạn có thể đạt được điều đó bằng cách tạo ra một tùy chỉnh MockMaker.

MockMaker là một điểm mở rộng mà làm cho nó có thể sử dụng proxy tùy chỉnh năng động và tránh sử dụng mặc định cglib/asm/objenesis thực hiện

đại biểu thực hiện tùy chỉnh của chúng tôi tất cả những thứ phức tạp để mặc định MockMaker: CglibMockMaker. Nó "trang trí" chỉ bằng phương pháp createMock bằng cách đăng ký tham số settings một InvocationListener. Người nghe này sẽ được thông báo khi một số invocation được thực hiện cho phép sử dụng để gọi validateArguments và validateReturnValue.

import org.mockito.internal.creation.CglibMockMaker; 
import org.mockito.invocation.Invocation; 
import org.mockito.invocation.MockHandler; 
import org.mockito.listeners.InvocationListener; 
import org.mockito.listeners.MethodInvocationReport; 
import org.mockito.mock.MockCreationSettings; 
import org.mockito.plugins.MockMaker; 

public class ValidationMockMaker implements MockMaker { 
    private final MockMaker delegate = new CglibMockMaker(); 

    public ValidationMockMaker() { 
    } 
    @Override 
    public <T> T createMock(MockCreationSettings<T> settings, MockHandler handler) { 
     settings.getInvocationListeners().add(new InvocationListener() { 

      @Override 
      public void reportInvocation(MethodInvocationReport methodInvocationReport) { 
       Invocation invocation = (Invocation) methodInvocationReport.getInvocation(); 
       validateArguments(invocation.getArguments()); 
       validateReturnValue(methodInvocationReport.getReturnedValue()); 
      } 
     }); 
     return delegate.createMock(settings, handler); 
    } 

    @Override 
    public MockHandler getHandler(Object mock) { 
     return delegate.getHandler(mock); 
    } 

    @Override 
    public void resetMock(Object mock, MockHandler newHandler, MockCreationSettings settings) { 
     delegate.resetMock(mock, newHandler, settings); 
    } 

    protected void validateArguments(Object... arguments) { 
//  Arrays.stream(arguments).forEach(Objects::requireNonNull); 
    } 

    private void validateReturnValue(Object result) { 
//  Objects.requireNonNull(result); 
    } 
} 

---

Cuối cùng nhưng không kém, chúng ta cần phải nói đến Mockito sử dụng thực hiện của chúng tôi. Điều này có thể bằng cách thêm một tập tin

mockito-extensions/org.mockito.plugins.MockMaker 

chứa tên lớp MockMaker của chúng tôi:

ValidationMockMaker 

Xem Using the extension point phần trong javadoc.