atst/tests/routes/test_errors.py
Patrick Smith 0425df3d19 Use fixture for CSRF-enabled app
We don't want CSRF protection to stick around after the test runs.
2018-10-16 21:31:03 -04:00

22 lines
536 B
Python

import pytest
@pytest.fixture
def csrf_enabled_app(app):
app.config.update({"WTF_CSRF_ENABLED": True})
yield app
app.config.update({"WTF_CSRF_ENABLED": False})
def test_csrf_error(csrf_enabled_app, client):
response = client.post(
"/requests/new/1",
headers={"Content-Type": "application/x-www-form-urlencoded"},
data="csrf_token=invalid_token",
follow_redirects=True,
)
body = response.data.decode()
assert "Session Expired" in body
assert "Log in Required" in body