48 lines
1.3 KiB
Python
48 lines
1.3 KiB
Python
import re
|
|
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
from app import create_app
|
|
from boker.db import db
|
|
|
|
|
|
class ErrorPageTests(unittest.TestCase):
|
|
def setUp(self):
|
|
self.tmpdir = tempfile.TemporaryDirectory()
|
|
db_path = Path(self.tmpdir.name) / "test.sqlite3"
|
|
self.app = create_app(
|
|
{
|
|
"TESTING": True,
|
|
"PROPAGATE_EXCEPTIONS": False,
|
|
"SQLALCHEMY_DATABASE_URI": f"sqlite:///{db_path}",
|
|
"WTF_CSRF_ENABLED": False,
|
|
}
|
|
)
|
|
|
|
@self.app.get("/explode")
|
|
def explode():
|
|
raise RuntimeError("boom")
|
|
|
|
self.client = self.app.test_client()
|
|
|
|
def tearDown(self):
|
|
with self.app.app_context():
|
|
db.session.remove()
|
|
db.drop_all()
|
|
db.engine.dispose()
|
|
self.tmpdir.cleanup()
|
|
|
|
def test_unexpected_errors_render_crash_page_with_incident_id(self):
|
|
response = self.client.get("/explode")
|
|
|
|
self.assertEqual(response.status_code, 500)
|
|
self.assertIn(b"Something went wrong.", response.data)
|
|
self.assertIn(b"Incident", response.data)
|
|
self.assertNotIn(b"RuntimeError", response.data)
|
|
self.assertRegex(response.get_data(as_text=True), re.compile(r"Incident [a-f0-9]{12}"))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|