在java中模拟从私有方法返回的对象

Question

我需要模拟从私有方法返回的对象。我在我的项目中与junit和mockito合作。我不能在这里粘贴实际代码，但示例代码如下所示。这里的限制是代码是遗留的，我现在无法重构

要测试的类

 public class TestService {

   public String test() {
     TestDao testDao = getDaoObject();
     int num = testDao.getData();
     if (num < 10) {
       return "hey you loose";
     } else {
       return "hey you win";
     }
   }

   private TestDao getDaoObject() {
     return new TestDao();
   }
 }

道类

public class TestDao {

  public int getData() {
    return 5;
  }
}

测试类别

 public class JUnitServiceTestExample {
   @Test
   public void test() {
     TestDao testDao = Mockito.mock(TestDao.class);
     TestService test = new TestService();
     when(testDao.getData()).thenReturn(20);
     assertEquals(test.test(), "hey you win");
   }
 }   

请帮帮忙

Answer 1

您可以稍微修改getDaoObject从Private到Package-Private，如下所示：

TestDao getDaoObject() {
   return new TestDao();
}

然后使用Mockito。spy（new TestService（））来存根getDaoObject（）并返回模拟的testDao。

 public class JUnitServiceTestExample {
   @Test
   public void test() {
     TestDao testDao = Mockito.mock(TestDao.class);
     TestService test = Mockito.spy(new TestService());
     when(testDao.getData()).thenReturn(20);
     doReturn(testDao).when(test).getDaoObject();

     String result = test.test();

     assertEquals("hey you win", result);
   }
 }   

给你一个提示：assertEquals的正确用法是assertEquals（预期的，实际的）而不是assertEquals（实际的，预期的）