planetiler/planetiler-core/src/test/java/com/onthegomap/planetiler/mbtiles/VerifyTest.java

package com.onthegomap.planetiler.mbtiles;

import static com.onthegomap.planetiler.TestUtils.newPolygon;
import static com.onthegomap.planetiler.geo.GeoUtils.point;
import static com.onthegomap.planetiler.util.Gzip.gzip;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertTrue;

import com.onthegomap.planetiler.VectorTile;
import com.onthegomap.planetiler.archive.TileEncodingResult;
import com.onthegomap.planetiler.geo.TileCoord;
import java.io.IOException;
import java.util.List;
import java.util.Map;
import java.util.OptionalLong;
import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;

class VerifyTest {

  private Mbtiles mbtiles;

  @BeforeEach
  public void setup() {
    mbtiles = Mbtiles.newInMemoryDatabase();
  }

  @AfterEach
  public void teardown() throws IOException {
    mbtiles.close();
  }

  @Test
  void testEmptyFileInvalid() {
    assertInvalid(mbtiles);
  }

  @Test
  void testEmptyTablesInvalid() {
    mbtiles.createTablesWithIndexes();
    assertInvalid(mbtiles);
  }

  @Test
  void testValidWithNameAndOneTile() throws IOException {
    mbtiles.createTablesWithIndexes();
    mbtiles.metadata().setName("name");
    try (var writer = mbtiles.newTileWriter()) {
      VectorTile tile = new VectorTile();
      tile.addLayerFeatures("layer", List.of(new VectorTile.Feature(
        "layer",
        1,
        VectorTile.encodeGeometry(point(0, 0)),
        Map.of()
      )));
      writer.write(new TileEncodingResult(TileCoord.ofXYZ(0, 0, 0), gzip(tile.encode()), OptionalLong.empty()));
    }
    assertValid(mbtiles);
  }

  @Test
  void testInvalidGeometry() throws IOException {
    mbtiles.createTablesWithIndexes();
    mbtiles.metadata().setName("name");
    try (var writer = mbtiles.newTileWriter()) {
      VectorTile tile = new VectorTile();
      tile.addLayerFeatures("layer", List.of(new VectorTile.Feature(
        "layer",
        1,
        // self-intersecting bow-tie shape
        VectorTile.encodeGeometry(newPolygon(
          0, 0,
          10, 0,
          0, 10,
          10, 10,
          0, 0
        )),
        Map.of()
      )));
      writer.write(new TileEncodingResult(TileCoord.ofXYZ(0, 0, 0), gzip(tile.encode()), OptionalLong.empty()));
    }
    assertInvalid(mbtiles);
  }

  private void assertInvalid(Mbtiles mbtiles) {
    assertTrue(Verify.verify(mbtiles).numErrors() > 0);
  }

  private void assertValid(Mbtiles mbtiles) {
    assertEquals(0, Verify.verify(mbtiles).numErrors());
  }
}