def test_error_message_encoding(pgconn):
res = pgconn.exec_(b"set client_encoding to latin9")
assert res.status == pq.ExecStatus.COMMAND_OK
res = pgconn.exec_('select 1 from "foo\u20acbar"'.encode("latin9"))
assert res.status == pq.ExecStatus.FATAL_ERROR
msg = pq.error_message(pgconn)
assert "foo\u20acbar" in msg
msg = pq.error_message(res)
assert "foo\ufffdbar" in msg
msg = pq.error_message(res, encoding="latin9")
assert "foo\u20acbar" in msg
msg = pq.error_message(res, encoding="ascii")
assert "foo\ufffdbar" in msg
def error_from_result(result: PGresult, encoding: str = "utf-8") -> Error:
from psycopg import pq
state = result.error_field(DiagnosticField.SQLSTATE) or b""
cls = _class_for_state(state.decode("ascii"))
return cls(
pq.error_message(result, encoding=encoding),
info=result,
encoding=encoding,
)
def test_error_message(pgconn):
res = pgconn.exec_(b"wat")
assert res.status == pq.ExecStatus.FATAL_ERROR
msg = pq.error_message(pgconn)
assert msg == 'syntax error at or near "wat"\nLINE 1: wat\n ^'
assert msg == pq.error_message(res)
primary = res.error_field(pq.DiagnosticField.MESSAGE_PRIMARY)
assert primary.decode("ascii") in msg
with pytest.raises(TypeError):
pq.error_message(None)
res.clear()
assert pq.error_message(res) == "no details available"
pgconn.finish()
assert "NULL" in pq.error_message(pgconn)