mirror of https://github.com/encode/starlette.git
444 lines
14 KiB
Python
444 lines
14 KiB
Python
import os
|
|
import pathlib
|
|
import stat
|
|
import time
|
|
|
|
import anyio
|
|
import pytest
|
|
|
|
from starlette.applications import Starlette
|
|
from starlette.exceptions import HTTPException
|
|
from starlette.middleware import Middleware
|
|
from starlette.middleware.base import BaseHTTPMiddleware
|
|
from starlette.requests import Request
|
|
from starlette.routing import Mount
|
|
from starlette.staticfiles import StaticFiles
|
|
|
|
|
|
def test_staticfiles(tmpdir, test_client_factory):
|
|
path = os.path.join(tmpdir, "example.txt")
|
|
with open(path, "w") as file:
|
|
file.write("<file content>")
|
|
|
|
app = StaticFiles(directory=tmpdir)
|
|
client = test_client_factory(app)
|
|
response = client.get("/example.txt")
|
|
assert response.status_code == 200
|
|
assert response.text == "<file content>"
|
|
|
|
|
|
def test_staticfiles_with_pathlib(tmpdir, test_client_factory):
|
|
base_dir = pathlib.Path(tmpdir)
|
|
path = base_dir / "example.txt"
|
|
with open(path, "w") as file:
|
|
file.write("<file content>")
|
|
|
|
app = StaticFiles(directory=base_dir)
|
|
client = test_client_factory(app)
|
|
response = client.get("/example.txt")
|
|
assert response.status_code == 200
|
|
assert response.text == "<file content>"
|
|
|
|
|
|
def test_staticfiles_head_with_middleware(tmpdir, test_client_factory):
|
|
"""
|
|
see https://github.com/encode/starlette/pull/935
|
|
"""
|
|
path = os.path.join(tmpdir, "example.txt")
|
|
with open(path, "w") as file:
|
|
file.write("x" * 100)
|
|
|
|
async def does_nothing_middleware(request: Request, call_next):
|
|
response = await call_next(request)
|
|
return response
|
|
|
|
routes = [Mount("/static", app=StaticFiles(directory=tmpdir), name="static")]
|
|
middleware = [Middleware(BaseHTTPMiddleware, dispatch=does_nothing_middleware)]
|
|
app = Starlette(routes=routes, middleware=middleware)
|
|
|
|
client = test_client_factory(app)
|
|
response = client.head("/static/example.txt")
|
|
assert response.status_code == 200
|
|
assert response.headers.get("content-length") == "100"
|
|
|
|
|
|
def test_staticfiles_with_package(test_client_factory):
|
|
app = StaticFiles(packages=["tests"])
|
|
client = test_client_factory(app)
|
|
response = client.get("/example.txt")
|
|
assert response.status_code == 200
|
|
assert response.text == "123\n"
|
|
|
|
app = StaticFiles(packages=[("tests", "statics")])
|
|
client = test_client_factory(app)
|
|
response = client.get("/example.txt")
|
|
assert response.status_code == 200
|
|
assert response.text == "123\n"
|
|
|
|
|
|
def test_staticfiles_post(tmpdir, test_client_factory):
|
|
path = os.path.join(tmpdir, "example.txt")
|
|
with open(path, "w") as file:
|
|
file.write("<file content>")
|
|
|
|
routes = [Mount("/", app=StaticFiles(directory=tmpdir), name="static")]
|
|
app = Starlette(routes=routes)
|
|
client = test_client_factory(app)
|
|
|
|
response = client.post("/example.txt")
|
|
assert response.status_code == 405
|
|
assert response.text == "Method Not Allowed"
|
|
|
|
|
|
def test_staticfiles_with_directory_returns_404(tmpdir, test_client_factory):
|
|
path = os.path.join(tmpdir, "example.txt")
|
|
with open(path, "w") as file:
|
|
file.write("<file content>")
|
|
|
|
routes = [Mount("/", app=StaticFiles(directory=tmpdir), name="static")]
|
|
app = Starlette(routes=routes)
|
|
client = test_client_factory(app)
|
|
|
|
response = client.get("/")
|
|
assert response.status_code == 404
|
|
assert response.text == "Not Found"
|
|
|
|
|
|
def test_staticfiles_with_missing_file_returns_404(tmpdir, test_client_factory):
|
|
path = os.path.join(tmpdir, "example.txt")
|
|
with open(path, "w") as file:
|
|
file.write("<file content>")
|
|
|
|
routes = [Mount("/", app=StaticFiles(directory=tmpdir), name="static")]
|
|
app = Starlette(routes=routes)
|
|
client = test_client_factory(app)
|
|
|
|
response = client.get("/404.txt")
|
|
assert response.status_code == 404
|
|
assert response.text == "Not Found"
|
|
|
|
|
|
def test_staticfiles_instantiated_with_missing_directory(tmpdir):
|
|
with pytest.raises(RuntimeError) as exc_info:
|
|
path = os.path.join(tmpdir, "no_such_directory")
|
|
StaticFiles(directory=path)
|
|
assert "does not exist" in str(exc_info.value)
|
|
|
|
|
|
def test_staticfiles_configured_with_missing_directory(tmpdir, test_client_factory):
|
|
path = os.path.join(tmpdir, "no_such_directory")
|
|
app = StaticFiles(directory=path, check_dir=False)
|
|
client = test_client_factory(app)
|
|
with pytest.raises(RuntimeError) as exc_info:
|
|
client.get("/example.txt")
|
|
assert "does not exist" in str(exc_info.value)
|
|
|
|
|
|
def test_staticfiles_configured_with_file_instead_of_directory(
|
|
tmpdir, test_client_factory
|
|
):
|
|
path = os.path.join(tmpdir, "example.txt")
|
|
with open(path, "w") as file:
|
|
file.write("<file content>")
|
|
|
|
app = StaticFiles(directory=path, check_dir=False)
|
|
client = test_client_factory(app)
|
|
with pytest.raises(RuntimeError) as exc_info:
|
|
client.get("/example.txt")
|
|
assert "is not a directory" in str(exc_info.value)
|
|
|
|
|
|
def test_staticfiles_config_check_occurs_only_once(tmpdir, test_client_factory):
|
|
app = StaticFiles(directory=tmpdir)
|
|
client = test_client_factory(app)
|
|
assert not app.config_checked
|
|
|
|
with pytest.raises(HTTPException):
|
|
client.get("/")
|
|
|
|
assert app.config_checked
|
|
|
|
with pytest.raises(HTTPException):
|
|
client.get("/")
|
|
|
|
|
|
def test_staticfiles_prevents_breaking_out_of_directory(tmpdir):
|
|
directory = os.path.join(tmpdir, "foo")
|
|
os.mkdir(directory)
|
|
|
|
path = os.path.join(tmpdir, "example.txt")
|
|
with open(path, "w") as file:
|
|
file.write("outside root dir")
|
|
|
|
app = StaticFiles(directory=directory)
|
|
# We can't test this with 'requests', so we test the app directly here.
|
|
path = app.get_path({"path": "/../example.txt"})
|
|
scope = {"method": "GET"}
|
|
|
|
with pytest.raises(HTTPException) as exc_info:
|
|
anyio.run(app.get_response, path, scope)
|
|
|
|
assert exc_info.value.status_code == 404
|
|
assert exc_info.value.detail == "Not Found"
|
|
|
|
|
|
def test_staticfiles_never_read_file_for_head_method(tmpdir, test_client_factory):
|
|
path = os.path.join(tmpdir, "example.txt")
|
|
with open(path, "w") as file:
|
|
file.write("<file content>")
|
|
|
|
app = StaticFiles(directory=tmpdir)
|
|
client = test_client_factory(app)
|
|
response = client.head("/example.txt")
|
|
assert response.status_code == 200
|
|
assert response.content == b""
|
|
assert response.headers["content-length"] == "14"
|
|
|
|
|
|
def test_staticfiles_304_with_etag_match(tmpdir, test_client_factory):
|
|
path = os.path.join(tmpdir, "example.txt")
|
|
with open(path, "w") as file:
|
|
file.write("<file content>")
|
|
|
|
app = StaticFiles(directory=tmpdir)
|
|
client = test_client_factory(app)
|
|
first_resp = client.get("/example.txt")
|
|
assert first_resp.status_code == 200
|
|
last_etag = first_resp.headers["etag"]
|
|
second_resp = client.get("/example.txt", headers={"if-none-match": last_etag})
|
|
assert second_resp.status_code == 304
|
|
assert second_resp.content == b""
|
|
|
|
|
|
def test_staticfiles_304_with_last_modified_compare_last_req(
|
|
tmpdir, test_client_factory
|
|
):
|
|
path = os.path.join(tmpdir, "example.txt")
|
|
file_last_modified_time = time.mktime(
|
|
time.strptime("2013-10-10 23:40:00", "%Y-%m-%d %H:%M:%S")
|
|
)
|
|
with open(path, "w") as file:
|
|
file.write("<file content>")
|
|
os.utime(path, (file_last_modified_time, file_last_modified_time))
|
|
|
|
app = StaticFiles(directory=tmpdir)
|
|
client = test_client_factory(app)
|
|
# last modified less than last request, 304
|
|
response = client.get(
|
|
"/example.txt", headers={"If-Modified-Since": "Thu, 11 Oct 2013 15:30:19 GMT"}
|
|
)
|
|
assert response.status_code == 304
|
|
assert response.content == b""
|
|
# last modified greater than last request, 200 with content
|
|
response = client.get(
|
|
"/example.txt", headers={"If-Modified-Since": "Thu, 20 Feb 2012 15:30:19 GMT"}
|
|
)
|
|
assert response.status_code == 200
|
|
assert response.content == b"<file content>"
|
|
|
|
|
|
def test_staticfiles_html_normal(tmpdir, test_client_factory):
|
|
path = os.path.join(tmpdir, "404.html")
|
|
with open(path, "w") as file:
|
|
file.write("<h1>Custom not found page</h1>")
|
|
path = os.path.join(tmpdir, "dir")
|
|
os.mkdir(path)
|
|
path = os.path.join(path, "index.html")
|
|
with open(path, "w") as file:
|
|
file.write("<h1>Hello</h1>")
|
|
|
|
app = StaticFiles(directory=tmpdir, html=True)
|
|
client = test_client_factory(app)
|
|
|
|
response = client.get("/dir/")
|
|
assert response.url == "http://testserver/dir/"
|
|
assert response.status_code == 200
|
|
assert response.text == "<h1>Hello</h1>"
|
|
|
|
response = client.get("/dir")
|
|
assert response.url == "http://testserver/dir/"
|
|
assert response.status_code == 200
|
|
assert response.text == "<h1>Hello</h1>"
|
|
|
|
response = client.get("/dir/index.html")
|
|
assert response.url == "http://testserver/dir/index.html"
|
|
assert response.status_code == 200
|
|
assert response.text == "<h1>Hello</h1>"
|
|
|
|
response = client.get("/missing")
|
|
assert response.status_code == 404
|
|
assert response.text == "<h1>Custom not found page</h1>"
|
|
|
|
|
|
def test_staticfiles_html_without_index(tmpdir, test_client_factory):
|
|
path = os.path.join(tmpdir, "404.html")
|
|
with open(path, "w") as file:
|
|
file.write("<h1>Custom not found page</h1>")
|
|
path = os.path.join(tmpdir, "dir")
|
|
os.mkdir(path)
|
|
|
|
app = StaticFiles(directory=tmpdir, html=True)
|
|
client = test_client_factory(app)
|
|
|
|
response = client.get("/dir/")
|
|
assert response.url == "http://testserver/dir/"
|
|
assert response.status_code == 404
|
|
assert response.text == "<h1>Custom not found page</h1>"
|
|
|
|
response = client.get("/dir")
|
|
assert response.url == "http://testserver/dir"
|
|
assert response.status_code == 404
|
|
assert response.text == "<h1>Custom not found page</h1>"
|
|
|
|
response = client.get("/missing")
|
|
assert response.status_code == 404
|
|
assert response.text == "<h1>Custom not found page</h1>"
|
|
|
|
|
|
def test_staticfiles_html_without_404(tmpdir, test_client_factory):
|
|
path = os.path.join(tmpdir, "dir")
|
|
os.mkdir(path)
|
|
path = os.path.join(path, "index.html")
|
|
with open(path, "w") as file:
|
|
file.write("<h1>Hello</h1>")
|
|
|
|
app = StaticFiles(directory=tmpdir, html=True)
|
|
client = test_client_factory(app)
|
|
|
|
response = client.get("/dir/")
|
|
assert response.url == "http://testserver/dir/"
|
|
assert response.status_code == 200
|
|
assert response.text == "<h1>Hello</h1>"
|
|
|
|
response = client.get("/dir")
|
|
assert response.url == "http://testserver/dir/"
|
|
assert response.status_code == 200
|
|
assert response.text == "<h1>Hello</h1>"
|
|
|
|
with pytest.raises(HTTPException) as exc_info:
|
|
response = client.get("/missing")
|
|
assert exc_info.value.status_code == 404
|
|
|
|
|
|
def test_staticfiles_html_only_files(tmpdir, test_client_factory):
|
|
path = os.path.join(tmpdir, "hello.html")
|
|
with open(path, "w") as file:
|
|
file.write("<h1>Hello</h1>")
|
|
|
|
app = StaticFiles(directory=tmpdir, html=True)
|
|
client = test_client_factory(app)
|
|
|
|
with pytest.raises(HTTPException) as exc_info:
|
|
response = client.get("/")
|
|
assert exc_info.value.status_code == 404
|
|
|
|
response = client.get("/hello.html")
|
|
assert response.status_code == 200
|
|
assert response.text == "<h1>Hello</h1>"
|
|
|
|
|
|
def test_staticfiles_cache_invalidation_for_deleted_file_html_mode(
|
|
tmpdir, test_client_factory
|
|
):
|
|
path_404 = os.path.join(tmpdir, "404.html")
|
|
with open(path_404, "w") as file:
|
|
file.write("<p>404 file</p>")
|
|
path_some = os.path.join(tmpdir, "some.html")
|
|
with open(path_some, "w") as file:
|
|
file.write("<p>some file</p>")
|
|
|
|
common_modified_time = time.mktime(
|
|
time.strptime("2013-10-10 23:40:00", "%Y-%m-%d %H:%M:%S")
|
|
)
|
|
os.utime(path_404, (common_modified_time, common_modified_time))
|
|
os.utime(path_some, (common_modified_time, common_modified_time))
|
|
|
|
app = StaticFiles(directory=tmpdir, html=True)
|
|
client = test_client_factory(app)
|
|
|
|
resp_exists = client.get("/some.html")
|
|
assert resp_exists.status_code == 200
|
|
assert resp_exists.text == "<p>some file</p>"
|
|
|
|
resp_cached = client.get(
|
|
"/some.html",
|
|
headers={"If-Modified-Since": resp_exists.headers["last-modified"]},
|
|
)
|
|
assert resp_cached.status_code == 304
|
|
|
|
os.remove(path_some)
|
|
|
|
resp_deleted = client.get(
|
|
"/some.html",
|
|
headers={"If-Modified-Since": resp_exists.headers["last-modified"]},
|
|
)
|
|
assert resp_deleted.status_code == 404
|
|
assert resp_deleted.text == "<p>404 file</p>"
|
|
|
|
|
|
def test_staticfiles_with_invalid_dir_permissions_returns_401(
|
|
tmpdir, test_client_factory
|
|
):
|
|
path = os.path.join(tmpdir, "example.txt")
|
|
with open(path, "w") as file:
|
|
file.write("<file content>")
|
|
|
|
os.chmod(tmpdir, stat.S_IRWXO)
|
|
|
|
routes = [Mount("/", app=StaticFiles(directory=tmpdir), name="static")]
|
|
app = Starlette(routes=routes)
|
|
client = test_client_factory(app)
|
|
|
|
response = client.get("/example.txt")
|
|
assert response.status_code == 401
|
|
assert response.text == "Unauthorized"
|
|
|
|
|
|
def test_staticfiles_with_missing_dir_returns_404(tmpdir, test_client_factory):
|
|
path = os.path.join(tmpdir, "example.txt")
|
|
with open(path, "w") as file:
|
|
file.write("<file content>")
|
|
|
|
routes = [Mount("/", app=StaticFiles(directory=tmpdir), name="static")]
|
|
app = Starlette(routes=routes)
|
|
client = test_client_factory(app)
|
|
|
|
response = client.get("/foo/example.txt")
|
|
assert response.status_code == 404
|
|
assert response.text == "Not Found"
|
|
|
|
|
|
def test_staticfiles_access_file_as_dir_returns_404(tmpdir, test_client_factory):
|
|
path = os.path.join(tmpdir, "example.txt")
|
|
with open(path, "w") as file:
|
|
file.write("<file content>")
|
|
|
|
routes = [Mount("/", app=StaticFiles(directory=tmpdir), name="static")]
|
|
app = Starlette(routes=routes)
|
|
client = test_client_factory(app)
|
|
|
|
response = client.get("/example.txt/foo")
|
|
assert response.status_code == 404
|
|
assert response.text == "Not Found"
|
|
|
|
|
|
def test_staticfiles_unhandled_os_error_returns_500(
|
|
tmpdir, test_client_factory, monkeypatch
|
|
):
|
|
def mock_timeout(*args, **kwargs):
|
|
raise TimeoutError
|
|
|
|
path = os.path.join(tmpdir, "example.txt")
|
|
with open(path, "w") as file:
|
|
file.write("<file content>")
|
|
|
|
routes = [Mount("/", app=StaticFiles(directory=tmpdir), name="static")]
|
|
app = Starlette(routes=routes)
|
|
client = test_client_factory(app, raise_server_exceptions=False)
|
|
|
|
monkeypatch.setattr("starlette.staticfiles.StaticFiles.lookup_path", mock_timeout)
|
|
|
|
response = client.get("/example.txt")
|
|
assert response.status_code == 500
|
|
assert response.text == "Internal Server Error"
|