aws lambda has configurable tile_path and pmtiles_path

This commit is contained in:
Brandon Liu
2022-07-18 16:13:00 +08:00
parent 2d96d92925
commit 2b7f6b6c9f
2 changed files with 74 additions and 16 deletions

View File

@@ -1,18 +1,46 @@
import unittest
from lambda_function import parse_tile_uri
from lambda_function import parse_tile_path, pmtiles_path
class TestLambda(unittest.TestCase):
def test_parse_regex(self):
tileset, tile = parse_tile_uri("/0/0/0.pbf")
self.assertEqual(tileset, None)
self.assertEqual(tile.x, 0)
self.assertEqual(tile.y, 0)
self.assertEqual(tile.z, 0)
tileset, tile = parse_tile_uri("abcd")
def test_parse_tile_default(self):
tileset, tile = parse_tile_path(None, "abcd")
self.assertEqual(tile, None)
tileset, tile = parse_tile_path(None, "/foo/11/22/33.pbf")
self.assertEqual(tileset, "foo")
self.assertEqual(tile.z, 11)
self.assertEqual(tile.x, 22)
self.assertEqual(tile.y, 33)
def test_parse_tile_path_setting(self):
tileset, tile = parse_tile_path(
"/{tileset}/{z}/{y}/{x}.pbf", "/foo/11/22/33.pbf"
)
self.assertEqual(tile.x, 33)
self.assertEqual(tile.y, 22)
tileset, tile = parse_tile_path(
"/tiles/{tileset}/{z}/{x}/{y}.mvt", "/tiles/foo/4/2/3.mvt"
)
self.assertEqual(tileset, "foo")
self.assertEqual(tile.z, 4)
self.assertEqual(tile.x, 2)
self.assertEqual(tile.y, 3)
def test_parse_tile_path_setting_special_chars(self):
tileset, tile = parse_tile_path(
"/folder(new/{tileset}/{z}/{y}/{x}.pbf", "/folder(new/foo/11/22/33.pbf"
)
self.assertEqual(tileset, "foo")
def test_pmtiles_path(self):
self.assertEqual(pmtiles_path(None, "foo"), "foo.pmtiles")
self.assertEqual(
pmtiles_path("folder/{tileset}/file.pmtiles", "foo"),
"folder/foo/file.pmtiles",
)
if __name__ == "__main__":
unittest.main()