Function TestHumanInTheLoopMiddleware_Edit

pkg/middleware/hitl_test.go:207–267

TestHumanInTheLoopMiddleware_Edit 测试编辑参数

(t *testing.T)

// TestHumanInTheLoopMiddleware_Edit 测试编辑参数
func TestHumanInTheLoopMiddleware_Edit(t *testing.T) {
	ctx := context.Background()

	middleware, err := NewHumanInTheLoopMiddleware(&HumanInTheLoopMiddlewareConfig{
		InterruptOn: map[string]any{
			"editable_tool": true,
		},
		ApprovalHandler: func(ctx context.Context, request *ReviewRequest) ([]Decision, error) {
			// 编辑参数
			editedInput := make(map[string]any)
			maps.Copy(editedInput, request.ActionRequests[0].Input)
			editedInput["param"] = "edited_value"

			return []Decision{
				{
					Type:        DecisionEdit,
					EditedInput: editedInput,
					Reason:      "Parameter adjusted",
				},
			}, nil
		},
	})
	if err != nil {
		t.Fatalf("Failed to create middleware: %v", err)
	}

	req := &ToolCallRequest{
		ToolCallID: "test-edit",
		ToolName:   "editable_tool",
		ToolInput: map[string]any{
			"param": "original_value",
		},
		Tool: &mockTool{name: "editable_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.Error("Expected ok=true")
	}

	// 验证参数已被编辑
	if resultMap["param"] != "edited_value" {