planetiler/planetiler-examples/src/test/java/com/onthegomap/planetiler/examples/ToiletsProfileTest.java

73 wiersze
2.6 KiB
Java

package com.onthegomap.planetiler.examples;
import static com.onthegomap.planetiler.TestUtils.assertContains;
import static org.junit.jupiter.api.Assertions.assertEquals;
import com.onthegomap.planetiler.FeatureCollector;
import com.onthegomap.planetiler.TestUtils;
import com.onthegomap.planetiler.config.Arguments;
import com.onthegomap.planetiler.geo.GeoUtils;
import com.onthegomap.planetiler.mbtiles.Mbtiles;
import com.onthegomap.planetiler.reader.SimpleFeature;
import java.nio.file.Path;
import java.util.List;
import java.util.Map;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.io.TempDir;
import org.locationtech.jts.geom.Point;
class ToiletsProfileTest {
private final ToiletsOverlay profile = new ToiletsOverlay();
@Test
void testSourceFeatureProcessing() {
var node = SimpleFeature.create(
TestUtils.newPoint(1, 2),
Map.of("amenity", "toilets")
);
List<FeatureCollector.Feature> mapFeatures = TestUtils.processSourceFeature(node, profile);
// verify output feature
assertEquals(1, mapFeatures.size());
var feature = mapFeatures.get(0);
assertEquals("toilets", feature.getLayer());
// no attributes
assertEquals(Map.of(), feature.getAttrsAtZoom(14));
assertEquals(0, feature.getMinZoom());
assertEquals(14, feature.getMaxZoom());
assertEquals(1, feature.getSortKey());
// at z12 - use label grid to limit output
assertEquals(4, feature.getPointLabelGridLimitAtZoom(12));
assertEquals(32, feature.getPointLabelGridPixelSizeAtZoom(12));
assertEquals(32, feature.getBufferPixelsAtZoom(12));
// at z13 - no label grid (0 disables filtering)
assertEquals(0, feature.getPointLabelGridLimitAtZoom(13));
assertEquals(0, feature.getPointLabelGridPixelSizeAtZoom(13));
assertEquals(4, feature.getBufferPixelsAtZoom(13));
}
@Test
void integrationTest(@TempDir Path tmpDir) throws Exception {
Path dbPath = tmpDir.resolve("output.mbtiles");
ToiletsOverlay.run(Arguments.of(
// Override input source locations
"osm_path", TestUtils.pathToResource("monaco-latest.osm.pbf"),
// Override temp dir location
"tmp", tmpDir.toString(),
// Override output location
"output", dbPath.toString()
));
try (Mbtiles mbtiles = Mbtiles.newReadOnlyDatabase(dbPath)) {
Map<String, String> metadata = mbtiles.metadata().toMap();
assertEquals("Toilets Overlay", metadata.get("name"));
assertContains("openstreetmap.org/copyright", metadata.get("attribution"));
TestUtils.assertNumFeatures(mbtiles, "toilets", 14, Map.of(), GeoUtils.WORLD_LAT_LON_BOUNDS,
34, Point.class);
}
}
}