35 lines
1003 B
Go
35 lines
1003 B
Go
package store
|
|
|
|
import (
|
|
"path/filepath"
|
|
"testing"
|
|
)
|
|
|
|
func TestSetWhitelistEnabled(t *testing.T) {
|
|
st, err := Open(filepath.Join(t.TempDir(), "e.db"))
|
|
if err != nil {
|
|
t.Fatalf("open: %v", err)
|
|
}
|
|
defer st.Close()
|
|
k := make([]byte, 32)
|
|
if err := st.InitKeys(k, k); err != nil {
|
|
t.Fatalf("InitKeys: %v", err)
|
|
}
|
|
if _, err := st.AddAccount(Account{Name: "a", Mode: "RO", IMAPHost: "h", IMAPPort: 993, IMAPSecurity: "tls", AuthType: "password", Username: "u@x.com"}); err != nil {
|
|
t.Fatalf("AddAccount: %v", err)
|
|
}
|
|
if err := st.SetWhitelistEnabled("a", DirIn, true); err != nil {
|
|
t.Fatalf("SetWhitelistEnabled: %v", err)
|
|
}
|
|
got, err := st.GetAccount("a")
|
|
if err != nil {
|
|
t.Fatalf("GetAccount: %v", err)
|
|
}
|
|
if !got.WhitelistInEnabled || got.WhitelistOutEnabled {
|
|
t.Fatalf("flags wrong: in=%v out=%v", got.WhitelistInEnabled, got.WhitelistOutEnabled)
|
|
}
|
|
if err := st.SetWhitelistEnabled("missing", DirIn, true); err == nil {
|
|
t.Fatal("enabling on a missing account must error")
|
|
}
|
|
}
|