From b5812a7f2f53e61d6d794533ae9d850429e57f51 Mon Sep 17 00:00:00 2001 From: Max Date: Sat, 25 Oct 2025 16:43:28 +0800 Subject: [PATCH] Add Neo loading in test preparation - Integrated Neo loading in the Prepare function of testutils.go to ensure the Neo environment is initialized before running tests. - Updated error handling to log failures in loading Neo, enhancing test reliability and feedback. - Adjusted variable assignment for Knowledge Base loading to maintain consistency in error handling. --- openapi/tests/testutils/testutils.go | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/openapi/tests/testutils/testutils.go b/openapi/tests/testutils/testutils.go index b2dbedb3..7ccb8e3c 100644 --- a/openapi/tests/testutils/testutils.go +++ b/openapi/tests/testutils/testutils.go @@ -16,6 +16,7 @@ import ( "github.com/gin-gonic/gin" "github.com/yaoapp/yao/config" "github.com/yaoapp/yao/kb" + "github.com/yaoapp/yao/neo" "github.com/yaoapp/yao/openapi" "github.com/yaoapp/yao/openapi/oauth" "github.com/yaoapp/yao/openapi/oauth/types" @@ -153,8 +154,14 @@ func Prepare(t *testing.T) string { // Step 1: Initialize base test environment with all Yao dependencies test.Prepare(t, config.Conf) + // Load Neo + err := neo.Load(config.Conf) + if err != nil { + t.Fatalf("Failed to load Neo: %v", err) + } + // Step 1.5: Initialize Knowledge Base (must be done before OpenAPI load) - _, err := kb.Load(config.Conf) + _, err = kb.Load(config.Conf) if err != nil { // KB loading failure is not fatal for tests, just log it t.Logf("Warning: Failed to load Knowledge Base: %v", err)