Function TestHumanInTheLoopMiddleware_BasicApproval

pkg/middleware/hitl_test.go:48–143

TestHumanInTheLoopMiddleware_BasicApproval 测试基本审核流程

(t *testing.T)

// TestHumanInTheLoopMiddleware_BasicApproval 测试基本审核流程
func TestHumanInTheLoopMiddleware_BasicApproval(t *testing.T) {
	ctx := context.Background()

	// 创建中间件
	middleware, err := NewHumanInTheLoopMiddleware(&HumanInTheLoopMiddlewareConfig{
		InterruptOn: map[string]any{
			"sensitive_tool": true,  // 需要审核
			"safe_tool":      false, // 不需要审核
		},
		ApprovalHandler: func(ctx context.Context, request *ReviewRequest) ([]Decision, error) {
			// 自动批准
			decisions := make([]Decision, len(request.ActionRequests))
			for i := range decisions {
				decisions[i] = Decision{
					Type:   DecisionApprove,
					Reason: "Test approval",
				}
			}
			return decisions, nil
		},
	})
	if err != nil {
		t.Fatalf("Failed to create middleware: %v", err)
	}

	// 测试1: sensitive_tool 需要审核,但被批准
	t.Run("Approve sensitive tool", func(t *testing.T) {
		req := &ToolCallRequest{
			ToolCallID: "test-1",
			ToolName:   "sensitive_tool",
			ToolInput: map[string]any{
				"param": "value1",
			},
			Tool: &mockTool{name: "sensitive_tool"},
		}

		handler := func(ctx context.Context, req *ToolCallRequest) (*ToolCallResponse, error) {
			// 执行工具
			result, _ := req.Tool.Execute(ctx, req.ToolInput, nil)
			return &ToolCallResponse{
				Result: result,
			}, nil
		}

		resp, err := middleware.WrapToolCall(ctx, req, handler)
		if err != nil {
			t.Fatalf("WrapToolCall failed: %v", err)
		}

		resultMap, ok := resp.Result.(map[string]any)
		if !ok {
			t.Fatal("Result is not a map")
		}

		if !resultMap["ok"].(bool) {
			t.Errorf("Expected ok=true, got ok=false")
		}