尝试使用 PowerMockito 模拟 ProcessBuilder 的构造函数时出错

Question

我正在尝试模拟 ProcessBuilder 的构造函数。问题是当调用构造函数时它返回 null。

班级代码：

 public static void enable() throws IOException, InterruptedException {


        logger.info("Enable NTP server...");

        String ntpAddress = AppConfig.getInstance().getString(AppConfig.NTP_SERVER, "");
        AppConfig.getInstance().getBoolean(AppConfig.NTP_ENABLED, true);
        String enableNtp = "chkconfig ntpd on " + SEPARATOR + " service ntpd stop " + SEPARATOR + " ntpdate " + ntpAddress + " " + SEPARATOR + " service ntpd start";

        String[] commandArr = {"bash", "-c", enableNtp};

        ProcessBuilder pb = new ProcessBuilder(commandArr);
        pb.redirectErrorStream(true);
        Process proc = pb.start();
        try (BufferedReader in = new BufferedReader(new InputStreamReader(
                proc.getInputStream()))) {

            String line;
            while ((line = in.readLine()) != null) {
                logger.info(line);
            }
        } catch (IOException ex) {
            logger.log(Level.SEVERE, "Error while trying to enable NTP Server");
        }

        proc.waitFor();
        proc.destroy();
        logger.info("NTP server has been enabled");


    }

测试代码：

@RunWith(PowerMockRunner.class)
@PrepareForTest({NtpServerUtil.class, ProcessBuilder.class})
public class NtpServerUtilTest extends AbstractDbTest {

    @Test
    public void testEnableNtp() throws Exception {
        ProcessBuilder pb = PowerMockito.mock(ProcessBuilder.class);
        PowerMockito.whenNew(ProcessBuilder.class).withAnyArguments().thenReturn(pb);

        NtpServerUtil.enable();
        PowerMockito.verifyNew(ProcessBuilder.class).withArguments(Matchers.anyString());
    }

}

因此，当它使用 new ProcessBuilder(command) 时，结果为 null。之后，当调用 processBuilder.start() 时，会抛出异常。我尝试了一些模拟该构造函数的方法。有什么想法吗？

Answer 1

在我的团队中，禁止对任何新编写的代码使用 PowerMock，因为它表明代码编写得不好（无法测试）。如果你把它当作一条规则，你通常会得到更干净的代码。

因此，对于您的情况，您的问题是您正在构造 的具体新实例ProcessBuilder，但您的代码并不真正关心它是在此类的具体实例上运行，还是在为您需要的所有方法定义契约的接口上运行。实际上你只使用了start方法，所以首先定义一个相应的接口（Java 还没有定义它真是太糟糕了）：

public interface ProcessStarter {
    Process start() throws IOException;
}

然后，如果您不喜欢将包可见字段用于测试目的，则向您的方法添加一个包可见字段或参数：Function<String[], ProcessStarter> processStarterProvider并在代码中使用它：

ProcessStarter starter = processStarterProvider.apply(commandArr);
Process proc = starter.start();

最后，提供默认实现。如果你去一个领域：

Function<String[], ProcessStarter> processStarterProvider = (commandArr) -> {
    ProcessBuilder pb = new ProcessBuilder(commandArr);
    pb.redirectErrorStream(true);
    return (ProcessStarter) pb::start;
};

现在您不需要任何 PowerMock，只需进行简单的模拟即可！

再想一想，尽管上述使用接口的方法通常适用，并且我建议在整个过程中使用它，但在这种特殊情况下，如果您愿意将 IOException 包装到运行时异常中，那么您可以使用单个Function<String[], Process>接口和以下默认实现：

Function<String[], Process> processProvider = (commandArr) -> {
    ProcessBuilder pb = new ProcessBuilder(commandArr);
    pb.redirectErrorStream(true);
    try {
        return pb.start();
    } catch (IOException ex) {
        throw new RuntimeException(ex);
};

它比上面的更短并且可测试。为了清楚起见，我可能会坚持上面较长的一个。

在这两种情况下，在测试中您都需要以某种方式提出一个实例Process，它本身也没有实现任何接口（设计不佳），因此可能需要一个类似的包装器接口，如上所示。Process但是，考虑到您的实例processStarterProvider在测试中可能如下所示：

Process mockedProcess = ...

myInstance.processStarterProvider = (commandArr) -> () -> mockedProcess;

在这种情况下Function<String[], Process>就更简单了：

Process mockedProcess = ...

myInstance.processProvider = (commandArr) -> mockedProcess;