mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-10-15 23:22:59 +08:00
26 lines
805 B
Go
26 lines
805 B
Go
package dto
|
|
|
|
import (
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"github.com/Wei-Shaw/sub2api/internal/service"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestGroupMapperExposesForceOpenAIFastOnlyToAdmins(t *testing.T) {
|
|
group := &service.Group{
|
|
ID: 7, Name: "fast", Platform: service.PlatformOpenAI, Status: service.StatusActive, ForceOpenAIFast: true, FreeOpenAIFast: true,
|
|
}
|
|
|
|
userJSON, err := json.Marshal(GroupFromService(group))
|
|
require.NoError(t, err)
|
|
require.NotContains(t, string(userJSON), "force_openai_fast")
|
|
require.NotContains(t, string(userJSON), "free_openai_fast")
|
|
|
|
adminJSON, err := json.Marshal(GroupFromServiceAdmin(group))
|
|
require.NoError(t, err)
|
|
require.Contains(t, string(adminJSON), `"force_openai_fast":true`)
|
|
require.Contains(t, string(adminJSON), `"free_openai_fast":true`)
|
|
}
|