zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-08-15 13:14:28 +08:00
Code Issues Packages Projects Releases Wiki Activity

Go: fix context (#18118)

Browse Source 
Signed-off-by: Jin Hai <haijin.chn@gmail.com>
This commit is contained in:
Jin Hai
2026-08-11 19:19:29 +08:00
committed by GitHub
parent 8bd5768ebc
commit c75edbfbe8
36 changed files with 492 additions and 352 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
internal/agent/component/switch_test.go
View File

@@ -17,7 +17,6 @@
package component
import (
"context"
"testing"
"ragflow/internal/agent/canvas"
@@ -56,7 +55,7 @@ func TestSwitch_AndMatches(t *testing.T) {
s, _ := NewSwitchComponent(nil)
state := canvas.NewCanvasState("run-1", "task-1")
state.Sys["x"] = "yes"
ctx := withStateForTest(context.Background(), state)
ctx := withStateForTest(t.Context(), state)
inputs := map[string]any{
"conditions": []any{
@@ -87,7 +86,7 @@ func TestSwitch_OrMatches(t *testing.T) {
state := canvas.NewCanvasState("run-2", "task-2")
state.Sys["score"] = "85"
state.Sys["flag"] = "no"
ctx := withStateForTest(context.Background(), state)
ctx := withStateForTest(t.Context(), state)
inputs := map[string]any{
"conditions": []any{
@@ -129,7 +128,7 @@ func TestSwitch_DefaultFallback(t *testing.T) {
s, _ := NewSwitchComponent(nil)
state := canvas.NewCanvasState("run-3", "task-3")
state.Sys["x"] = "no"
ctx := withStateForTest(context.Background(), state)
ctx := withStateForTest(t.Context(), state)
inputs := map[string]any{
"conditions": []any{
@@ -156,7 +155,7 @@ func TestSwitch_LegacyEndCpnIDsFallback(t *testing.T) {
s, _ := NewSwitchComponent(nil)
state := canvas.NewCanvasState("run-end-cpn", "task-end-cpn")
state.Sys["x"] = "no"
ctx := withStateForTest(context.Background(), state)
ctx := withStateForTest(t.Context(), state)
inputs := map[string]any{
"conditions": []any{
@@ -186,7 +185,7 @@ func TestSwitch_ContainsAndEmpty(t *testing.T) {
state := canvas.NewCanvasState("run-4", "task-4")
state.Sys["body"] = "hello world"
state.Sys["opt"] = ""
ctx := withStateForTest(context.Background(), state)
ctx := withStateForTest(t.Context(), state)
inputs := map[string]any{
"conditions": []any{
@@ -229,7 +228,7 @@ func TestSwitch_LegacyConditionsAndArrayTo(t *testing.T) {
})
state := canvas.NewCanvasState("run-legacy", "task-legacy")
state.SetVar("UserFillUp:Menu", "demo", "loop")
ctx := withStateForTest(context.Background(), state)
ctx := withStateForTest(t.Context(), state)
out, err := s.Invoke(ctx, nil, nil)
if err != nil {
@@ -253,7 +252,7 @@ func TestSwitch_NilUpstreamContainsEmptyNeedleMatches(t *testing.T) {
s, _ := NewSwitchComponent(nil)
state := canvas.NewCanvasState("run-nil-contains", "task-nil-contains")
state.Sys["answer"] = nil
ctx := withStateForTest(context.Background(), state)
ctx := withStateForTest(t.Context(), state)
inputs := map[string]any{
"conditions": []any{
@@ -284,7 +283,7 @@ func TestSwitch_NilUpstreamContainsNonEmptyDoesNotMatch(t *testing.T) {
s, _ := NewSwitchComponent(nil)
state := canvas.NewCanvasState("run-nil-needle", "task-nil-needle")
state.Sys["answer"] = nil
ctx := withStateForTest(context.Background(), state)
ctx := withStateForTest(t.Context(), state)
inputs := map[string]any{
"conditions": []any{
@@ -316,7 +315,7 @@ func TestSwitch_NilValueContainsDoesNotRaise(t *testing.T) {
s, _ := NewSwitchComponent(nil)
state := canvas.NewCanvasState("run-nil-value", "task-nil-value")
state.Sys["answer"] = "foobar"
ctx := withStateForTest(context.Background(), state)
ctx := withStateForTest(t.Context(), state)
inputs := map[string]any{
"conditions": []any{
@@ -350,7 +349,7 @@ func TestSwitch_NilUpstreamStartWithEndWithDoNotCrash(t *testing.T) {
s, _ := NewSwitchComponent(nil)
state := canvas.NewCanvasState("run-nil-start-end", "task-nil-start-end")
state.Sys["answer"] = nil
ctx := withStateForTest(context.Background(), state)
ctx := withStateForTest(t.Context(), state)
for _, tc := range []struct {
name string
@@ -393,7 +392,7 @@ func TestSwitch_MultiTargetTo(t *testing.T) {
s, _ := NewSwitchComponent(nil)
state := canvas.NewCanvasState("run-multi", "task-multi")
state.SetVar("UserFillUp:Menu", "demo", "data_ops")
ctx := withStateForTest(context.Background(), state)
ctx := withStateForTest(t.Context(), state)
inputs := map[string]any{
"conditions": []any{
@@ -430,7 +429,7 @@ func TestSwitch_MultiTargetTo(t *testing.T) {
func TestSwitch_EmptyAndConditionFallsThrough(t *testing.T) {
s, _ := NewSwitchComponent(nil)
state := canvas.NewCanvasState("run-empty-and", "task-1")
ctx := withStateForTest(context.Background(), state)
ctx := withStateForTest(t.Context(), state)
// Empty clauses: must not match. Should fall through to default.
inputs := map[string]any{
@@ -462,7 +461,7 @@ func TestSwitch_EmptyAndConditionFallsThrough(t *testing.T) {
func TestSwitch_LegacyEmptyItemsFallsThrough(t *testing.T) {
s, _ := NewSwitchComponent(nil)
state := canvas.NewCanvasState("run-legacy-empty", "task-1")
ctx := withStateForTest(context.Background(), state)
ctx := withStateForTest(t.Context(), state)
inputs := map[string]any{
"conditions": []any{
@@ -496,7 +495,7 @@ func TestSwitch_SatisfiedAndConditionStillRoutes(t *testing.T) {
s, _ := NewSwitchComponent(nil)
state := canvas.NewCanvasState("run-and-ok", "task-1")
state.Sys["greeting"] = "hello world"
ctx := withStateForTest(context.Background(), state)
ctx := withStateForTest(t.Context(), state)
inputs := map[string]any{
"conditions": []any{