From d857e80fffe685cd2b9611f13f7d0b88510fd6ab Mon Sep 17 00:00:00 2001 From: duwenxin Date: Tue, 10 Feb 2026 10:08:54 -0500 Subject: [PATCH] serverless spark --- .../serverless_spark_integration_test.go | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) diff --git a/tests/serverlessspark/serverless_spark_integration_test.go b/tests/serverlessspark/serverless_spark_integration_test.go index 777b012bc6..faf9b30bb6 100644 --- a/tests/serverlessspark/serverless_spark_integration_test.go +++ b/tests/serverlessspark/serverless_spark_integration_test.go @@ -1017,13 +1017,14 @@ func testError(t *testing.T, toolName string, request map[string]any, wantCode i t.Fatalf("failed to unmarshal outer response: %v", err) } - resultStr, ok := body["result"].(string) - if !ok { - if errMsg, ok := body["error"].(string); ok { - resultStr = errMsg - } else { - t.Fatalf("unable to find result string in response: %s", string(bodyBytes)) - } + var resultStr string + if res, ok := body["result"].(string); ok { + resultStr = res + } else if errMsg, ok := body["error"].(string); ok { + resultStr = errMsg + } else { + // If neither exists, check the raw bytes as a last resort + resultStr = string(bodyBytes) } if !strings.Contains(resultStr, wantMsg) {