48 lines
1.1 KiB
Go
48 lines
1.1 KiB
Go
package wallet
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/btcsuite/btcd/chaincfg"
|
|
"github.com/btcsuite/btcutil/hdkeychain"
|
|
)
|
|
|
|
// testWallet creates a test wallet and unlocks it.
|
|
func testWallet(t *testing.T) (*Wallet, func()) {
|
|
// Set up a wallet.
|
|
dir, err := ioutil.TempDir("", "test_wallet")
|
|
if err != nil {
|
|
t.Fatalf("Failed to create db dir: %v", err)
|
|
}
|
|
|
|
cleanup := func() {
|
|
if err := os.RemoveAll(dir); err != nil {
|
|
t.Fatalf("could not cleanup test: %v", err)
|
|
}
|
|
}
|
|
|
|
seed, err := hdkeychain.GenerateSeed(hdkeychain.MinSeedBytes)
|
|
if err != nil {
|
|
t.Fatalf("unable to create seed: %v", err)
|
|
}
|
|
|
|
pubPass := []byte("hello")
|
|
privPass := []byte("world")
|
|
|
|
loader := NewLoader(&chaincfg.TestNet3Params, dir, true, 250)
|
|
w, err := loader.CreateNewWallet(pubPass, privPass, seed, time.Now())
|
|
if err != nil {
|
|
t.Fatalf("unable to create wallet: %v", err)
|
|
}
|
|
chainClient := &mockChainClient{}
|
|
w.chainClient = chainClient
|
|
if err := w.Unlock(privPass, time.After(10*time.Minute)); err != nil {
|
|
t.Fatalf("unable to unlock wallet: %v", err)
|
|
}
|
|
|
|
return w, cleanup
|
|
}
|