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")
	}
}