如何对使用 ProcessBuilder 和 Process 的 Java 方法进行单元测试？

Question

我有一个 Java 方法，它使用 ProcessBuilder 启动一个进程，并将其输出传输到一个字节数组中，然后在进程完成时返回其字节数组。

伪代码：

ProcessBuilder b = new ProcessBuilder("my.exe")
Process p = b.start();
... // get output from process, close process

对该方法进行单元测试的最佳方法是什么？ 我还没有找到一种方法来模拟 ProcessBuilder （这是最终的），即使有令人难以置信的很棒的 JMockit，它给了我一个 NoClassDefFoundError ：

java.lang.NoClassDefFoundError: test/MockProcessBuilder
    at java.lang.ProcessBuilder.<init>(ProcessBuilder.java)
    at mypackage.MyProcess.start(ReportReaderWrapperImpl.java:97)
    at test.MyProcessTest.testStart(ReportReaderWrapperImplTest.java:28)

有什么想法吗？

---

答案 - 正如 Olaf 所建议的，我最终将这些行重构为一个接口，

Process start(String param) throws IOException;

现在我将该接口的一个实例传递到我想要测试的类中（在其构造函数中），通常使用默认实现原始线路。 当我想测试时，我只需使用接口的模拟实现。 就像一个魅力，虽然我确实想知道我是否在这里过度交互......

Answer 1

保护自己免受嘲笑。 创建一个接口来执行您真正想要的操作（例如隐藏涉及外部进程的事实）或仅用于 Process 和 ProcessBuilder。

您不想测试 ProcessBuilder 和 Process 是否工作，只想测试您可以使用它们的输出。 当您创建接口时，一个简单的实现（可以轻松检查）委托给 ProcessBuilder 和 Process，另一个实现会模拟此行为。 稍后您甚至可能有另一个实现来完成您需要的操作，而无需启动另一个进程。

Answer 2

使用较新版本的 JMockit (0.98+)，您应该能够轻松模拟 JRE 类，例如 Process 和 ProcessBuilder。 因此，无需仅为测试而创建接口...

完整示例（使用 JMockit 1.16）：

public class MyProcessTest
{
    public static class MyProcess {
        public byte[] run() throws IOException, InterruptedException {
            Process process = new ProcessBuilder("my.exe").start();
            process.waitFor();

            // Simplified example solution:
            InputStream processOutput = process.getInputStream();
            byte[] output = new byte[8192];
            int bytesRead = processOutput.read(output);

            return Arrays.copyOf(output, bytesRead);
        }
   }

    @Test
    public void runProcessReadingItsOutput(@Mocked final ProcessBuilder pb)
        throws Exception
    {
        byte[] expectedOutput = "mocked output".getBytes();
        final InputStream output = new ByteArrayInputStream(expectedOutput);
        new Expectations() {{ pb.start().getInputStream(); result = output; }};

        byte[] processOutput = new MyProcess().run();

        assertArrayEquals(expectedOutput, processOutput);
    }
}