mirror of
https://github.com/coder/coder.git
synced 2025-07-08 11:39:50 +00:00
118 lines
3.0 KiB
Go
118 lines
3.0 KiB
Go
package coderd
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/coder/coder/v2/codersdk"
|
|
)
|
|
|
|
func TestApplyIDPSyncMappingDiff(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
t.Run("with UUIDs", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
id := []uuid.UUID{
|
|
uuid.MustParse("00000000-b8bd-46bb-bb6c-6c2b2c0dd2ea"),
|
|
uuid.MustParse("01000000-fbe8-464c-9429-fe01a03f3644"),
|
|
uuid.MustParse("02000000-0926-407b-9998-39af62e3d0c5"),
|
|
uuid.MustParse("03000000-92f6-4bfd-bba6-0f54667b131c"),
|
|
}
|
|
|
|
mapping := applyIDPSyncMappingDiff(map[string][]uuid.UUID{},
|
|
[]codersdk.IDPSyncMapping[uuid.UUID]{
|
|
{Given: "wibble", Gets: id[0]},
|
|
{Given: "wibble", Gets: id[1]},
|
|
{Given: "wobble", Gets: id[0]},
|
|
{Given: "wobble", Gets: id[1]},
|
|
{Given: "wobble", Gets: id[2]},
|
|
{Given: "wobble", Gets: id[3]},
|
|
{Given: "wooble", Gets: id[0]},
|
|
},
|
|
// Remove takes priority over Add, so `3` should not actually be added.
|
|
[]codersdk.IDPSyncMapping[uuid.UUID]{
|
|
{Given: "wobble", Gets: id[3]},
|
|
},
|
|
)
|
|
|
|
expected := map[string][]uuid.UUID{
|
|
"wibble": {id[0], id[1]},
|
|
"wobble": {id[0], id[1], id[2]},
|
|
"wooble": {id[0]},
|
|
}
|
|
|
|
require.Equal(t, expected, mapping)
|
|
|
|
mapping = applyIDPSyncMappingDiff(mapping,
|
|
[]codersdk.IDPSyncMapping[uuid.UUID]{
|
|
{Given: "wibble", Gets: id[2]},
|
|
{Given: "wobble", Gets: id[3]},
|
|
{Given: "wooble", Gets: id[0]},
|
|
},
|
|
[]codersdk.IDPSyncMapping[uuid.UUID]{
|
|
{Given: "wibble", Gets: id[0]},
|
|
{Given: "wobble", Gets: id[1]},
|
|
},
|
|
)
|
|
|
|
expected = map[string][]uuid.UUID{
|
|
"wibble": {id[1], id[2]},
|
|
"wobble": {id[0], id[2], id[3]},
|
|
"wooble": {id[0]},
|
|
}
|
|
|
|
require.Equal(t, expected, mapping)
|
|
})
|
|
|
|
t.Run("with strings", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
mapping := applyIDPSyncMappingDiff(map[string][]string{},
|
|
[]codersdk.IDPSyncMapping[string]{
|
|
{Given: "wibble", Gets: "group-00"},
|
|
{Given: "wibble", Gets: "group-01"},
|
|
{Given: "wobble", Gets: "group-00"},
|
|
{Given: "wobble", Gets: "group-01"},
|
|
{Given: "wobble", Gets: "group-02"},
|
|
{Given: "wobble", Gets: "group-03"},
|
|
{Given: "wooble", Gets: "group-00"},
|
|
},
|
|
// Remove takes priority over Add, so `3` should not actually be added.
|
|
[]codersdk.IDPSyncMapping[string]{
|
|
{Given: "wobble", Gets: "group-03"},
|
|
},
|
|
)
|
|
|
|
expected := map[string][]string{
|
|
"wibble": {"group-00", "group-01"},
|
|
"wobble": {"group-00", "group-01", "group-02"},
|
|
"wooble": {"group-00"},
|
|
}
|
|
|
|
require.Equal(t, expected, mapping)
|
|
|
|
mapping = applyIDPSyncMappingDiff(mapping,
|
|
[]codersdk.IDPSyncMapping[string]{
|
|
{Given: "wibble", Gets: "group-02"},
|
|
{Given: "wobble", Gets: "group-03"},
|
|
{Given: "wooble", Gets: "group-00"},
|
|
},
|
|
[]codersdk.IDPSyncMapping[string]{
|
|
{Given: "wibble", Gets: "group-00"},
|
|
{Given: "wobble", Gets: "group-01"},
|
|
},
|
|
)
|
|
|
|
expected = map[string][]string{
|
|
"wibble": {"group-01", "group-02"},
|
|
"wobble": {"group-00", "group-02", "group-03"},
|
|
"wooble": {"group-00"},
|
|
}
|
|
|
|
require.Equal(t, expected, mapping)
|
|
})
|
|
}
|