feat(server): handle snowflake ProgrammingError

fakesnow/server.py

@@ -6,6 +6,7 @@
 from base64 import b64encode
 from dataclasses import dataclass
 
+import snowflake.connector.errors
 from starlette.applications import Starlette
 from starlette.concurrency import run_in_threadpool
 from starlette.requests import Request
@@ -45,8 +46,22 @@ async def query_request(request: Request) -> JSONResponse:
 
         sql_text = body_json["sqlText"]
 
-        # only a single sql statement is sent at a time by the python snowflake connector
-        cur = await run_in_threadpool(conn.cursor().execute, sql_text)
+        try:
+            # only a single sql statement is sent at a time by the python snowflake connector
+            cur = await run_in_threadpool(conn.cursor().execute, sql_text)
+        except snowflake.connector.errors.ProgrammingError as e:
+            code = f"{e.errno:06d}"
+            return JSONResponse(
+                {
+                    "data": {
+                        "errorCode": code,
+                        "sqlState": e.sqlstate,
+                    },
+                    "code": code,
+                    "message": e.msg,
+                    "success": False,
+                }
+            )
 
         rowtype = describe_as_rowtype(cur._describe_last_sql())  # noqa: SLF001
 

tests/test_server.py

@@ -111,3 +111,14 @@ def test_server_abort_request(server: dict) -> None:
         conn1.cursor() as cur,
     ):
         cur.execute("select 'will abort'")
+
+
+def test_server_errors(scur: snowflake.connector.cursor.SnowflakeCursor) -> None:
+    cur = scur
+    with pytest.raises(snowflake.connector.errors.ProgrammingError) as excinfo:
+        cur.execute("select * from this_table_does_not_exist")
+
+    assert excinfo.value.errno == 2003
+    assert excinfo.value.sqlstate == "42S02"
+    assert excinfo.value.msg
+    assert "THIS_TABLE_DOES_NOT_EXIST" in excinfo.value.msg