feat: publish v1.0.0-rc.36-custom.1

service/task_plugin_view_test.go

@@ -0,0 +1,85 @@
+package service
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+
+	"github.com/QuantumNous/new-api/common"
+	"github.com/QuantumNous/new-api/model"
+)
+
+func TestBuildTaskPluginViewRewritesOnlyStructuredTaskIDFields(t *testing.T) {
+	const (
+		privateTaskID = "upstream-task-123"
+		publicTaskID  = "task_public_123"
+		resultURL     = "https://cdn.example.com/results/upstream-task-123/video.mp4"
+	)
+
+	taskData, err := common.Marshal(map[string]any{
+		"task_id": privateTaskID,
+		"id":      privateTaskID,
+		"taskId":  privateTaskID,
+		"url":     resultURL,
+		"message": "completed upstream-task-123",
+		"nested": []any{
+			map[string]any{
+				"task_id": privateTaskID,
+				"url":     resultURL,
+			},
+			privateTaskID,
+		},
+		privateTaskID: "opaque map key",
+	})
+	require.NoError(t, err)
+	task := &model.Task{
+		TaskID: publicTaskID,
+		PrivateData: model.TaskPrivateData{
+			UpstreamTaskID: privateTaskID,
+		},
+		Data: taskData,
+	}
+
+	view, err := BuildTaskPluginView(task)
+	require.NoError(t, err)
+
+	data, ok := view.Data.(map[string]any)
+	require.True(t, ok)
+	assert.Equal(t, publicTaskID, data["task_id"])
+	assert.Equal(t, publicTaskID, data["id"])
+	assert.Equal(t, publicTaskID, data["taskId"])
+	assert.Equal(t, resultURL, data["url"])
+	assert.Equal(t, "completed upstream-task-123", data["message"])
+	assert.Equal(t, "opaque map key", data[privateTaskID])
+
+	nested, ok := data["nested"].([]any)
+	require.True(t, ok)
+	nestedData, ok := nested[0].(map[string]any)
+	require.True(t, ok)
+	assert.Equal(t, publicTaskID, nestedData["task_id"])
+	assert.Equal(t, resultURL, nestedData["url"])
+	assert.Equal(t, privateTaskID, nested[1])
+
+}
+
+func TestBuildTaskPluginViewOmitsPrivatePollState(t *testing.T) {
+	task := &model.Task{
+		TaskID: "task_public_view",
+		Data:   []byte(`{"ok":true}`),
+		PrivateData: model.TaskPrivateData{
+			PluginState:  []byte(`{"req_key":"secret"}`),
+			PollFailures: 7,
+		},
+	}
+
+	view, err := BuildTaskPluginView(task)
+	require.NoError(t, err)
+	encoded, err := common.Marshal(view)
+	require.NoError(t, err)
+	var payload map[string]any
+	require.NoError(t, common.Unmarshal(encoded, &payload))
+	assert.NotContains(t, payload, "plugin_state")
+	assert.NotContains(t, payload, "poll_failures")
+	assert.NotContains(t, payload, "private_data")
+}