blob: e4ef9b833946d9338454602d639e5cf44c2a5874 (
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
|
package cz.crcs.ectester.reader.output;
import cz.crcs.ectester.common.output.TeeTestWriter;
import cz.crcs.ectester.common.output.TestWriter;
import javax.xml.parsers.ParserConfigurationException;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.PrintStream;
import java.util.regex.Pattern;
/**
* @author Jan Jancar johny@neuromancer.sk
*/
public class FileTestWriter extends TeeTestWriter {
private static final Pattern PREFIX = Pattern.compile("(text|xml|yaml|yml):.+");
public FileTestWriter(String defaultFormat, boolean systemOut, String[] files) throws ParserConfigurationException, FileNotFoundException {
int fLength = files == null ? 0 : files.length;
writers = new TestWriter[systemOut ? fLength + 1 : fLength];
if (systemOut) {
writers[0] = createWriter(defaultFormat, System.out);
}
for (int i = 0; i < fLength; ++i) {
String fName = files[i];
String format = null;
if (PREFIX.matcher(fName).matches()) {
String[] split = fName.split(":",2);
format = split[0];
fName = split[1];
}
writers[i + 1] = createWriter(format, new PrintStream(new FileOutputStream(fName)));
}
}
private TestWriter createWriter(String format, PrintStream out) throws ParserConfigurationException {
if (format == null) {
return new TextTestWriter(out);
}
switch (format) {
case "text":
return new TextTestWriter(out);
case "xml":
return new XMLTestWriter(out);
case "yaml":
case "yml":
return new YAMLTestWriter(out);
default:
return null;
}
}
}
|
