模拟无法模拟方法调用

Question

我试图在测试中模拟一个调用，但我得到了一个错误，因为它调用了真正的方法，而不是模拟它。

这是我的方法

@Value("${omega.aws.nonprod-profile}")
private String nonProdProfile;

@Autowired
AwsService awsService;

public List<SecurityGroup> getAllSecurityGroups() {          
    AmazonEC2 ec2 = configSetter();     
    return awsService.getAllSecurityGroups(ec2);  
}

protected AmazonEC2 configSetter() {

    ProfileCredentialsProvider credentials = new ProfileCredentialsProvider(nonProdProfile);
    ClientConfiguration clientCfg = new ClientConfiguration();
    clientCfg.setProxyHost(this.proxyHost);
    clientCfg.setProxyPort(Integer.valueOf(proxyPort)); 
    clientCfg.setProtocol(Protocol.HTTPS);
    return new AmazonEC2Client(credentials, clientCfg);       

}

这是我的测试课

@InjectMocks
    private AwsConfigurationLocal subject;

    @Mock
    private AwsService awsService;

    @Test
    public void TestgetAllSecurityGroups() throws Exception {

        ec2 = Mockito.mock(AmazonEC2Client.class);
        securityGroup = new SecurityGroup();
        List<SecurityGroup> result = Collections.singletonList(securityGroup);
        Mockito.when(awsService.getAllSecurityGroups(ec2)).thenReturn(result);  
        List<SecurityGroup> actual = subject.getAllSecurityGroups();
        assertThat(actual, CoreMatchers.is(equals(result)));   
    }

测试实际上调用了受保护的方法config Setter，并在设置代理时失败。帮助我理解我在这里做错了什么。

Answer 1

尝试使用powerMockito返回AmazonEC2

@RunWith(PowerMockRunner.class)
@PrepareForTest({AwsConfigurationLocal.class})//assuming this is your test class

@InjectMocks
private AwsConfigurationLocal subject=PowerMockito.spy(AwsConfigurationLocal.class);//or try Mockito.spy instead, whichever works

@Test
    public void TestgetAllSecurityGroups() throws Exception {

        ec2 = Mockito.mock(AmazonEC2Client.class);
        PowerMockito.doReturn(ec2).when(subject).configSetter().withNoArguments();
        //your code
    }

Answer 2

主题。getAllSecurityGroups（）调用realconfigSetter（）返回realAmazonEC2，然后将其传递给awsService。GetAllSecurityGroup（ec2） 。该参数与mockec2不匹配，因此默认的mock实现将作为实际的返回（我猜它是空的）。

所以问题是：没有任何东西会阻止调用configSetter（）的真正实现。如果你用@Spy注释主题，然后

Mockito.when(subject.configSetter()).then(ec2);

它应该像预期的那样工作。

也就是说，有很多设置只是为了检查委托的简单调用。这是由于AwsConfigurationlocal中的两个职责混合在一起——创建Amazon EC2Client并提供getAllSecurityGroup（）。如果您将前者移动到单独的类中（比如Amazon EC2ClientFactor），那么一切都应该就位。