zydronium/google-nomulus
1
0
Fork 0
mirror of https://github.com/google/nomulus.git synced 2025-04-30 03:57:51 +02:00
Code Issues Projects Releases Packages Wiki Activity

Added forgotten command description

Browse source
This commit is contained in:
Michael Muller 2019-08-16 14:45:24 -04:00
parent 2cc8fbec6e
commit 09326cb8ac
1 changed files with 45 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

47
core/src/test/java/google/registry/tools/CommandTestCase.java
View file

@ -36,6 +36,7 @@ import google.registry.tools.params.ParameterFactory;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.IOException;
import java.io.OutputStream;
import java.io.PrintStream;
import java.util.List;
import org.junit.Before;
@ -75,8 +76,10 @@ public abstract class CommandTestCase<C extends Command> {
// Ensure the UNITTEST environment has been set before constructing a new command instance.
RegistryToolEnvironment.UNITTEST.setup(systemPropertyRule);
command = newCommandInstance();
System.setOut(new PrintStream(stdout));
System.setErr(new PrintStream(stderr));
System.out.println("XXX Before stealing output...");
System.setOut(new PrintStream(new OutputSplitter(System.out, stdout)));
System.setErr(new PrintStream(new OutputSplitter(System.err, stderr)));
System.out.println("XXX After stealing output...");
}
void runCommandInEnvironment(RegistryToolEnvironment env, String... args) throws Exception {
@ -221,4 +224,44 @@ public abstract class CommandTestCase<C extends Command> {
throw new RuntimeException(e);
}
}
static class OutputSplitter extends OutputStream {
OutputStream a, b;
OutputSplitter(OutputStream a, OutputStream b) {
this.a = a;
this.b = b;
}
@Override
public void write(byte[] data) throws IOException {
a.write(data);
b.write(data);
}
@Override
public void write(byte[] data, int off, int len) throws IOException {
a.write(data, off, len);
b.write(data, off, len);
}
@Override
public void close() throws IOException {
a.close();
b.close();
}
@Override
public void flush() throws IOException {
a.close();
b.close();
}
@Override
public void write(int val) throws IOException {
a.write(val);
b.write(val);
}
}
}