d1311c53ee
* Use RWMutex instead of Mutex * Use time.Duration instead of int for unlock time * Use time.After with select instead of time.Sleep
83 lines
1.8 KiB
Go
83 lines
1.8 KiB
Go
package accounts
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/ethereum/go-ethereum/crypto"
|
|
"github.com/ethereum/go-ethereum/crypto/randentropy"
|
|
"github.com/ethereum/go-ethereum/ethutil"
|
|
"time"
|
|
)
|
|
|
|
func TestAccountManager(t *testing.T) {
|
|
ks := crypto.NewKeyStorePlain(ethutil.DefaultDataDir() + "/testaccounts")
|
|
am := NewAccountManager(ks, 100)
|
|
pass := "" // not used but required by API
|
|
a1, err := am.NewAccount(pass)
|
|
toSign := randentropy.GetEntropyCSPRNG(32)
|
|
_, err = am.SignLocked(a1, pass, toSign)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Cleanup
|
|
time.Sleep(time.Millisecond * 150) // wait for locking
|
|
|
|
accounts, err := am.Accounts()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
for _, account := range accounts {
|
|
err := am.DeleteAccount(account.Address, pass)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestAccountManagerLocking(t *testing.T) {
|
|
ks := crypto.NewKeyStorePassphrase(ethutil.DefaultDataDir() + "/testaccounts")
|
|
am := NewAccountManager(ks, 200)
|
|
pass := "foo"
|
|
a1, err := am.NewAccount(pass)
|
|
toSign := randentropy.GetEntropyCSPRNG(32)
|
|
|
|
// Signing without passphrase fails because account is locked
|
|
_, err = am.Sign(a1, toSign)
|
|
if err != ErrLocked {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Signing with passphrase works
|
|
_, err = am.SignLocked(a1, pass, toSign)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Signing without passphrase works because account is temp unlocked
|
|
_, err = am.Sign(a1, toSign)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Signing without passphrase fails after automatic locking
|
|
time.Sleep(time.Millisecond * time.Duration(250))
|
|
|
|
_, err = am.Sign(a1, toSign)
|
|
if err != ErrLocked {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Cleanup
|
|
accounts, err := am.Accounts()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
for _, account := range accounts {
|
|
err := am.DeleteAccount(account.Address, pass)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
}
|