Sử dụng Mockito để mô phỏng biến cục bộ của phương thức

Question

Tôi có lớp học A cần thử nghiệm. Sau đây là định nghĩa của A:

public class A { 
    public void methodOne(int argument) { 
     //some operations 
     methodTwo(int argument); 
     //some operations 
    } 

    private void methodTwo(int argument) { 
     DateTime dateTime = new DateTime(); 
     //use dateTime to perform some operations 
    } 
} 

Và dựa trên giá trị dateTime một số dữ liệu là để được thao tác, lấy ra từ cơ sở dữ liệu. Đối với cơ sở dữ liệu này, các giá trị được duy trì thông qua một tệp JSON.

Điều này làm phức tạp mọi thứ. Những gì tôi cần là đặt dateTime cho một số ngày cụ thể trong khi nó đang được thử nghiệm. Có cách nào tôi có thể giả định một giá trị của biến địa phương bằng cách sử dụng mockito?

Answer 1

Bạn không thể giả lập biến cục bộ. Những gì bạn có thể làm, tuy nhiên, là trích xuất sáng tạo của nó đến một phương pháp protected và spy nó:

public class A { 
    public void methodOne(int argument) { 
    //some operations 
    methodTwo(int argument); 
    //some operations 
    } 

    private void methodTwo(int argument) { 
    DateTime dateTime = createDateTime(); 
    //use dateTime to perform some operations 
    } 

    protected DateTime createDateTime() { 
    return new DateTime(); 
    } 
} 

public class ATest { 
    @Test 
    public void testMethodOne() { 
    DateTime dt = new DateTime (/* some known parameters... */); 
    A a = Mockito.spy(new A()); 
    doReturn(dt).when(a).createDateTime(); 
    int arg = 0; // Or some meaningful value... 
    a.methodOne(arg); 
    // assert the result 
} 

Answer 2

Cách tốt nhất để đối phó với một vấn đề như vậy là để sử dụng một tiêm Clock dịch vụ, được sử dụng để có được các trường hợp mới của DateTime . Bằng cách đó, kiểm tra của bạn có thể tiêm một đồng hồ giả, trả về một DateTime cụ thể thay vì thời gian hiện tại.

Lưu ý rằng API thời gian 8 Java mới định nghĩa một Clock class, cụ thể cho mục đích đó.