Skip to content

Commit 6494f0e

Browse files
vincentsaragovincentsarago
committed
add tests
1 parent 0d1c6aa commit 6494f0e

File tree

5 files changed

+181
-47
lines changed

5 files changed

+181
-47
lines changed

‎CHANGES.md‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
## 0.15.0 (2025-02-26)
1+
## 0.15.0 (2025-02-27)
22

33
* add support for `VSIFile` backend (https://github.com/developmentseed/tilebench/pull/27)
44

‎pyproject.toml‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@ test = [
3535
"pytest-cov",
3636
"pytest-asyncio",
3737
"requests",
38+
"vsifile",
3839
]
3940
dev = [
4041
"pre-commit",

‎tests/test_middleware.py‎

Lines changed: 76 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,10 @@
11
"""Tests for tilebench."""
22

3+
import rasterio
34
from fastapi import FastAPI
45
from rio_tiler.io import Reader
56
from starlette.testclient import TestClient
7+
from vsifile.rasterio import opener
68

79
from tilebench.middleware import NoCacheMiddleware, VSIStatsMiddleware
810

@@ -33,26 +35,77 @@ def tile():
3335
def skip():
3436
return "I've been skipped"
3537

36-
client = TestClient(app)
37-
38-
response = client.get("/info")
39-
assert response.status_code == 200
40-
assert response.headers["content-type"] == "application/json"
41-
assert response.headers["Cache-Control"] == "no-cache"
42-
assert response.headers["VSI-Stats"]
43-
stats = response.headers["VSI-Stats"]
44-
assert "head;count=" in stats
45-
assert "get;count=" in stats
46-
47-
response = client.get("/tile")
48-
assert response.status_code == 200
49-
assert response.headers["content-type"] == "application/json"
50-
assert response.headers["VSI-Stats"]
51-
stats = response.headers["VSI-Stats"]
52-
assert "head;count=" in stats
53-
assert "get;count=" in stats
54-
55-
response = client.get("/skip")
56-
assert response.status_code == 200
57-
assert response.headers["content-type"] == "application/json"
58-
assert "VSI-Stats" not in response.headers
38+
with TestClient(app) as client:
39+
response = client.get("/info")
40+
assert response.status_code == 200
41+
assert response.headers["content-type"] == "application/json"
42+
assert response.headers["Cache-Control"] == "no-cache"
43+
assert response.headers["VSI-Stats"]
44+
stats = response.headers["VSI-Stats"]
45+
assert "head;count=" in stats
46+
assert "get;count=" in stats
47+
48+
response = client.get("/tile")
49+
assert response.status_code == 200
50+
assert response.headers["content-type"] == "application/json"
51+
assert response.headers["VSI-Stats"]
52+
stats = response.headers["VSI-Stats"]
53+
assert "head;count=" in stats
54+
assert "get;count=" in stats
55+
56+
response = client.get("/skip")
57+
assert response.status_code == 200
58+
assert response.headers["content-type"] == "application/json"
59+
assert "VSI-Stats" not in response.headers
60+
61+
62+
def test_middleware_vsifile():
63+
"""Simple test."""
64+
app = FastAPI()
65+
app.add_middleware(NoCacheMiddleware)
66+
app.add_middleware(
67+
VSIStatsMiddleware, config={}, exclude_paths=["/skip"], io="vsifile"
68+
)
69+
70+
@app.get("/info")
71+
def head():
72+
"""Get info."""
73+
with rasterio.open(COG_PATH, opener=opener) as src:
74+
with Reader(None, dataset=src) as cog:
75+
cog.info()
76+
return "I got info"
77+
78+
@app.get("/tile")
79+
def tile():
80+
"""Read tile."""
81+
with rasterio.open(COG_PATH, opener=opener) as src:
82+
with Reader(None, dataset=src) as cog:
83+
cog.tile(36460, 52866, 17)
84+
return "I got tile"
85+
86+
@app.get("/skip")
87+
def skip():
88+
return "I've been skipped"
89+
90+
with TestClient(app) as client:
91+
response = client.get("/info")
92+
assert response.status_code == 200
93+
assert response.headers["content-type"] == "application/json"
94+
assert response.headers["Cache-Control"] == "no-cache"
95+
assert response.headers["VSI-Stats"]
96+
stats = response.headers["VSI-Stats"]
97+
assert "head;count=" in stats
98+
assert "get;count=" in stats
99+
100+
response = client.get("/tile")
101+
assert response.status_code == 200
102+
assert response.headers["content-type"] == "application/json"
103+
assert response.headers["VSI-Stats"]
104+
stats = response.headers["VSI-Stats"]
105+
assert "head;count=" in stats
106+
assert "get;count=" in stats
107+
108+
response = client.get("/skip")
109+
assert response.status_code == 200
110+
assert response.headers["content-type"] == "application/json"
111+
assert "VSI-Stats" not in response.headers

‎tests/test_reader.py‎

Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,8 @@
11
"""Tests for tilebench."""
22

3+
import rasterio
34
from rio_tiler.io import Reader
5+
from vsifile.rasterio import opener
46

57
from tilebench import profile as profiler
68

@@ -41,3 +43,28 @@ def _read_tile(src_path: str, x: int, y: int, z: int, tilesize: int = 256):
4143
assert stats.get("GET")
4244
assert stats.get("Timing")
4345
assert stats.get("WarpKernels")
46+
47+
48+
def test_vsifile():
49+
"""Checkout profile output."""
50+
51+
@profiler(
52+
kernels=True,
53+
add_to_return=True,
54+
quiet=True,
55+
config={"GDAL_DISABLE_READDIR_ON_OPEN": "EMPTY_DIR"},
56+
io="vsifile",
57+
)
58+
def _read_tile(src_path: str, x: int, y: int, z: int, tilesize: int = 256):
59+
with rasterio.open(src_path, opener=opener) as src:
60+
with Reader(None, dataset=src) as cog:
61+
return cog.tile(x, y, z, tilesize=tilesize)
62+
63+
(data, mask), stats = _read_tile(COG_PATH, 36460, 52866, 17)
64+
assert data.shape
65+
assert mask.shape
66+
assert stats
67+
assert "HEAD" in stats
68+
assert stats.get("GET")
69+
assert stats.get("Timing")
70+
assert "WarpKernels" in stats

‎tests/test_viz.py‎

Lines changed: 76 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,10 @@
11
"""Tests for tilebench."""
22

3+
import attr
4+
import rasterio
5+
from rio_tiler.io import Reader
36
from starlette.testclient import TestClient
7+
from vsifile.rasterio import opener
48

59
from tilebench.viz import TileDebug
610

@@ -17,26 +21,75 @@ def test_viz():
1721
assert app.endpoint == "http://127.0.0.1:8080"
1822
assert app.template_url == "http://127.0.0.1:8080"
1923

20-
client = TestClient(app.app)
21-
22-
response = client.get("/tiles/17/36460/52866")
23-
assert response.status_code == 200
24-
assert response.headers["content-type"] == "application/json"
25-
assert response.headers["Cache-Control"] == "no-cache"
26-
assert response.headers["VSI-Stats"]
27-
stats = response.headers["VSI-Stats"]
28-
assert "head;count=" in stats
29-
assert "get;count=" in stats
30-
31-
response = client.get("/info.geojson")
32-
assert response.status_code == 200
33-
assert response.headers["content-type"] == "application/geo+json"
34-
assert "VSI-Stats" not in response.headers
35-
36-
response = client.get("/tiles.geojson?ovr_level=0")
37-
assert response.status_code == 200
38-
assert response.headers["content-type"] == "application/geo+json"
39-
40-
response = client.get("/tiles.geojson?ovr_level=1")
41-
assert response.status_code == 200
42-
assert response.headers["content-type"] == "application/geo+json"
24+
with TestClient(app.app) as client:
25+
response = client.get("/tiles/17/36460/52866")
26+
assert response.status_code == 200
27+
assert response.headers["content-type"] == "application/json"
28+
assert response.headers["Cache-Control"] == "no-cache"
29+
assert response.headers["VSI-Stats"]
30+
stats = response.headers["VSI-Stats"]
31+
assert "head;count=" in stats
32+
assert "get;count=" in stats
33+
34+
response = client.get("/info.geojson")
35+
assert response.status_code == 200
36+
assert response.headers["content-type"] == "application/geo+json"
37+
assert "VSI-Stats" not in response.headers
38+
39+
response = client.get("/tiles.geojson?ovr_level=0")
40+
assert response.status_code == 200
41+
assert response.headers["content-type"] == "application/geo+json"
42+
43+
response = client.get("/tiles.geojson?ovr_level=1")
44+
assert response.status_code == 200
45+
assert response.headers["content-type"] == "application/geo+json"
46+
47+
48+
def test_viz_vsifile():
49+
"""Should work as expected (create TileServer object)."""
50+
51+
@attr.s
52+
class VSIReader(Reader):
53+
"""Rasterio Reader with VSIFILE opener."""
54+
55+
dataset = attr.ib(default=None, init=False) # type: ignore
56+
57+
def __attrs_post_init__(self):
58+
"""Use vsifile.rasterio.opener as Python file opener."""
59+
self.dataset = self._ctx_stack.enter_context(
60+
rasterio.open(self.input, opener=opener)
61+
)
62+
super().__attrs_post_init__()
63+
64+
app = TileDebug(
65+
src_path=COG_PATH,
66+
config={"GDAL_DISABLE_READDIR_ON_OPEN": "EMPTY_DIR"},
67+
reader=VSIReader,
68+
io_backend="vsifile",
69+
)
70+
assert app.port == 8080
71+
assert app.endpoint == "http://127.0.0.1:8080"
72+
assert app.template_url == "http://127.0.0.1:8080"
73+
74+
with TestClient(app.app) as client:
75+
response = client.get("/tiles/17/36460/52866")
76+
assert response.status_code == 200
77+
assert response.headers["content-type"] == "application/json"
78+
assert response.headers["Cache-Control"] == "no-cache"
79+
assert response.headers["VSI-Stats"]
80+
stats = response.headers["VSI-Stats"]
81+
assert "head;count=" in stats
82+
assert "get;count=" in stats
83+
84+
response = client.get("/info.geojson")
85+
assert response.status_code == 200
86+
assert response.headers["content-type"] == "application/geo+json"
87+
assert "VSI-Stats" not in response.headers
88+
89+
response = client.get("/tiles.geojson?ovr_level=0")
90+
assert response.status_code == 200
91+
assert response.headers["content-type"] == "application/geo+json"
92+
93+
response = client.get("/tiles.geojson?ovr_level=1")
94+
assert response.status_code == 200
95+
assert response.headers["content-type"] == "application/geo+json"

0 commit comments

Comments
 (0)