opensource/sub2api
3
0
Fork 0
mirror of https://github.com/Wei-Shaw/sub2api.git synced 2026-09-24 16:05:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: test

Browse Source
This commit is contained in:
xueshiji
2026-07-01 18:09:20 +08:00
parent a0eaf12a01
commit be26e6ff2c
2 changed files with 2 additions and 107 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/internal/server/api_contract_test.go
+2 -1
View File
@@ -317,6 +317,7 @@ func TestAPIContracts(t *testing.T) {
Description: "desc",
Platform: service.PlatformAnthropic,
RateMultiplier: 1.5,
PeakRateMultiplier: 1.0,
IsExclusive: false,
Status: service.StatusActive,
SubscriptionType: service.SubscriptionTypeStandard,
@@ -347,7 +348,7 @@ func TestAPIContracts(t *testing.T) {
"peak_rate_enabled": false,
"peak_start": "",
"peak_end": "",
"peak_rate_multiplier": 0,
"peak_rate_multiplier": 1,
"is_exclusive": false,
"status": "active",
"subscription_type": "standard",
backend/internal/service/admin_service_peak_rate_test.go
-106
View File
@@ -1,106 +0,0 @@
package service
import (
"context"
"testing"
"github.com/Wei-Shaw/sub2api/internal/pkg/pagination"
"github.com/stretchr/testify/require"
)
type peakRateGroupRepoStub struct {
getByID *Group
updated *Group
}
func (s *peakRateGroupRepoStub) Create(context.Context, *Group) error {
panic("unexpected Create call")
}
func (s *peakRateGroupRepoStub) GetByID(context.Context, int64) (*Group, error) {
return s.getByID, nil
}
func (s *peakRateGroupRepoStub) GetByIDLite(context.Context, int64) (*Group, error) {
return s.getByID, nil
}
func (s *peakRateGroupRepoStub) Update(_ context.Context, group *Group) error {
s.updated = group
return nil
}
func (s *peakRateGroupRepoStub) Delete(context.Context, int64) error {
panic("unexpected Delete call")
}
func (s *peakRateGroupRepoStub) DeleteCascade(context.Context, int64) ([]int64, error) {
panic("unexpected DeleteCascade call")
}
func (s *peakRateGroupRepoStub) List(context.Context, pagination.PaginationParams) ([]Group, *pagination.PaginationResult, error) {
panic("unexpected List call")
}
func (s *peakRateGroupRepoStub) ListWithFilters(context.Context, pagination.PaginationParams, string, string, string, *bool) ([]Group, *pagination.PaginationResult, error) {
panic("unexpected ListWithFilters call")
}
func (s *peakRateGroupRepoStub) ListActive(context.Context) ([]Group, error) {
panic("unexpected ListActive call")
}
func (s *peakRateGroupRepoStub) ListActiveByPlatform(context.Context, string) ([]Group, error) {
panic("unexpected ListActiveByPlatform call")
}
func (s *peakRateGroupRepoStub) ExistsByName(context.Context, string) (bool, error) {
panic("unexpected ExistsByName call")
}
func (s *peakRateGroupRepoStub) GetAccountCount(context.Context, int64) (int64, int64, error) {
panic("unexpected GetAccountCount call")
}
func (s *peakRateGroupRepoStub) DeleteAccountGroupsByGroupID(context.Context, int64) (int64, error) {
panic("unexpected DeleteAccountGroupsByGroupID call")
}
func (s *peakRateGroupRepoStub) GetAccountIDsByGroupIDs(context.Context, []int64) ([]int64, error) {
panic("unexpected GetAccountIDsByGroupIDs call")
}
func (s *peakRateGroupRepoStub) BindAccountsToGroup(context.Context, int64, []int64) error {
panic("unexpected BindAccountsToGroup call")
}
func (s *peakRateGroupRepoStub) UpdateSortOrders(context.Context, []GroupSortOrderUpdate) error {
panic("unexpected UpdateSortOrders call")
}
func TestAdminService_UpdateGroup_ClearsPeakRateWhenChangingToStandardDefault(t *testing.T) {
repo := &peakRateGroupRepoStub{getByID: &Group{
ID: 1,
Name: "existing-group",
Platform: PlatformOpenAI,
Status: StatusActive,
SubscriptionType: SubscriptionTypeSubscription,
PeakRateEnabled: true,
PeakStart: "14:00",
PeakEnd: "18:00",
PeakRateMultiplier: 3,
}}
svc := &adminServiceImpl{groupRepo: repo}
group, err := svc.UpdateGroup(context.Background(), 1, &UpdateGroupInput{
SubscriptionType: SubscriptionTypeStandard,
})
require.NoError(t, err)
require.NotNil(t, group)
require.NotNil(t, repo.updated)
require.Equal(t, SubscriptionTypeStandard, repo.updated.SubscriptionType)
require.False(t, repo.updated.PeakRateEnabled)
require.Equal(t, "", repo.updated.PeakStart)
require.Equal(t, "", repo.updated.PeakEnd)
require.Equal(t, 1.0, repo.updated.PeakRateMultiplier)
}