32 lines
906 B
Go
32 lines
906 B
Go
|
package userstore
|
||
|
|
|
||
|
|
import (
|
||
|
|
"reflect"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestAttachAllowedLibraries(t *testing.T) {
|
||
|
|
profiles := []Profile{
|
||
|
|
{ID: "p1", AllowedLibraryIDs: []int{99}},
|
||
|
|
{ID: "p2", AllowedLibraryIDs: []int{99}},
|
||
|
|
{ID: "p3", AllowedLibraryIDs: []int{99}},
|
||
|
|
}
|
||
|
|
allowedLibraries := []ProfileAllowedLibrary{
|
||
|
|
{ProfileID: "p2", LibraryID: 1},
|
||
|
|
{ProfileID: "p1", LibraryID: 2},
|
||
|
|
{ProfileID: "p1", LibraryID: 5},
|
||
|
|
}
|
||
|
|
|
||
|
|
AttachAllowedLibraries(profiles, allowedLibraries)
|
||
|
|
|
||
|
|
if got, want := profiles[0].AllowedLibraryIDs, []int{2, 5}; !reflect.DeepEqual(got, want) {
|
||
|
|
t.Fatalf("p1 AllowedLibraryIDs = %v, want %v", got, want)
|
||
|
|
}
|
||
|
|
if got, want := profiles[1].AllowedLibraryIDs, []int{1}; !reflect.DeepEqual(got, want) {
|
||
|
|
t.Fatalf("p2 AllowedLibraryIDs = %v, want %v", got, want)
|
||
|
|
}
|
||
|
|
if profiles[2].AllowedLibraryIDs != nil {
|
||
|
|
t.Fatalf("p3 AllowedLibraryIDs = %v, want nil", profiles[2].AllowedLibraryIDs)
|
||
|
|
}
|
||
|
|
}
|