Write a test

src/app/routes.py

@@ -47,7 +47,7 @@ def fluff_results():
     try:
         linted = lint(sql, dialect=dialect)
         fixed_sql = fix(sql, dialect=dialect)
-    except RuntimeError as e:  # pragma: no cover
+    except RuntimeError as e:
         linted = [
             {
                 "start_line_no": 1,

test/test_app.py

@@ -4,6 +4,7 @@
 import pytest
 from app.routes import sql_encode
 from bs4 import BeautifulSoup
+from unittest.mock import patch
 
 
 @pytest.fixture
@@ -95,6 +96,20 @@ def test_newlines_in_error(client):
     )
 
 
+@patch("app.routes.lint")
+def test_runtime_error(mock_lint, client):
+    """Test that a runtime error is handled."""
+    mock_lint.side_effect = RuntimeError("This is a test error")
+    sql_encoded = sql_encode("select * from table")
+    rv = client.get("/fluffed", query_string=f"""dialect=ansi&sql={sql_encoded}""")
+    html = rv.data.decode().lower()
+    assert "sqlfluff online" in html
+    assert "fixed sql" in html
+    assert "select * from table" in html
+    assert "runtimeerror" in html
+    assert "this is a test error" in html
+
+
 def test_security_headers(client):
     """Test flask-talisman is setting the security headers"""
     rv = client.get("/")