38 lines
1013 B
Go
38 lines
1013 B
Go
package repository
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/Wei-Shaw/sub2api/internal/service"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestFilterSchedulerCredentialsKeepsSubscriptionPlanType(t *testing.T) {
|
|
filtered := filterSchedulerCredentials(map[string]any{
|
|
"plan_type": "plus",
|
|
"access_token": "secret-access-token",
|
|
"refresh_token": "secret-refresh-token",
|
|
})
|
|
|
|
require.Equal(t, "plus", filtered["plan_type"])
|
|
require.NotContains(t, filtered, "access_token")
|
|
require.NotContains(t, filtered, "refresh_token")
|
|
}
|
|
|
|
func TestSchedulerMetadataAccountKeepsOpenAISubscriptionIdentity(t *testing.T) {
|
|
account := service.Account{
|
|
ID: 24,
|
|
Platform: service.PlatformOpenAI,
|
|
Type: service.AccountTypeOAuth,
|
|
Credentials: map[string]any{
|
|
"plan_type": "plus",
|
|
"access_token": "secret-access-token",
|
|
},
|
|
}
|
|
|
|
metadata := buildSchedulerMetadataAccount(account)
|
|
|
|
require.True(t, metadata.IsOpenAIChatGPTSubscription())
|
|
require.Empty(t, metadata.GetCredential("access_token"))
|
|
}
|