72 lines
2.5 KiB
Java
72 lines
2.5 KiB
Java
package net.sourceforge.plantuml.servlet;
|
|
|
|
import java.io.IOException;
|
|
import java.net.HttpURLConnection;
|
|
import java.net.URL;
|
|
|
|
import net.sourceforge.plantuml.servlet.utils.WebappTestCase;
|
|
|
|
|
|
public class TestOldProxy extends WebappTestCase {
|
|
|
|
/**
|
|
* Verifies the proxified reception of the default Bob and Alice diagram
|
|
*/
|
|
public void testDefaultProxy() throws IOException {
|
|
final URL url = new URL(getServerUrl() + "/proxy/" + getTestDiagramUrl());
|
|
final HttpURLConnection conn = (HttpURLConnection)url.openConnection();
|
|
// Analyze response
|
|
// Verifies HTTP status code and the Content-Type
|
|
assertEquals("Bad HTTP status received", 200, conn.getResponseCode());
|
|
assertEquals(
|
|
"Response content type is not PNG",
|
|
"image/png",
|
|
conn.getContentType().toLowerCase()
|
|
);
|
|
// Get the image and verify its size (~2000 bytes)
|
|
byte[] inMemoryImage = getContentAsBytes(conn);
|
|
int diagramLen = inMemoryImage.length;
|
|
assertTrue(diagramLen > 2000);
|
|
assertTrue(diagramLen < 3000);
|
|
}
|
|
|
|
public void testProxyWithFormat() throws IOException {
|
|
final URL url = new URL(getServerUrl() + "/proxy/svg/" + getTestDiagramUrl());
|
|
final HttpURLConnection conn = (HttpURLConnection)url.openConnection();
|
|
// Analyze response
|
|
// Verifies HTTP status code and the Content-Type
|
|
assertEquals("Bad HTTP status received", 200, conn.getResponseCode());
|
|
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 that the HTTP header of a diagram incites the browser to cache it.
|
|
*/
|
|
public void testInvalidUrl() throws IOException {
|
|
final URL url = new URL(getServerUrl() + "/proxy/invalidURL");
|
|
final HttpURLConnection conn = (HttpURLConnection)url.openConnection();
|
|
// Analyze response
|
|
// Check if status code is 400
|
|
assertEquals(
|
|
"Bad HTTP status received",
|
|
400,
|
|
conn.getResponseCode()
|
|
);
|
|
// Check error message
|
|
assertTrue(
|
|
"Response is not malformed URL",
|
|
getContentText(conn.getErrorStream()).contains("URL malformed.")
|
|
);
|
|
}
|
|
|
|
}
|