Sub2API v1.0 - AI API 网关（二开初始版本，基于上游 Wei-Shaw/sub2api）

backend/internal/service/ops_query_mode_test.go

@@ -0,0 +1,66 @@
+//go:build unit
+
+package service
+
+import (
+	"errors"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestShouldFallbackOpsPreagg(t *testing.T) {
+	preaggErr := ErrOpsPreaggregatedNotPopulated
+	otherErr := errors.New("some other error")
+
+	autoFilter := &OpsDashboardFilter{QueryMode: OpsQueryModeAuto}
+	rawFilter := &OpsDashboardFilter{QueryMode: OpsQueryModeRaw}
+	preaggFilter := &OpsDashboardFilter{QueryMode: OpsQueryModePreagg}
+
+	tests := []struct {
+		name   string
+		filter *OpsDashboardFilter
+		err    error
+		want   bool
+	}{
+		{"auto mode + preagg error => fallback", autoFilter, preaggErr, true},
+		{"auto mode + other error => no fallback", autoFilter, otherErr, false},
+		{"auto mode + nil error => no fallback", autoFilter, nil, false},
+		{"raw mode + preagg error => no fallback", rawFilter, preaggErr, false},
+		{"preagg mode + preagg error => no fallback", preaggFilter, preaggErr, false},
+		{"nil filter => no fallback", nil, preaggErr, false},
+		{"wrapped preagg error => fallback", autoFilter, errors.Join(preaggErr, otherErr), true},
+	}
+
+	for _, tc := range tests {
+		t.Run(tc.name, func(t *testing.T) {
+			got := shouldFallbackOpsPreagg(tc.filter, tc.err)
+			require.Equal(t, tc.want, got)
+		})
+	}
+}
+
+func TestCloneOpsFilterWithMode(t *testing.T) {
+	t.Run("nil filter returns nil", func(t *testing.T) {
+		require.Nil(t, cloneOpsFilterWithMode(nil, OpsQueryModeRaw))
+	})
+
+	t.Run("cloned filter has new mode", func(t *testing.T) {
+		groupID := int64(42)
+		original := &OpsDashboardFilter{
+			StartTime: time.Now(),
+			EndTime:   time.Now().Add(time.Hour),
+			Platform:  "anthropic",
+			GroupID:   &groupID,
+			QueryMode: OpsQueryModeAuto,
+		}
+
+		cloned := cloneOpsFilterWithMode(original, OpsQueryModeRaw)
+		require.Equal(t, OpsQueryModeRaw, cloned.QueryMode)
+		require.Equal(t, OpsQueryModeAuto, original.QueryMode, "original should not be modified")
+		require.Equal(t, original.Platform, cloned.Platform)
+		require.Equal(t, original.StartTime, cloned.StartTime)
+		require.Equal(t, original.GroupID, cloned.GroupID)
+	})
+}