-
Notifications
You must be signed in to change notification settings - Fork 1.1k
/
ConsoleOutput.java
71 lines (58 loc) · 1.83 KB
/
ConsoleOutput.java
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
/*******************************************************************************
* Copyright (c) 2009, 2020 Mountainminds GmbH & Co. KG and Contributors
* This program and the accompanying materials are made available under
* the terms of the Eclipse Public License 2.0 which is available at
* http://www.eclipse.org/legal/epl-2.0
*
* SPDX-License-Identifier: EPL-2.0
*
* Contributors:
* Marc R. Hoffmann - initial API and implementation
*
*******************************************************************************/
package org.jacoco.examples;
import static org.hamcrest.CoreMatchers.containsString;
import static org.hamcrest.MatcherAssert.assertThat;
import static org.hamcrest.core.Is.is;
import static org.junit.Assert.fail;
import java.io.ByteArrayOutputStream;
import java.io.PrintStream;
import java.io.UnsupportedEncodingException;
import org.hamcrest.Matcher;
import org.junit.rules.ExternalResource;
/**
* In-Memory buffer to assert console output.
*/
public class ConsoleOutput extends ExternalResource {
private final ByteArrayOutputStream buffer;
public final PrintStream stream;
public ConsoleOutput() {
this.buffer = new ByteArrayOutputStream();
try {
this.stream = new PrintStream(buffer, true, "UTF-8");
} catch (UnsupportedEncodingException e) {
throw new AssertionError(e.getMessage());
}
}
@Override
protected void after() {
buffer.reset();
}
public static Matcher<String> containsLine(String line) {
return containsString(String.format("%s%n", line));
}
public static Matcher<String> isEmpty() {
return is("");
}
public String getContents() {
try {
return new String(buffer.toByteArray(), "UTF-8");
} catch (UnsupportedEncodingException e) {
fail(e.getMessage());
return "";
}
}
public void expect(Matcher<String> matcher) {
assertThat(getContents(), matcher);
}
}