package org.mapfish.print.processor.map;
import com.google.common.base.Predicate;
import com.google.common.io.Files;
import jsr166y.ForkJoinPool;
import jsr166y.ForkJoinTask;
import org.junit.Test;
import org.mapfish.print.AbstractMapfishSpringTest;
import org.mapfish.print.TestHttpClientFactory;
import org.mapfish.print.config.Configuration;
import org.mapfish.print.config.ConfigurationFactory;
import org.mapfish.print.config.Template;
import org.mapfish.print.output.Values;
import org.mapfish.print.parser.MapfishParser;
import org.mapfish.print.test.util.ImageSimilarity;
import org.mapfish.print.wrapper.json.PJsonObject;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpMethod;
import org.springframework.mock.http.client.MockClientHttpRequest;
import org.springframework.test.annotation.DirtiesContext;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.net.URI;
import java.util.List;
import static org.junit.Assert.assertEquals;
/**
* Basic test of the Map processor.
* <p></p>
* Created by Jesse on 3/26/14.
*/
public class CreateMapProcessorGridFixedNumlinesPointRotatedTest extends AbstractMapfishSpringTest {
public static final String BASE_DIR = "grid_numlines_points_fixedscale_rotated/";
@Autowired
private ConfigurationFactory configurationFactory;
@Autowired
private TestHttpClientFactory requestFactory;
@Autowired
private MapfishParser parser;
@Autowired
private ForkJoinPool forkJoinPool;
@Test
@DirtiesContext
public void testExecute() throws Exception {
final String host = "center_osm_fixedscale";
requestFactory.registerHandler(
new Predicate<URI>() {
@Override
public boolean apply(URI input) {
return (("" + input.getHost()).contains(host + ".osm")) || input.getAuthority().contains(host + ".osm");
}
}, new TestHttpClientFactory.Handler() {
@Override
public MockClientHttpRequest handleRequest(URI uri, HttpMethod httpMethod) throws Exception {
try {
byte[] bytes = Files.toByteArray(getFile("/map-data/osm" + uri.getPath()));
return ok(uri, bytes, httpMethod);
} catch (AssertionError e) {
return error404(uri, httpMethod);
}
}
}
);
final Configuration config = configurationFactory.getConfig(getFile(BASE_DIR + "config.yaml"));
final Template template = config.getTemplate("main");
PJsonObject requestData = loadJsonRequestData();
PJsonObject map = requestData.getJSONObject("attributes").getJSONObject("map");
int[] rotationsToTest = {23, 90, 123, 180, 203, 270, 310, 360};
for (int rotation : rotationsToTest) {
map.getInternalObj().put("rotation", rotation);
Values values = new Values(requestData, template, this.parser, getTaskDirectory(), this.requestFactory, new File("."));
final ForkJoinTask<Values> taskFuture = this.forkJoinPool.submit(
template.getProcessorGraph().createTask(values));
taskFuture.get();
@SuppressWarnings("unchecked")
List<URI> layerGraphics = (List<URI>) values.getObject("layerGraphics", List.class);
assertEquals(2, layerGraphics.size());
final BufferedImage referenceImage = ImageSimilarity.mergeImages(layerGraphics, 780, 330);
String imageName = getExpectedImageName("_" + rotation, referenceImage, BASE_DIR);
try {
new ImageSimilarity(referenceImage, 2).assertSimilarity(getFile(BASE_DIR + imageName), 30);
} catch (AssertionError e) {
// throw new AssertionError("Image Similarity test failed for the rotation: " + rotation);
}
}
}
private static PJsonObject loadJsonRequestData() throws IOException {
return parseJSONObjectFromFile(CreateMapProcessorGridFixedNumlinesPointRotatedTest.class, BASE_DIR + "requestData.json");
}
}