模拟私有方法[重复]

Question

我有一个私有的方法。现在，我不想在对execute() 方法进行单元测试时调用这个私有方法。我已经尝试过使用 PowerMockito 和所有方法，但是对于所有类型的模拟，它仍然会进入私有方法。

请提供可行的测试用例。会很感激。

@Component 
public class Employee implements SuperClass {

    @Autowired
    private FileTraverse fileTraverse;

    @Override 
    public void execute() throws Exception {
        List<String> traverse = fileTraverse.getFiles();
        Boolean t = isFileTraversed(traverse);
    } 

    private Boolean isFileTraversed(List<String> param1) {
        Boolean flag;
        //do some DB operation and return flag;
    }
}

Answer 1

@glytching 是对的。最好的变体是在新服务/组件中提取方法并为其创建模拟。在这种情况下，你的代码是可测试的，你可以重复使用这个组件......

但是，如果您对此方法只有一个用例，并且您不想仅为一种方法（辅助方法）创建服务/组件，则可以将方法可见性级别从私有更改为受保护或包-默认。在这种情况下，您可以在子类中重写此方法以进行测试并使用此子类。你应该做什么： 为您要测试的类创建一个子类，并使用该子类的实例而不是目标类。

--service that you have and need to test one
public class MainService {
    @Autowired
    private SecondService secondService;

    public Object getResultFromMainService(){
        return getResultFromMainServiceFromPrivate();
    }

    --here I changed 'private' into 'default-package' 
    Object getResultFromMainServiceFromPrivate(){
        return secondService.getResult();
    }
}

_

@RunWith(SpringRunner.class)
@ContextConfiguration(classes = ServiceOverrideTestConfiguration.class)
public class MainServiceTest {
    @Autowired
    @Qualifier("subMainService") // or add @Primary and don't use Qualifier
    private MainService service;
    @Autowired
    private SecondService secondService;

    @Test
    public void test(){
        Object result = service.getResultFromMainService();
        --here, method getResultFromMainService call inside the overrided 
            method that we can change
        assertNotNull(result);
    }    
}

@ContextConfiguration
@Import(ApplicationConfigure.class)
class ServiceOverrideTestConfiguration {

    @Bean("subMainService")
    // or add @Primary and don't use Qualifier
    MainService mainServiceSubBean(){
        return new MainServiceUnderTest();
    }
}

class MainServiceUnderTest extends MainService{    
    @Override
    Object getResultFromMainServiceFromPrivate(){
        return "SOME DEFAULT";
    }    
}

！请仅在需要模拟/存根某些方法并且不能使用 PowerMock 或任何其他库的极少数情况下将此方法视为一种解决方法。更好的是，尝试进行重构并在代码中引入可测试性

Answer 2

不要模拟私有方法。

请参阅以下建议：

@Component 
public class Employee implements SuperClass {

    @Autowired
    private FileTraverse fileTraverse;

    @Override 
    public void execute() throws Exception {
        List<String> traverse = fileTraverse.getFiles();
        Boolean t = isFileTraversed(traverse);
    } 

    private Boolean isFileTraversed(List<String> param1) {
        Boolean flag;
        //do some DB operation and return flag;
    }
}

所以在 isFileTraversed 里面 - 你将有一个 DB 操作。此操作可能会通过 DAO/Repository 对象执行。

所以你的代码可能看起来像：

@Component 
public class Employee implements SuperClass {

    @Autowired
    private FileTraverse fileTraverse;

    @Autowired
    private DatabaseAccessDao dbAccess;

    @Override 
    public void execute() throws Exception {
        List<String> traverse = fileTraverse.getFiles();
        Boolean t = isFileTraversed(traverse);
    } 

    @Override
    private Boolean isFileTraversed(List<String> param1) {
        Boolean flag;
        flag = dbAccess.checkFileTraversed(param1);
        return flag;
    }
}

您需要做的是模拟 DatabaseAccessDao 类上的公共 checkFileTraversed() 方法。

1) 不要在字段上使用 @Autowire - 更喜欢构造函数注入。

2) 您确定要返回布尔值吗？是否允许“null”作为返回值？如果不是 - 考虑使用原始布尔类型；

Answer 3

其他人都是对的。你应该尽量避免模拟私有方法。如果您真的需要模拟它，只需删除 private 以将其置于默认范围内。

但是

为了完整起见，您确实可以使用 PowerMock。这是一个使用 PowerMock 和 EasyMock 的示例。

public class Employee {

  public void execute() {
    // If our mock is working, isFileTraversed will return false
    assertThat(isFileTraversed(Collections.emptyList())).isFalse();
  }

  private Boolean isFileTraversed(List<String> param1) {
    return true;
  }

}

@RunWith(PowerMockRunner.class)
@PrepareForTest(Employee.class)
public class EmployeeTest {

  @Test
  public void execute() throws Exception {
    Employee employee = PowerMock.createPartialMockForAllMethodsExcept(Employee.class, "execute");

    PowerMock.expectPrivate(employee, "isFileTraversed", Collections.emptyList()).andReturn(false);
    PowerMock.replay(employee);

    employee.execute();

    PowerMock.verify(employee);
  }

}