package nonreg.scxml; import static java.nio.charset.StandardCharsets.UTF_8; import static org.junit.jupiter.api.Assertions.assertEquals; import static org.junit.jupiter.api.Assertions.assertTrue; import java.io.ByteArrayOutputStream; import java.io.IOException; import java.io.UnsupportedEncodingException; import java.nio.file.Files; import java.nio.file.Path; import java.nio.file.Paths; import java.util.Collection; import java.util.List; import java.util.Map; import java.util.TreeMap; import java.util.concurrent.atomic.AtomicInteger; import net.sourceforge.plantuml.FileFormat; import net.sourceforge.plantuml.FileFormatOption; import net.sourceforge.plantuml.SourceStringReader; import net.sourceforge.plantuml.core.DiagramDescription; public class ScXmlTest { private static final String TRIPLE_QUOTE = "\"\"\""; protected void checkXmlAndDescription(final String expectedDescription) throws IOException, UnsupportedEncodingException { final String actual = runPlantUML(expectedDescription, FileFormat.SCXML); final String expected = readStringFromSourceFile(getDiagramFile(), "{{{", "}}}"); if (sortString(actual).equals(sortString(expected)) == false) { assertEquals(expected, actual, "Generated ScXml is not ok"); } } private String sortString(String s) { final Map map = new TreeMap<>(); for (int i = 0; i < s.length(); i++) { final char ch = s.charAt(i); // We ignore non writable characters if (ch <= ' ') continue; AtomicInteger count = map.get(ch); if (count == null) map.put(ch, new AtomicInteger(1)); else count.addAndGet(1); } return map.toString(); } private String getLocalFolder() { return "test/" + getPackageName().replace(".", "/"); } private String getPackageName() { return getClass().getPackage().getName(); } private Path getDiagramFile() { return Paths.get(getLocalFolder(), getClass().getSimpleName() + ".java"); } private String runPlantUML(String expectedDescription, FileFormat format) throws IOException, UnsupportedEncodingException { final String diagramText = readStringFromSourceFile(getDiagramFile(), TRIPLE_QUOTE, TRIPLE_QUOTE); final SourceStringReader ssr = new SourceStringReader(diagramText, UTF_8); final ByteArrayOutputStream baos = new ByteArrayOutputStream(); final DiagramDescription diagramDescription = ssr.outputImage(baos, 0, new FileFormatOption(format)); assertEquals(expectedDescription, diagramDescription.getDescription(), "Bad description"); return new String(baos.toByteArray(), UTF_8); } private String readStringFromSourceFile(Path path, String startMarker, String endMarker) throws IOException { assertTrue(Files.exists(path), "Cannot find " + path); assertTrue(Files.isReadable(path), "Cannot read " + path); final List allLines = Files.readAllLines(path, UTF_8); final int first = allLines.indexOf(startMarker); final int last = allLines.lastIndexOf(endMarker); assertTrue(first != -1); assertTrue(last != -1); assertTrue(last > first); return packString(allLines.subList(first + 1, last)); } private String packString(Collection list) { final StringBuilder sb = new StringBuilder(); for (String s : list) { sb.append(s); sb.append("\n"); } return sb.toString(); } }