23 lines
507 B
Go
23 lines
507 B
Go
|
package oauth
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/QuantumNous/new-api/setting/system_setting"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestOIDCProvider_GetName(t *testing.T) {
|
||
|
|
settings := system_setting.GetOIDCSettings()
|
||
|
|
originalDisplayName := settings.DisplayName
|
||
|
|
defer func() { settings.DisplayName = originalDisplayName }()
|
||
|
|
|
||
|
|
p := &OIDCProvider{}
|
||
|
|
|
||
|
|
settings.DisplayName = ""
|
||
|
|
assert.Equal(t, "OIDC", p.GetName())
|
||
|
|
|
||
|
|
settings.DisplayName = " Acme SSO "
|
||
|
|
assert.Equal(t, "Acme SSO", p.GetName())
|
||
|
|
}
|