Mockito Cách chỉ chế nhạo lệnh gọi của một phương thức của lớp cha

Tôi đang sử dụng Mockito trong một số thử nghiệm.

Tôi có các lớp sau:

class BaseService {  
    public void save() {...}  
}

public Childservice extends BaseService {  
    public void save(){  
        //some code  
        super.save();
    }  
}   

Tôi chỉ muốn chế nhạo cuộc gọi thứ hai ( super.save) của ChildService. Lần gọi đầu tiên phải gọi phương thức thực. Có cách nào làm được việc này không?

Không, Mockito không hỗ trợ điều này.

Đây có thể không phải là câu trả lời bạn đang tìm kiếm, nhưng những gì bạn thấy là dấu hiệu của việc không áp dụng nguyên tắc thiết kế:

Thành phần ủng hộ hơn thừa kế

Nếu bạn trích xuất một chiến lược thay vì mở rộng một siêu lớp, vấn đề sẽ không còn nữa.

Tuy nhiên, nếu bạn không được phép thay đổi mã, nhưng bạn vẫn phải kiểm tra nó, và theo cách khó xử này, vẫn có hy vọng. Với một số công cụ AOP (ví dụ: AspectJ), bạn có thể đan mã vào phương thức siêu lớp và tránh hoàn toàn việc thực thi nó (yuck). Điều này không hiệu quả nếu bạn đang sử dụng proxy, bạn phải sử dụng sửa đổi mã bytecode (dệt thời gian tải hoặc dệt thời gian biên dịch). Có những khuôn khổ chế nhạo cũng hỗ trợ loại thủ thuật này, như PowerMock và PowerMockito.

Tôi khuyên bạn nên tiến hành tái cấu trúc, nhưng nếu đó không phải là một lựa chọn, bạn đang ở trong một số trò vui hack nghiêm trọng.

Nếu bạn thực sự không có lựa chọn để tái cấu trúc, bạn có thể mô phỏng / khai thác mọi thứ trong lệnh gọi siêu phương thức, ví dụ:

    class BaseService {

        public void validate(){
            fail(" I must not be called");
        }

        public void save(){
            //Save method of super will still be called.
            validate();
        }
    }

    class ChildService extends BaseService{

        public void load(){}

        public void save(){
            super.save();
            load();
        }
    }

    @Test
    public void testSave() {
        ChildService classToTest = Mockito.spy(new ChildService());

        // Prevent/stub logic in super.save()
        Mockito.doNothing().when((BaseService)classToTest).validate();

        // When
        classToTest.save();

        // Then
        verify(classToTest).load();
    }

Hãy xem xét việc cấu trúc lại mã từ phương thức ChildService.save () thành phương thức khác và kiểm tra phương thức mới đó thay vì thử nghiệm ChildService.save (), bằng cách này bạn sẽ tránh được việc gọi siêu phương thức không cần thiết.

Thí dụ:

class BaseService {  
    public void save() {...}  
}

public Childservice extends BaseService {  
    public void save(){  
        newMethod();    
        super.save();
    }
    public void newMethod(){
       //some codes
    }
} 

tạo một phương thức được bảo vệ bằng gói (giả sử lớp thử nghiệm trong cùng một gói) trong lớp con gọi phương thức siêu lớp và sau đó gọi phương thức đó trong phương thức lớp con bị ghi đè của bạn. sau đó bạn có thể đặt kỳ vọng vào phương pháp này trong thử nghiệm của mình thông qua việc sử dụng mẫu gián điệp. không đẹp nhưng chắc chắn tốt hơn là phải đối phó với tất cả các kỳ vọng đặt ra cho phương pháp siêu việt trong bài kiểm tra của bạn

Ngay cả khi tôi hoàn toàn đồng ý với phản hồi của iwein (

thành phần ưu tiên hơn thừa kế

), tôi thừa nhận rằng có một số lần kế thừa dường như chỉ là tự nhiên và tôi không cảm thấy bị phá vỡ hoặc cấu trúc lại nó chỉ vì lợi ích của một bài kiểm tra đơn vị.

Vì vậy, gợi ý của tôi:

/**
 * BaseService is now an asbtract class encapsulating 
 * some common logic callable by child implementations
 */
abstract class BaseService {  
    protected void commonSave() {
        // Put your common work here
    }

    abstract void save();
}

public ChildService extends BaseService {  
    public void save() {
        // Put your child specific work here
        // ...

        this.commonSave();
    }  
}

Và sau đó, trong bài kiểm tra đơn vị:

    ChildService childSrv = Mockito.mock(ChildService.class, Mockito.CALLS_REAL_METHODS);

    Mockito.doAnswer(new Answer<Void>() {
        @Override
        public Boolean answer(InvocationOnMock invocation)
                throws Throwable {
            // Put your mocked behavior of BaseService.commonSave() here
            return null;
        }
    }).when(childSrv).commonSave();

    childSrv.save();

    Mockito.verify(childSrv, Mockito.times(1)).commonSave();

    // Put any other assertions to check child specific work is done

Lý do là lớp cơ sở của bạn không phải là public-ed, sau đó Mockito không thể chặn nó do khả năng hiển thị, nếu bạn thay đổi lớp cơ sở thành public hoặc @Override trong lớp con (dưới dạng công khai), thì Mockito có thể mô phỏng nó một cách chính xác.

public class BaseService{
  public boolean foo(){
    return true;
  }
}

public ChildService extends BaseService{
}

@Test
@Mock ChildService childService;
public void testSave() {
  Mockito.when(childService.foo()).thenReturn(false);

  // When
  assertFalse(childService.foo());
}

Có thể tùy chọn dễ dàng nhất nếu kế thừa có ý nghĩa là tạo một phương thức mới (gói riêng tư ??) để gọi super (cho phép gọi nó là superFindall), theo dõi phiên bản thực và sau đó giả lập phương thức superFindAll () theo cách bạn muốn mô phỏng lớp cha một. Nó không phải là giải pháp hoàn hảo về mức độ phủ sóng và khả năng hiển thị nhưng nó sẽ hoạt động tốt và rất dễ áp ​​dụng.

 public Childservice extends BaseService {
    public void save(){
        //some code
        superSave();
    }

    void superSave(){
        super.save();
    }
}