mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-10-14 12:02:27 +08:00
88 lines
3.3 KiB
Go
88 lines
3.3 KiB
Go
package handler
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/Wei-Shaw/sub2api/internal/server/middleware"
|
|
"github.com/Wei-Shaw/sub2api/internal/service"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
type channelMonitorV2GroupAuthorizerStub struct {
|
|
groups []service.Group
|
|
err error
|
|
calls []int64
|
|
}
|
|
|
|
func (s *channelMonitorV2GroupAuthorizerStub) GetAvailableGroups(_ context.Context, userID int64) ([]service.Group, error) {
|
|
s.calls = append(s.calls, userID)
|
|
return s.groups, s.err
|
|
}
|
|
|
|
func TestChannelMonitorV2QueryListSupportsRepeatedAndCommaValues(t *testing.T) {
|
|
c, _ := gin.CreateTestContext(nil)
|
|
c.Request = httptest.NewRequest("GET", "/?platform=openai,grok&platform=anthropic", nil)
|
|
require.Equal(t, []string{"openai", "grok", "anthropic"}, queryList(c, "platform"))
|
|
}
|
|
|
|
func TestChannelMonitorV2GroupByQueryDefaultsAndRejectsInvalid(t *testing.T) {
|
|
groupBy, err := service.ParseChannelMonitorV2GroupBy("")
|
|
require.NoError(t, err)
|
|
require.Equal(t, service.ChannelMonitorV2GroupByPlatformGroup, groupBy)
|
|
_, err = service.ParseChannelMonitorV2GroupBy("invalid")
|
|
require.Error(t, err)
|
|
}
|
|
|
|
func TestChannelMonitorV2MatrixHandlerRejectsInvalidGroupBy(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
recorder := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(recorder)
|
|
c.Request = httptest.NewRequest(http.MethodGet, "/channel-monitor-v2/matrix?group_by=invalid", nil)
|
|
h := NewChannelMonitorV2Handler(service.NewChannelMonitorV2Service(nil), nil)
|
|
h.Matrix(c)
|
|
require.Equal(t, http.StatusBadRequest, recorder.Code)
|
|
}
|
|
|
|
func TestChannelMonitorV2ScopeFilterUsesAvailableGroupsForOrdinaryUser(t *testing.T) {
|
|
recorder := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(recorder)
|
|
c.Request = httptest.NewRequest(http.MethodGet, "/channel-monitor-v2/snapshot", nil)
|
|
c.Set(string(middleware.ContextKeyUser), middleware.AuthSubject{UserID: 42})
|
|
authorizer := &channelMonitorV2GroupAuthorizerStub{groups: []service.Group{{ID: 3}, {ID: 7}}}
|
|
h := &ChannelMonitorV2Handler{apiKeyService: authorizer}
|
|
filter := service.ChannelMonitorV2Filter{GroupIDs: []int64{7, 9}}
|
|
|
|
require.True(t, h.scopeFilter(c, &filter, false))
|
|
require.True(t, filter.RestrictGroups)
|
|
require.Equal(t, []int64{3, 7}, filter.AllowedGroupIDs)
|
|
require.Equal(t, []int64{42}, authorizer.calls)
|
|
}
|
|
|
|
func TestChannelMonitorV2ScopeFilterPreservesEmptyOrdinaryUserScope(t *testing.T) {
|
|
recorder := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(recorder)
|
|
c.Request = httptest.NewRequest(http.MethodGet, "/channel-monitor-v2/snapshot", nil)
|
|
c.Set(string(middleware.ContextKeyUser), middleware.AuthSubject{UserID: 42})
|
|
h := &ChannelMonitorV2Handler{apiKeyService: &channelMonitorV2GroupAuthorizerStub{}}
|
|
filter := service.ChannelMonitorV2Filter{}
|
|
|
|
require.True(t, h.scopeFilter(c, &filter, false))
|
|
require.True(t, filter.RestrictGroups)
|
|
require.Empty(t, filter.AllowedGroupIDs)
|
|
}
|
|
|
|
func TestChannelMonitorV2ScopeFilterLeavesAdminUnrestricted(t *testing.T) {
|
|
authorizer := &channelMonitorV2GroupAuthorizerStub{}
|
|
h := &ChannelMonitorV2Handler{apiKeyService: authorizer}
|
|
filter := service.ChannelMonitorV2Filter{GroupIDs: []int64{9}}
|
|
|
|
require.True(t, h.scopeFilter(nil, &filter, true))
|
|
require.False(t, filter.RestrictGroups)
|
|
require.Nil(t, filter.AllowedGroupIDs)
|
|
require.Empty(t, authorizer.calls)
|
|
}
|