如何模擬在方法內創建的對象？

Question

考慮這個

public class UserManager { 
    private final CrudService crudService; 

    @Inject 
    public UserManager(@Nonnull final CrudService crudService) { 
     this.crudService = crudService; 
    } 

    @Nonnull 
    public List<UserPresentation> getUsersByState(@Nonnull final String state) { 
     return UserPresentation.getUserPresentations(new UserQueries(crudService).getUserByState(state)); 
    } 

} 

我想模擬出

new UserQueries(crudService) 

，這樣我可以模擬出其行爲

任何想法？

Answer 1

您可以注入一個創建UserQueries的工廠。

public class UserManager { 
private final CrudService crudService; 
private final UserQueriesFactory queriesFactory; 

@Inject 
public UserManager(@Nonnull final CrudService crudService,UserQueriesFactory queriesFactory) { 
    this.crudService = crudService; 
    this.queriesFactory = queriesFactory; 
} 

@Nonnull 
public List<UserPresentation> getUsersByState(@Nonnull final String state) { 
    return UserPresentation.getUserPresentations(queriesFactory.create(crudService).getUserByState(state)); 
} 

}

雖然它可能是更好的（如果你要做到這一點）以注入CrudService進廠。

Answer 2

與PowerMock你可以模擬構造函數。見example

我不是一個IDE的權利，但會是這樣的：

UserQueries userQueries = PowerMockito.mock(UserQueries.class); 
    PowerMockito.whenNew(UserQueries.class).withArguments(Mockito.any(CrudService.class)).thenReturn(userQueries); 

您需要使用PowerMockRunner運行測試（這些註釋添加到您的測試類）：

@RunWith(PowerMockRunner.class) 
@PrepareForTest(UserQueries .class) 

如果你不能使用PowerMock，你必須注入一個工廠，因爲它說@Briggo答案。

希望它可以幫助