106 lines
2.6 KiB
Go
106 lines
2.6 KiB
Go
package utils
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"goauthentik.io/api/v3"
|
|
)
|
|
|
|
func Test_stringify_nil(t *testing.T) {
|
|
var ex *string
|
|
assert.Equal(t, ex, stringify(nil))
|
|
}
|
|
|
|
func TestAKAttrsToLDAP_String(t *testing.T) {
|
|
u := api.User{}
|
|
|
|
// normal string
|
|
u.Attributes = map[string]interface{}{
|
|
"foo": "bar",
|
|
}
|
|
mapped := AttributesToLDAP(u.Attributes, func(key string) string {
|
|
return AttributeKeySanitize(key)
|
|
}, func(value []string) []string {
|
|
return value
|
|
})
|
|
assert.Equal(t, 1, len(mapped))
|
|
assert.Equal(t, "foo", mapped[0].Name)
|
|
assert.Equal(t, []string{"bar"}, mapped[0].Values)
|
|
// pointer string
|
|
u.Attributes = map[string]interface{}{
|
|
"foo": api.PtrString("bar"),
|
|
}
|
|
mapped = AttributesToLDAP(u.Attributes, func(key string) string {
|
|
return AttributeKeySanitize(key)
|
|
}, func(value []string) []string {
|
|
return value
|
|
})
|
|
assert.Equal(t, 1, len(mapped))
|
|
assert.Equal(t, "foo", mapped[0].Name)
|
|
assert.Equal(t, []string{"bar"}, mapped[0].Values)
|
|
}
|
|
|
|
func TestAKAttrsToLDAP_String_List(t *testing.T) {
|
|
u := api.User{}
|
|
// string list
|
|
u.Attributes = map[string]interface{}{
|
|
"foo": []string{"bar"},
|
|
}
|
|
mapped := AttributesToLDAP(u.Attributes, func(key string) string {
|
|
return AttributeKeySanitize(key)
|
|
}, func(value []string) []string {
|
|
return value
|
|
})
|
|
assert.Equal(t, 1, len(mapped))
|
|
assert.Equal(t, "foo", mapped[0].Name)
|
|
assert.Equal(t, []string{"bar"}, mapped[0].Values)
|
|
// pointer string list
|
|
u.Attributes = map[string]interface{}{
|
|
"foo": &[]string{"bar"},
|
|
}
|
|
mapped = AttributesToLDAP(u.Attributes, func(key string) string {
|
|
return AttributeKeySanitize(key)
|
|
}, func(value []string) []string {
|
|
return value
|
|
})
|
|
assert.Equal(t, 1, len(mapped))
|
|
assert.Equal(t, "foo", mapped[0].Name)
|
|
assert.Equal(t, []string{"bar"}, mapped[0].Values)
|
|
}
|
|
|
|
func TestAKAttrsToLDAP_Dict(t *testing.T) {
|
|
// dict
|
|
d := map[string]interface{}{
|
|
"foo": map[string]string{
|
|
"foo": "bar",
|
|
},
|
|
}
|
|
mapped := AttributesToLDAP(d, func(key string) string {
|
|
return AttributeKeySanitize(key)
|
|
}, func(value []string) []string {
|
|
return value
|
|
})
|
|
assert.Equal(t, 1, len(mapped))
|
|
assert.Equal(t, "foo", mapped[0].Name)
|
|
assert.Equal(t, []string{"map[foo:bar]"}, mapped[0].Values)
|
|
}
|
|
|
|
func TestAKAttrsToLDAP_Mixed(t *testing.T) {
|
|
// dict
|
|
d := map[string]interface{}{
|
|
"foo": []interface{}{
|
|
"foo",
|
|
6,
|
|
},
|
|
}
|
|
mapped := AttributesToLDAP(d, func(key string) string {
|
|
return AttributeKeySanitize(key)
|
|
}, func(value []string) []string {
|
|
return value
|
|
})
|
|
assert.Equal(t, 1, len(mapped))
|
|
assert.Equal(t, "foo", mapped[0].Name)
|
|
assert.Equal(t, []string{"foo", "6"}, mapped[0].Values)
|
|
}
|