有人能帮我模拟一个返回对象的静态方法吗？这个静态方法存在于最后一个类中

Question

下面的事情我需要帮助，

我必须使用PowerMock/Mockito为一个方法编写一个JUnit，该方法调用外部jar中存在的最终类的静态方法。

我需要为其编写JUnit测试的方法是：

public class SomeClass {
    private PrivateKey privateKeyFromPkcs8(String privateKeyPem) throws IOException {
        Reader reader = new StringReader(privateKeyPem);
        Section section = PemReader.readFirstSectionAndClose(reader, "PRIVATE KEY");
        if (section == null) {
            throw new IOException("Invalid PKCS8 data.");
        }
        byte[] bytes = section.getBase64DecodedBytes();
        PKCS8EncodedKeySpec keySpec = new PKCS8EncodedKeySpec(bytes);
        try {
            KeyFactory keyFactory = SecurityUtils.getRsaKeyFactory();
            PrivateKey privateKey = keyFactory.generatePrivate(keySpec);
            return privateKey;
        } catch (NoSuchAlgorithmException exception) {
        } catch (InvalidKeySpecException exception) {
        }
        throw new IOException("Unexpected exception reading PKCS data");
    }
}   

在上面的代码中PemReader是一个最终类，readFirstSectionAndClole（阅读器，“PRIVATE KEY”）是PemReader中的静态方法。

我已经尝试编写如下所示的测试，但调试时节对象（节）显示为null。可能调用的是实际的代码（PemReader.readFirstSectionAndClose（reader，“PRIVATE KEY”）），而不是模拟。

@RunWith(PowerMockRunner.class)
@PrepareForTest({SomeClass.class,PemReader.class})
public class SomeClassTest {
    @InjectMocks
    SomeClass mockSomeClass;

    @Mock
    private Reader mockReader;

    @Mock
    private Section mockSection;

    @Test
    public void testPrivateKeyFromPkcs8() throws Exception {   
        PowerMockito.mockStatic(PemReader.class);
        Mockito.when(PemReader.readFirstSectionAndClose(mockReader, "PRIVATE KEY")).thenReturn(mockSection);

        assertNotNull(mockSomeClass.privateKeyFromPkcs8(dummyPrivateKey));
    } 

 }

请帮助我使用powermockito/mockito编写Junit

Answer 1

你必须准备最后的静态类。

这是一个使用JUnit的PowerMock注释的示例：

@RunWith(PowerMockRunner.class)
@PrepareForTest({PemReader.class})
public class PemReaderTest {

    @Mock
    private Reader mockReader;

    @Mock
    private Section mockSection;

    @Test
    public void testMockingStatic() {
        PowerMockito.mockStatic(PemReader.class);

        Mockito.when(PemReader.readFirstSectionAndClose(mockReader, "PRIVATE KEY")).thenReturn(mockSection);

        Assert.assertEquals(mockSection, PemReader.readFirstSectionAndClose(mockReader, "PRIVATE KEY"));
    }
}

为了完整起见，下面是PemReader的定义：

public final class PemReader {

    public static Section readFirstSectionAndClose(Reader reader, String key) {
        return null;
    }
}

以上测试通过，版本如下：

• JUnit:4.12

更新1：根据您更新的问题。如果您只做以下更改，您的测试用例将通过（或至少在PemReader上调用。readFirstSectionAndClose将返回一些内容）：

Mockito.when(PemReader.readFirstSectionAndClose(
    Mockito.any(Reader.class), 
    Mockito.eq("PRIVATE KEY"))
).thenReturn(mockSection);

当前测试用例中此指令的版本依赖于代码传递到readFirstSectionAndClose的StringReader与测试用例提供的模拟读取器之间的相等匹配。这些值不“相等”，因此模拟调用的期望值未得到满足，并且不会返回您的模拟节。

一些不相关的注释：

• 不需要包含SomeClass。class在PrepareForTest中，您只需要在该注释中包含要模拟的类，因为SomeClass是您尝试测试的类，该类不需要模拟