zydronium/google-nomulus
1
0
Fork 0
mirror of https://github.com/google/nomulus.git synced 2025-06-27 14:54:51 +02:00
Code Issues Projects Releases Packages Wiki Activity

Add better assertions on registry_tool stdout

Browse source 
-------------
Created by MOE: https://github.com/google/moe
MOE_MIGRATED_REVID=127561648
This commit is contained in:
mcilwain 2016-07-15 11:38:50 -07:00 • committed by Ben McIlwain
parent bca9b2a806
commit 60180348cd
1 changed files with 8 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

11
javatests/google/registry/tools/CommandTestCase.java
View file

@ -145,17 +145,22 @@ public abstract class CommandTestCase<C extends Command> {
return ofy().load().type(PollMessage.class).count(); return ofy().load().type(PollMessage.class).count();
} }
protected void assertStdoutIs(String expected) throws Exception {
assertThat(getStdoutAsString()).isEqualTo(expected);
}
protected void assertInStdout(String... expected) throws Exception { protected void assertInStdout(String... expected) throws Exception {
String stdout = getStdoutAsString();
for (String line : expected) { for (String line : expected) {
assertThat(stdout.toString(UTF_8.toString())).contains(line); assertThat(stdout).contains(line);
} }
} }
void assertNotInStdout(String expected) throws Exception { void assertNotInStdout(String expected) throws Exception {
assertThat(stdout.toString(UTF_8.toString())).doesNotContain(expected); assertThat(getStdoutAsString()).doesNotContain(expected);
} }
String getStdoutAsString() { protected String getStdoutAsString() {
return new String(stdout.toByteArray(), UTF_8); return new String(stdout.toByteArray(), UTF_8);
} }