模拟内部类的局部变量-使用嘲笑/ powermock

Question

我正在尝试模拟一个innerclass（Class2.class）的局部变量，但是它不起作用。 我在下面共享我的代码，请帮助我运行第二个testCase。

包含Class2作为内部类的Class1.java

public final class Class1 {

    Class2 class2 = new Class2();

    public String echoString(String s) {
        return class2.echoString2(s);
    }

    public class Class2 {

        public String echoString2(String s) {
            Class3 class3 = new Class3(null);
            System.out.println("######## Class3 = " + class3);
            return class3.getResult(s);
        }
    }
}

Class3.java

public class Class3 {
    private String data;

    public Class3(String data){
        this.data=data;
    }

    public String getResult(String s) {
        return data;
    }
}

以下是我要运行第二个测试用例的TestNG测试用例-请帮助

@PrepareForTest({ Class1.class, Class2.class, Class3.class })
public class MockTest extends PowerMockTestCase {
    String testInput = "input";
    String mockedResult = "Mocked - " + testInput;

    @InjectMocks
    private Class1 tested;

    @Mock
    Class2 tested2;

    @BeforeMethod
    public void InjectingMocks() {
        MockitoAnnotations.initMocks(this); // Creates a fresh set of mocks
    }

    @Test
    public void mockTest() {

        PowerMockito.doReturn(mockedResult).when(tested2)
                .echoString2(testInput);

        Assert.assertEquals(tested.echoString(testInput), mockedResult);
    }

    @Test
    public void mockTest2() throws Exception {
        Class3 c3 = PowerMockito.mock(Class3.class);

        PowerMockito.whenNew(Class3.class).withAnyArguments().thenReturn(c3);

        PowerMockito.doReturn(mockedResult).when(c3).getResult(testInput);

        Assert.assertEquals(tested.echoString(testInput), mockedResult);
    }
}

显示错误

PASSED: mockTest
FAILED: mockTest2
java.lang.AssertionError: expected [Mocked - input] but found [null]

请帮忙-我如何在不对实际代码进行任何重大更改的情况下成功运行它。

Answer 1

怎么样：

@Test
    public void mockTest2() throws Exception {
        Class3 c3 = PowerMockito.mock(Class3.class);

        PowerMockito.whenNew(Class3.class).withAnyArguments().thenReturn(c3);

        PowerMockito.doReturn(mockedResult).when(c3).getResult(testInput);

        tested = new Class1();
        Assert.assertEquals(tested.echoString(testInput), mockedResult);
    }

我可能是错的，但我认为问题可能是在您使用PowerMockito模拟Class3构造函数之前调用了新的Class3（null）。