From 8d40e8bef752b44736421fa00600283081894a4a Mon Sep 17 00:00:00 2001
From: Robsdedude <rouven.bauer@neo4j.com>
Date: Fri, 4 Oct 2024 12:32:19 +0200
Subject: [PATCH] TestKit: serialize Neo4j error messages (#604)

---
 testkit-backend/backend.go | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/testkit-backend/backend.go b/testkit-backend/backend.go
index 604becdd..2b446461 100644
--- a/testkit-backend/backend.go
+++ b/testkit-backend/backend.go
@@ -218,11 +218,20 @@ func (b *backend) writeError(err error) {
 		neo4j.IsTransactionExecutionLimit(err)
 
 	if isDriverError {
+		var msg, errorType string
+		if neo4jError, ok := err.(*neo4j.Neo4jError); ok {
+			msg = neo4jError.Msg
+			errorType = "Neo4jError"
+		} else {
+			msg = err.Error()
+			errorType = strings.Split(err.Error(), ":")[0]
+		}
+
 		id := b.setError(err)
 		b.writeResponse("DriverError", map[string]any{
 			"id":        id,
-			"errorType": strings.Split(err.Error(), ":")[0],
-			"msg":       err.Error(),
+			"errorType": errorType,
+			"msg":       msg,
 			"code":      code,
 			"retryable": retriable,
 		})