package macaroons_test import ( "bytes" "context" "io/ioutil" "os" "path" "testing" "github.com/coreos/bbolt" "github.com/lightningnetwork/lnd/macaroons" "github.com/btcsuite/btcwallet/snacl" ) func TestStore(t *testing.T) { tempDir, err := ioutil.TempDir("", "macaroonstore-") if err != nil { t.Fatalf("Error creating temp dir: %v", err) } defer os.RemoveAll(tempDir) db, err := bbolt.Open(path.Join(tempDir, "weks.db"), 0600, bbolt.DefaultOptions) if err != nil { t.Fatalf("Error opening store DB: %v", err) } store, err := macaroons.NewRootKeyStorage(db) if err != nil { db.Close() t.Fatalf("Error creating root key store: %v", err) } defer store.Close() _, _, err = store.RootKey(context.TODO()) if err != macaroons.ErrStoreLocked { t.Fatalf("Received %v instead of ErrStoreLocked", err) } _, err = store.Get(context.TODO(), nil) if err != macaroons.ErrStoreLocked { t.Fatalf("Received %v instead of ErrStoreLocked", err) } pw := []byte("weks") err = store.CreateUnlock(&pw) if err != nil { t.Fatalf("Error creating store encryption key: %v", err) } key, id, err := store.RootKey(context.TODO()) if err != nil { t.Fatalf("Error getting root key from store: %v", err) } rootID := id key2, err := store.Get(context.TODO(), id) if err != nil { t.Fatalf("Error getting key with ID %s: %v", string(id), err) } if !bytes.Equal(key, key2) { t.Fatalf("Root key doesn't match: expected %v, got %v", key, key2) } badpw := []byte("badweks") err = store.CreateUnlock(&badpw) if err != macaroons.ErrAlreadyUnlocked { t.Fatalf("Received %v instead of ErrAlreadyUnlocked", err) } store.Close() // Between here and the re-opening of the store, it's possible to get // a double-close, but that's not such a big deal since the tests will // fail anyway in that case. db, err = bbolt.Open(path.Join(tempDir, "weks.db"), 0600, bbolt.DefaultOptions) if err != nil { t.Fatalf("Error opening store DB: %v", err) } store, err = macaroons.NewRootKeyStorage(db) if err != nil { db.Close() t.Fatalf("Error creating root key store: %v", err) } err = store.CreateUnlock(&badpw) if err != snacl.ErrInvalidPassword { t.Fatalf("Received %v instead of ErrInvalidPassword", err) } err = store.CreateUnlock(nil) if err != macaroons.ErrPasswordRequired { t.Fatalf("Received %v instead of ErrPasswordRequired", err) } _, _, err = store.RootKey(context.TODO()) if err != macaroons.ErrStoreLocked { t.Fatalf("Received %v instead of ErrStoreLocked", err) } _, err = store.Get(context.TODO(), nil) if err != macaroons.ErrStoreLocked { t.Fatalf("Received %v instead of ErrStoreLocked", err) } err = store.CreateUnlock(&pw) if err != nil { t.Fatalf("Error unlocking root key store: %v", err) } key, err = store.Get(context.TODO(), rootID) if err != nil { t.Fatalf("Error getting key with ID %s: %v", string(rootID), err) } if !bytes.Equal(key, key2) { t.Fatalf("Root key doesn't match: expected %v, got %v", key2, key) } key, id, err = store.RootKey(context.TODO()) if err != nil { t.Fatalf("Error getting root key from store: %v", err) } if !bytes.Equal(key, key2) { t.Fatalf("Root key doesn't match: expected %v, got %v", key2, key) } if !bytes.Equal(rootID, id) { t.Fatalf("Root ID doesn't match: expected %v, got %v", rootID, id) } }