mirror of https://github.com/encode/starlette.git
94 lines
2.7 KiB
Python
94 lines
2.7 KiB
Python
|
from starlette import TestClient
|
||
|
|
from starlette.staticfiles import StaticFile, StaticFiles
|
||
|
|
import os
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
|
||
|
|
def test_staticfile(tmpdir):
|
||
|
|
path = os.path.join(tmpdir, "example.txt")
|
||
|
|
with open(path, "w") as file:
|
||
|
|
file.write("<file content>")
|
||
|
|
|
||
|
|
app = StaticFile(path=path)
|
||
|
|
client = TestClient(app)
|
||
|
|
response = client.get("/")
|
||
|
|
assert response.status_code == 200
|
||
|
|
assert response.text == '<file content>'
|
||
|
|
|
||
|
|
|
||
|
|
def test_staticfile_post(tmpdir):
|
||
|
|
path = os.path.join(tmpdir, "example.txt")
|
||
|
|
with open(path, "w") as file:
|
||
|
|
file.write("<file content>")
|
||
|
|
|
||
|
|
app = StaticFile(path=path)
|
||
|
|
client = TestClient(app)
|
||
|
|
response = client.post("/")
|
||
|
|
assert response.status_code == 406
|
||
|
|
assert response.text == 'Method not allowed'
|
||
|
|
|
||
|
|
|
||
|
|
def test_staticfile_with_directory_raises_error(tmpdir):
|
||
|
|
app = StaticFile(path=tmpdir)
|
||
|
|
client = TestClient(app)
|
||
|
|
with pytest.raises(RuntimeError) as exc:
|
||
|
|
response = client.get("/")
|
||
|
|
assert 'is not a file' in str(exc)
|
||
|
|
|
||
|
|
|
||
|
|
def test_staticfile_with_missing_file_raises_error(tmpdir):
|
||
|
|
path = os.path.join(tmpdir, '404.txt')
|
||
|
|
app = StaticFile(path=path)
|
||
|
|
client = TestClient(app)
|
||
|
|
with pytest.raises(RuntimeError) as exc:
|
||
|
|
response = client.get("/")
|
||
|
|
assert 'does not exist' in str(exc)
|
||
|
|
|
||
|
|
|
||
|
|
def test_staticfiles(tmpdir):
|
||
|
|
path = os.path.join(tmpdir, "example.txt")
|
||
|
|
with open(path, "w") as file:
|
||
|
|
file.write("<file content>")
|
||
|
|
|
||
|
|
app = StaticFiles(directory=tmpdir)
|
||
|
|
client = TestClient(app)
|
||
|
|
response = client.get("/example.txt")
|
||
|
|
assert response.status_code == 200
|
||
|
|
assert response.text == '<file content>'
|
||
|
|
|
||
|
|
|
||
|
|
def test_staticfiles_post(tmpdir):
|
||
|
|
path = os.path.join(tmpdir, "example.txt")
|
||
|
|
with open(path, "w") as file:
|
||
|
|
file.write("<file content>")
|
||
|
|
|
||
|
|
app = StaticFiles(directory=tmpdir)
|
||
|
|
client = TestClient(app)
|
||
|
|
response = client.post("/example.txt")
|
||
|
|
assert response.status_code == 406
|
||
|
|
assert response.text == 'Method not allowed'
|
||
|
|
|
||
|
|
|
||
|
|
def test_staticfiles_with_directory_returns_404(tmpdir):
|
||
|
|
path = os.path.join(tmpdir, "example.txt")
|
||
|
|
with open(path, "w") as file:
|
||
|
|
file.write("<file content>")
|
||
|
|
|
||
|
|
app = StaticFiles(directory=tmpdir)
|
||
|
|
client = TestClient(app)
|
||
|
|
response = client.get("/")
|
||
|
|
assert response.status_code == 404
|
||
|
|
assert response.text == 'Not found'
|
||
|
|
|
||
|
|
|
||
|
|
def test_staticfiles_with_missing_file_returns_404(tmpdir):
|
||
|
|
path = os.path.join(tmpdir, "example.txt")
|
||
|
|
with open(path, "w") as file:
|
||
|
|
file.write("<file content>")
|
||
|
|
|
||
|
|
app = StaticFiles(directory=tmpdir)
|
||
|
|
client = TestClient(app)
|
||
|
|
response = client.get("/404.txt")
|
||
|
|
assert response.status_code == 404
|
||
|
|
assert response.text == 'Not found'
|