121 lines
4.1 KiB
Java
121 lines
4.1 KiB
Java
package net.sourceforge.plantuml.servlet;
|
|
|
|
import java.io.IOException;
|
|
import java.io.InputStream;
|
|
import java.io.OutputStreamWriter;
|
|
import java.net.HttpURLConnection;
|
|
import java.net.URL;
|
|
import java.net.URLConnection;
|
|
import java.util.Scanner;
|
|
|
|
import net.sourceforge.plantuml.servlet.utils.TestUtils;
|
|
import net.sourceforge.plantuml.servlet.utils.WebappTestCase;
|
|
|
|
|
|
public class TestSVG extends WebappTestCase {
|
|
|
|
/**
|
|
* Verifies the generation of the SVG for the Bob -> Alice sample
|
|
*/
|
|
public void testSimpleSequenceDiagram() throws IOException {
|
|
final URL url = new URL(getServerUrl() + "/svg/" + TestUtils.SEQBOB);
|
|
final URLConnection conn = url.openConnection();
|
|
// Analyze response
|
|
// Verifies the Content-Type header
|
|
assertEquals(
|
|
"Response content type is not SVG",
|
|
"image/svg+xml",
|
|
conn.getContentType().toLowerCase()
|
|
);
|
|
// Get the content and verify its size
|
|
String diagram = getContentText(conn);
|
|
int diagramLen = diagram.length();
|
|
assertTrue(diagramLen > 1000);
|
|
assertTrue(diagramLen < 3000);
|
|
}
|
|
|
|
/**
|
|
* Verifies the generation of the SVG for the Bob -> Alice sample
|
|
*/
|
|
public void testPostedSequenceDiagram() throws IOException {
|
|
final URL url = new URL(getServerUrl() + "/svg");
|
|
final HttpURLConnection conn = (HttpURLConnection)url.openConnection();
|
|
conn.setRequestMethod("POST");
|
|
conn.setDoOutput(true);
|
|
conn.setRequestProperty("Content-type", "text/plain");
|
|
try (final OutputStreamWriter writer = new OutputStreamWriter(conn.getOutputStream())) {
|
|
writer.write("@startuml\nBob -> Alice\n@enduml");
|
|
writer.flush();
|
|
}
|
|
// Analyze response
|
|
// HTTP response 200
|
|
assertEquals(
|
|
"Bad HTTP status received",
|
|
200,
|
|
conn.getResponseCode()
|
|
);
|
|
// Verifies the Content-Type header
|
|
assertEquals(
|
|
"Response content type is not SVG",
|
|
"image/svg+xml",
|
|
conn.getContentType().toLowerCase()
|
|
);
|
|
// Get the content and verify its size
|
|
String diagram = getContentText(conn.getInputStream());
|
|
int diagramLen = diagram.length();
|
|
assertTrue(diagramLen > 1000);
|
|
assertTrue(diagramLen < 3000);
|
|
}
|
|
|
|
/**
|
|
* Verifies the generation of the SVG for the Bob -> Alice sample
|
|
*/
|
|
public void testPostedInvalidSequenceDiagram() throws IOException {
|
|
final URL url = new URL(getServerUrl() + "/svg");
|
|
final HttpURLConnection conn = (HttpURLConnection)url.openConnection();
|
|
conn.setRequestMethod("POST");
|
|
conn.setDoOutput(true);
|
|
conn.setRequestProperty("Content-type", "text/plain");
|
|
try (final OutputStreamWriter writer = new OutputStreamWriter(conn.getOutputStream())) {
|
|
writer.write("@startuml\n[Bob\n@enduml");
|
|
writer.flush();
|
|
}
|
|
// Analyze response
|
|
// HTTP response 400
|
|
assertEquals(
|
|
"Bad HTTP status received",
|
|
400,
|
|
conn.getResponseCode()
|
|
);
|
|
}
|
|
|
|
/**
|
|
* Check the content of the SVG
|
|
*/
|
|
public void testSequenceDiagramContent() throws IOException {
|
|
final URL url = new URL(getServerUrl() + "/svg/" + TestUtils.SEQBOB);
|
|
// Analyze response
|
|
// Get the data contained in the XML
|
|
try (
|
|
final InputStream responseStream = url.openStream();
|
|
final Scanner scanner = new Scanner(responseStream).useDelimiter("(<([^<>]+)>)+")
|
|
) {
|
|
String token;
|
|
int bobCounter = 0;
|
|
int aliceCounter = 0;
|
|
while (scanner.hasNext()) {
|
|
token = scanner.next();
|
|
if (token.startsWith("Bob")) {
|
|
bobCounter++;
|
|
}
|
|
if (token.startsWith("Alice")) {
|
|
aliceCounter++;
|
|
}
|
|
}
|
|
assertTrue(bobCounter == 2);
|
|
assertTrue(aliceCounter == 2);
|
|
}
|
|
}
|
|
|
|
}
|