dfe617e05d
This a refactor of the btcwallet main package to create a new wallet package. The main feature of this package is the integration of all the other wallet components (waddrmgr, txstore, and chain) and the Wallet type is 'runnable', so it will be continuously updating itself against changes notified by the remote btcd instance. It also includes several methods which provide access to information necessary to run a wallet RPC server.
137 lines
3.2 KiB
Go
137 lines
3.2 KiB
Go
// TODO(jrick) Due to the extra encapsulation added during the switch
|
|
// to the new txstore, structures can no longer be mocked due to private
|
|
// members. Since all members for RecvTxOut and SignedTx are private, the
|
|
// simplist solution would be to make RecvTxOut an interface and create
|
|
// our own types satisifying the interface for this test package. Until
|
|
// then, disable this test.
|
|
//
|
|
// +build ignore
|
|
|
|
package wallet
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/btcsuite/btcd/txscript"
|
|
"github.com/btcsuite/btcd/wire"
|
|
"github.com/btcsuite/btcutil"
|
|
"github.com/btcsuite/btcwallet/tx"
|
|
)
|
|
|
|
func init() {
|
|
cfg = &Config{
|
|
KeypoolSize: 100,
|
|
}
|
|
}
|
|
|
|
type allowFreeTest struct {
|
|
name string
|
|
inputs []*tx.Utxo
|
|
curHeight int32
|
|
txSize int
|
|
free bool
|
|
}
|
|
|
|
var allowFreeTests = []allowFreeTest{
|
|
{
|
|
name: "priority < 57,600,000",
|
|
inputs: []*tx.Utxo{
|
|
{
|
|
Amt: btcutil.SatoshiPerBitcoin,
|
|
Height: 0,
|
|
},
|
|
},
|
|
curHeight: 142, // 143 confirmations
|
|
txSize: 250,
|
|
free: false,
|
|
},
|
|
{
|
|
name: "priority == 57,600,000",
|
|
inputs: []*tx.Utxo{
|
|
{
|
|
Amt: btcutil.SatoshiPerBitcoin,
|
|
Height: 0,
|
|
},
|
|
},
|
|
curHeight: 143, // 144 confirmations
|
|
txSize: 250,
|
|
free: false,
|
|
},
|
|
{
|
|
name: "priority > 57,600,000",
|
|
inputs: []*tx.Utxo{
|
|
{
|
|
Amt: btcutil.SatoshiPerBitcoin,
|
|
Height: 0,
|
|
},
|
|
},
|
|
curHeight: 144, // 145 confirmations
|
|
txSize: 250,
|
|
free: true,
|
|
},
|
|
}
|
|
|
|
func TestAllowFree(t *testing.T) {
|
|
for _, test := range allowFreeTests {
|
|
calcFree := allowFree(test.curHeight, test.inputs, test.txSize)
|
|
if calcFree != test.free {
|
|
t.Errorf("Allow free test '%v' failed.", test.name)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestFakeTxs(t *testing.T) {
|
|
// First we need a wallet.
|
|
w, err := keystore.NewStore("banana wallet", "", []byte("banana"),
|
|
wire.MainNet, &keystore.BlockStamp{}, 100)
|
|
if err != nil {
|
|
t.Errorf("Can not create encrypted wallet: %s", err)
|
|
return
|
|
}
|
|
a := &Wallet{
|
|
Wallet: w,
|
|
lockedOutpoints: map[wire.OutPoint]struct{}{},
|
|
}
|
|
|
|
w.Unlock([]byte("banana"))
|
|
|
|
// Create and add a fake Utxo so we have some funds to spend.
|
|
//
|
|
// This will pass validation because txcscript is unaware of invalid
|
|
// tx inputs, however, this example would fail in btcd.
|
|
utxo := &tx.Utxo{}
|
|
addr, err := w.NextChainedAddress(&keystore.BlockStamp{}, 100)
|
|
if err != nil {
|
|
t.Errorf("Cannot get next address: %s", err)
|
|
return
|
|
}
|
|
copy(utxo.AddrHash[:], addr.ScriptAddress())
|
|
ophash := (wire.ShaHash)([...]byte{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11,
|
|
12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27,
|
|
28, 29, 30, 31, 32})
|
|
out := wire.NewOutPoint(&ophash, 0)
|
|
utxo.Out = tx.OutPoint(*out)
|
|
ss, err := txscript.PayToAddrScript(addr)
|
|
if err != nil {
|
|
t.Errorf("Could not create utxo PkScript: %s", err)
|
|
return
|
|
}
|
|
utxo.Subscript = tx.PkScript(ss)
|
|
utxo.Amt = 1000000
|
|
utxo.Height = 12345
|
|
a.UtxoStore = append(a.UtxoStore, utxo)
|
|
|
|
// Fake our current block height so btcd doesn't need to be queried.
|
|
curBlock.BlockStamp.Height = 12346
|
|
|
|
// Create the transaction.
|
|
pairs := map[string]int64{
|
|
"17XhEvq9Nahdj7Xe1nv6oRe1tEmaHUuynH": 5000,
|
|
}
|
|
_, err = a.txToPairs(pairs, 1)
|
|
if err != nil {
|
|
t.Errorf("Tx creation failed: %s", err)
|
|
return
|
|
}
|
|
}
|