39 lines
957 B
Go
39 lines
957 B
Go
package store
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
func TestWhitelistAddListRemove(t *testing.T) {
|
|
s := openTemp(t)
|
|
_, _ = s.AddAccount(sampleAccount())
|
|
|
|
if err := s.AddWhitelist("work", DirIn, "Bob@Example.com"); err != nil {
|
|
t.Fatalf("add: %v", err)
|
|
}
|
|
_ = s.AddWhitelist("work", DirIn, "@trusted.com")
|
|
|
|
got, err := s.ListWhitelist("work", DirIn)
|
|
if err != nil {
|
|
t.Fatalf("list: %v", err)
|
|
}
|
|
want := []string{"@trusted.com", "bob@example.com"} // lower-cased, sorted
|
|
if !reflect.DeepEqual(got, want) {
|
|
t.Fatalf("got %v want %v", got, want)
|
|
}
|
|
|
|
// Out list is independent.
|
|
if out, _ := s.ListWhitelist("work", DirOut); len(out) != 0 {
|
|
t.Fatalf("out list should be empty, got %v", out)
|
|
}
|
|
|
|
if err := s.RemoveWhitelist("work", DirIn, "bob@example.com"); err != nil {
|
|
t.Fatalf("remove: %v", err)
|
|
}
|
|
got, _ = s.ListWhitelist("work", DirIn)
|
|
if !reflect.DeepEqual(got, []string{"@trusted.com"}) {
|
|
t.Fatalf("after remove got %v", got)
|
|
}
|
|
}
|