octoleo/plantuml
1
0
Fork 0
mirror of https://github.com/octoleo/plantuml.git synced 2024-12-27 12:42:47 +00:00
Code Issues Releases Wiki Activity

Add PlantUmlTestUtils.exportDiagram()

Browse Source
This commit is contained in:
matthew16550 2021-10-14 14:25:48 +11:00
parent 7d1c70dfe9
commit 85311858ad
4 changed files with 130 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
test/net/sourceforge/plantuml/help/HelpTest.java
View File

@ -1,6 +1,6 @@
package net.sourceforge.plantuml.help;
import static net.sourceforge.plantuml.test.TestUtils.renderUmlAsUnicode;
import static net.sourceforge.plantuml.test.PlantUmlTestUtils.exportDiagram;
import static org.assertj.core.api.Assertions.assertThat;
import org.junit.jupiter.api.Test;
@ -10,7 +10,11 @@ class HelpTest {
@Test
public void test_help_themes() throws Exception {
final String output = renderUmlAsUnicode("help themes");
final String output = exportDiagram(
"@startuml",
"help themes",
"@enduml"
).asString();
assertThat(output)
.startsWith("Help on themes")

76
test/net/sourceforge/plantuml/test/PlantUmlTestUtils.java Normal file
View File

@ -0,0 +1,76 @@
package net.sourceforge.plantuml.test;
import static java.nio.charset.StandardCharsets.UTF_8;
import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.util.List;
import net.sourceforge.plantuml.BlockUml;
import net.sourceforge.plantuml.FileFormat;
import net.sourceforge.plantuml.FileFormatOption;
import net.sourceforge.plantuml.SourceStringReader;
import net.sourceforge.plantuml.core.Diagram;
import net.sourceforge.plantuml.error.PSystemError;
import net.sourceforge.plantuml.security.ImageIO;
public class PlantUmlTestUtils {
public static ExportDiagram exportDiagram(String... source) {
final SourceStringReader ssr = new SourceStringReader(StringTestUtils.join("\n", source));
final List<BlockUml> blocks = ssr.getBlocks();
if (blocks.isEmpty()) throw new AssertionError("There is no diagram");
final Diagram diagram = blocks.get(0).getDiagram();
return new ExportDiagram(diagram);
}
public static class ExportDiagram {
private final Diagram diagram;
private boolean metadata;
public ExportDiagram(Diagram diagram) {
this.diagram = diagram;
}
public ExportDiagram assertNoError() {
if (diagram instanceof PSystemError) {
final PSystemError error = (PSystemError) this.diagram;
throw new AssertionError("Diagram has an error: " + StringTestUtils.join("\n", error.getPureAsciiFormatted()));
}
return this;
}
public byte[] asByteArray(FileFormat fileFormat) throws IOException {
final ByteArrayOutputStream os = new ByteArrayOutputStream();
stream(os, fileFormat);
return os.toByteArray();
}
public BufferedImage asImage() throws IOException {
return ImageIO.read(asByteArray(FileFormat.PNG));
}
public String asString() throws IOException {
return asString(FileFormat.UTXT);
}
public String asString(FileFormat fileFormat) throws IOException {
return new String(asByteArray(fileFormat), UTF_8);
}
public ExportDiagram stream(OutputStream os, FileFormat fileFormat) throws IOException {
final FileFormatOption fileFormatOption = new FileFormatOption(fileFormat, metadata);
diagram.exportDiagram(os, 0, fileFormatOption);
return this;
}
public ExportDiagram withMetadata(boolean metadata) {
this.metadata = metadata;
return this;
}
}
}

48
test/net/sourceforge/plantuml/test/StringTestUtils.java Normal file
View File

@ -0,0 +1,48 @@
package net.sourceforge.plantuml.test;
import static java.util.Objects.requireNonNull;
import java.util.Iterator;
public class StringTestUtils {
/**
* This can be replaced by String.join() when we move to Java 1.8
*/
public static String join(CharSequence delimiter, CharSequence... elements) {
requireNonNull(delimiter);
requireNonNull(elements);
if (elements.length == 0) return "";
final StringBuilder b = new StringBuilder();
for (int i = 0; i < elements.length; i++) {
if (i > 0) b.append(delimiter);
b.append(elements[i]);
}
return b.toString();
}
/**
* This can be replaced by String.join() when we move to Java 1.8
*/
public static String join(CharSequence delimiter, Iterable<? extends CharSequence> elements) {
requireNonNull(delimiter);
requireNonNull(elements);
final Iterator<? extends CharSequence> i = elements.iterator();
if (!i.hasNext()) return "";
final StringBuilder b = new StringBuilder();
while(true) {
b.append(i.next());
if (i.hasNext()) {
b.append(delimiter);
}
else {
break;
}
}
return b.toString();
}
}

25
test/net/sourceforge/plantuml/test/TestUtils.java
View File

@ -2,37 +2,12 @@ package net.sourceforge.plantuml.test;
import static java.nio.charset.StandardCharsets.UTF_8;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import net.sourceforge.plantuml.FileFormat;
import net.sourceforge.plantuml.FileFormatOption;
import net.sourceforge.plantuml.Option;
import net.sourceforge.plantuml.SourceStringReader;
public class TestUtils {
public static String renderAsUnicode(String source, String... options) throws Exception {
final Option option = new Option(options);
option.setFileFormatOption(new FileFormatOption(FileFormat.UTXT));
final SourceStringReader ssr = new SourceStringReader(option.getDefaultDefines(), source, UTF_8.name(), option.getConfig());
final ByteArrayOutputStream os = new ByteArrayOutputStream();
ssr.getBlocks().get(0).getDiagram().exportDiagram(os, 0, option.getFileFormatOption());
return new String(os.toByteArray(), UTF_8);
}
public static String renderUmlAsUnicode(String source, String... options) throws Exception {
return renderAsUnicode("@startuml\n" + source + "\n@enduml\n", options);
}
public static void writeUtf8File(Path path, String string) throws IOException {
Files.createDirectories(path.getParent());