blob: 882519c247c596c360ee582114f1172994f74329 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
package org.zwobble.hobgoblin.compiler.testing;
import java.io.IOException;
import java.io.Reader;
import java.nio.charset.StandardCharsets;
import java.nio.file.Path;
import java.util.List;
public class TestProcessRunner {
public static TestProcessRunner command(List<String> command) {
return new TestProcessRunner(
new ProcessBuilder().command(command)
);
}
private final ProcessBuilder processBuilder;
private TestProcessRunner(ProcessBuilder processBuilder) {
this.processBuilder = processBuilder;
}
public TestProcessRunner cwd(Path cwd) {
this.processBuilder.directory(cwd.toFile());
return this;
}
public TestProcessRunner addEnvVar(String name, String value) {
this.processBuilder.environment().put(name, value);
return this;
}
/**
* Run a process until it exits, capturing its stdout and stderr.
*/
public ExecutionResult runCaptureOutput() throws IOException, InterruptedException {
var process = this.processBuilder.start();
var stdout = ThreadReader.start(process.inputReader(StandardCharsets.UTF_8));
var stderr = ThreadReader.start(process.errorReader(StandardCharsets.UTF_8));
var exitCode = process.waitFor();
return new ExecutionResult(exitCode, stdout.waitForString(), stderr.waitForString());
}
private static class ThreadReader {
public static ThreadReader start(Reader reader) {
var lock = new Object();
var output = new String[1];
Thread.startVirtualThread(() -> {
synchronized (lock) {
try (reader) {
output[0] = reader.readAllAsString();
} catch (IOException exception) {
// TODO: handle this more elegantly.
throw new RuntimeException(exception);
}
}
});
return new ThreadReader(lock, output);
}
private final Object lock;
private final String[] output;
public ThreadReader(Object lock, String[] output) {
this.lock = lock;
this.output = output;
}
public String waitForString() {
synchronized (this.lock) {
return this.output[0];
}
}
}
}
|
