模拟私有方法[重复]

Posted 2023-03-22

【中文标题】模拟私有方法[重复]

【英文标题】：Mock a private method [duplicate]

【发布时间】：2018-07-17 07:16:58

【问题描述】：

我有一个私有的方法。现在，我不想在对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;
    

【问题讨论】：

如果你可以选择 mocking 框架，你可以试试 jmockit - 它可以模拟私有方法

【参考方案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 或任何其他库的极少数情况下将此方法视为一种解决方法。更好的是，尝试进行重构并在代码中引入可测试性

【参考方案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”作为返回值？如果不是 - 考虑使用原始布尔类型；

【参考方案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);