From f3778e393131994a76a0c16b0862727a205abcd2 Mon Sep 17 00:00:00 2001
From: fengsilin <claude@anthropic.com>
Date: Wed, 17 Jun 2026 18:39:32 +0800
Subject: [PATCH] test(task-pricing): cover doubao usage parsing

---
 relay/channel/task/doubao/adaptor_test.go | 50 +++++++++++++++++++++++
 1 file changed, 50 insertions(+)
 create mode 100644 relay/channel/task/doubao/adaptor_test.go

diff --git a/relay/channel/task/doubao/adaptor_test.go b/relay/channel/task/doubao/adaptor_test.go
new file mode 100644
index 0000000..396f4d3
--- /dev/null
+++ b/relay/channel/task/doubao/adaptor_test.go
@@ -0,0 +1,50 @@
+package doubao
+
+import (
+	"testing"
+
+	"github.com/QuantumNous/new-api/model"
+	"github.com/stretchr/testify/require"
+)
+
+func TestParseTaskResult_SuccessWithUsage(t *testing.T) {
+	adaptor := &TaskAdaptor{}
+	taskInfo, err := adaptor.ParseTaskResult([]byte(`{
+		"status":"succeeded",
+		"content":{"video_url":"https://example.test/video.mp4"},
+		"usage":{"completion_tokens":12,"total_tokens":3456}
+	}`))
+
+	require.NoError(t, err)
+	require.Equal(t, string(model.TaskStatusSuccess), taskInfo.Status)
+	require.Equal(t, "https://example.test/video.mp4", taskInfo.Url)
+	require.Equal(t, 12, taskInfo.CompletionTokens)
+	require.Equal(t, 3456, taskInfo.TotalTokens)
+}
+
+func TestParseTaskResult_SuccessWithoutUsage(t *testing.T) {
+	adaptor := &TaskAdaptor{}
+	taskInfo, err := adaptor.ParseTaskResult([]byte(`{
+		"status":"succeeded",
+		"content":{"video_url":"https://example.test/video.mp4"}
+	}`))
+
+	require.NoError(t, err)
+	require.Equal(t, string(model.TaskStatusSuccess), taskInfo.Status)
+	require.Zero(t, taskInfo.TotalTokens)
+}
+
+func TestParseTaskResult_NonSuccessStatusesDoNotRequireUsage(t *testing.T) {
+	adaptor := &TaskAdaptor{}
+	cases := map[string]model.TaskStatus{
+		"queued":     model.TaskStatusQueued,
+		"processing": model.TaskStatusInProgress,
+		"failed":     model.TaskStatusFailure,
+	}
+	for status, want := range cases {
+		taskInfo, err := adaptor.ParseTaskResult([]byte(`{"status":"` + status + `"}`))
+		require.NoError(t, err)
+		require.Equal(t, string(want), taskInfo.Status)
+		require.Zero(t, taskInfo.TotalTokens)
+	}
+}