b9fd527d33
This commit is the result of several big changes being made to the wallet. In particular, the "handshake" (initial sync to the chain server) was quite racy and required proper synchronization. To make fixing this race easier, several other changes were made to the internal wallet data structures and much of the RPC server ended up being rewritten. First, all account support has been removed. The previous Account struct has been replaced with a Wallet structure, which includes a keystore for saving keys, and a txstore for storing relevant transactions. This decision has been made since it is the opinion of myself and other developers that bitcoind accounts are fundamentally broken (as accounts implemented by bitcoind support both arbitrary address groupings as well as moving balances between accounts -- these are fundamentally incompatible features), and since a BIP0032 keystore is soon planned to be implemented (at which point, "accounts" can return as HD extended keys). With the keystore handling the grouping of related keys, there is no reason have many different Account structs, and the AccountManager has been removed as well. All RPC handlers that take an account option will only work with "" (the default account) or "*" if the RPC allows specifying all accounts. Second, much of the RPC server has been cleaned up. The global variables for the RPC server and chain server client have been moved to part of the rpcServer struct, and the handlers for each RPC method that are looked up change depending on which components have been set. Passthrough requests are also no longer handled specially, but when the chain server is set, a handler to perform the passthrough will be returned if the method is not otherwise a wallet RPC. The notification system for websocket clients has also been rewritten so wallet components can send notifications through channels, rather than requiring direct access to the RPC server itself, or worse still, sending directly to a websocket client's send channel. In the future, this will enable proper registration of notifications, rather than unsolicited broadcasts to every connected websocket client (see issue #84). Finally, and the main reason why much of this cleanup was necessary, the races during intial sync with the chain server have been fixed. Previously, when the 'Handshake' was run, a rescan would occur which would perform modifications to Account data structures as notifications were received. Synchronization was provided with a single binary semaphore which serialized all access to wallet and account data. However, the Handshake itself was not able to run with this lock (or else notifications would block), and many data races would occur as both notifications were being handled. If GOMAXPROCS was ever increased beyond 1, btcwallet would always immediately crash due to invalid addresses caused by the data races on startup. To fix this, the single lock for all wallet access has been replaced with mutexes for both the keystore and txstore. Handling of btcd notifications and client requests may now occur simultaneously. GOMAXPROCS has also been set to the number of logical CPUs at the beginning of main, since with the data races fixed, there's no reason to prevent the extra parallelism gained by increasing it. Closes #78. Closes #101. Closes #110.
596 lines
18 KiB
Go
596 lines
18 KiB
Go
// Copyright (c) 2013, 2014 Conformal Systems LLC <info@conformal.com>
|
|
//
|
|
// Permission to use, copy, modify, and distribute this software for any
|
|
// purpose with or without fee is hereby granted, provided that the above
|
|
// copyright notice and this permission notice appear in all copies.
|
|
//
|
|
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
|
// WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
|
// MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
|
// ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
|
// WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
|
// ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
|
|
// OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
|
|
|
package txstore_test
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/hex"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/conformal/btcnet"
|
|
"github.com/conformal/btcutil"
|
|
. "github.com/conformal/btcwallet/txstore"
|
|
"github.com/conformal/btcwire"
|
|
)
|
|
|
|
// Received transaction output for mainnet outpoint
|
|
// 61d3696de4c888730cbe06b0ad8ecb6d72d6108e893895aa9bc067bd7eba3fad:0
|
|
var (
|
|
TstRecvSerializedTx, _ = hex.DecodeString("010000000114d9ff358894c486b4ae11c2a8cf7851b1df64c53d2e511278eff17c22fb7373000000008c493046022100995447baec31ee9f6d4ec0e05cb2a44f6b817a99d5f6de167d1c75354a946410022100c9ffc23b64d770b0e01e7ff4d25fbc2f1ca8091053078a247905c39fce3760b601410458b8e267add3c1e374cf40f1de02b59213a82e1d84c2b94096e22e2f09387009c96debe1d0bcb2356ffdcf65d2a83d4b34e72c62eccd8490dbf2110167783b2bffffffff0280969800000000001976a914479ed307831d0ac19ebc5f63de7d5f1a430ddb9d88ac38bfaa00000000001976a914dadf9e3484f28b385ddeaa6c575c0c0d18e9788a88ac00000000")
|
|
TstRecvTx, _ = btcutil.NewTxFromBytes(TstRecvSerializedTx)
|
|
TstRecvTxSpendingTxBlockHash, _ = btcwire.NewShaHashFromStr("00000000000000017188b968a371bab95aa43522665353b646e41865abae02a4")
|
|
TstRecvAmt = int64(10000000)
|
|
TstRecvIndex = 684
|
|
TstRecvTxBlockDetails = &Block{
|
|
Height: 276425,
|
|
Hash: *TstRecvTxSpendingTxBlockHash,
|
|
Time: time.Unix(1387737310, 0),
|
|
}
|
|
|
|
TstRecvCurrentHeight = int32(284498) // mainnet blockchain height at time of writing
|
|
TstRecvTxOutConfirms = 8074 // hardcoded number of confirmations given the above block height
|
|
|
|
TstSpendingSerializedTx, _ = hex.DecodeString("0100000003ad3fba7ebd67c09baa9538898e10d6726dcb8eadb006be0c7388c8e46d69d361000000006b4830450220702c4fbde5532575fed44f8d6e8c3432a2a9bd8cff2f966c3a79b2245a7c88db02210095d6505a57e350720cb52b89a9b56243c15ddfcea0596aedc1ba55d9fb7d5aa0012103cccb5c48a699d3efcca6dae277fee6b82e0229ed754b742659c3acdfed2651f9ffffffffdbd36173f5610e34de5c00ed092174603761595d90190f790e79cda3e5b45bc2010000006b483045022000fa20735e5875e64d05bed43d81b867f3bd8745008d3ff4331ef1617eac7c44022100ad82261fc57faac67fc482a37b6bf18158da0971e300abf5fe2f9fd39e107f58012102d4e1caf3e022757512c204bf09ff56a9981df483aba3c74bb60d3612077c9206ffffffff65536c9d964b6f89b8ef17e83c6666641bc495cb27bab60052f76cd4556ccd0d040000006a473044022068e3886e0299ffa69a1c3ee40f8b6700f5f6d463a9cf9dbf22c055a131fc4abc02202b58957fe19ff1be7a84c458d08016c53fbddec7184ac5e633f2b282ae3420ae012103b4e411b81d32a69fb81178a8ea1abaa12f613336923ee920ffbb1b313af1f4d2ffffffff02ab233200000000001976a91418808b2fbd8d2c6d022aed5cd61f0ce6c0a4cbb688ac4741f011000000001976a914f081088a300c80ce36b717a9914ab5ec8a7d283988ac00000000")
|
|
TstSpendingTx, _ = btcutil.NewTxFromBytes(TstSpendingSerializedTx)
|
|
TstSpendingTxBlockHeight = int32(279143)
|
|
TstSignedTxBlockHash, _ = btcwire.NewShaHashFromStr("00000000000000017188b968a371bab95aa43522665353b646e41865abae02a4")
|
|
TstSignedTxIndex = 123
|
|
TstSignedTxBlockDetails = &Block{
|
|
Height: TstSpendingTxBlockHeight,
|
|
Hash: *TstSignedTxBlockHash,
|
|
Time: time.Unix(1389114091, 0),
|
|
}
|
|
)
|
|
|
|
func TestInsertsCreditsDebitsRollbacks(t *testing.T) {
|
|
// Create a double spend of the received blockchain transaction.
|
|
dupRecvTx, _ := btcutil.NewTxFromBytes(TstRecvSerializedTx)
|
|
// Switch txout amount to 1 BTC. Transaction store doesn't
|
|
// validate txs, so this is fine for testing a double spend
|
|
// removal.
|
|
TstDupRecvAmount := int64(1e8)
|
|
newDupMsgTx := dupRecvTx.MsgTx()
|
|
newDupMsgTx.TxOut[0].Value = TstDupRecvAmount
|
|
TstDoubleSpendTx := btcutil.NewTx(newDupMsgTx)
|
|
|
|
// Create a "signed" (with invalid sigs) tx that spends output 0 of
|
|
// the double spend.
|
|
spendingTx := btcwire.NewMsgTx()
|
|
spendingTxIn := btcwire.NewTxIn(btcwire.NewOutPoint(TstDoubleSpendTx.Sha(), 0), []byte{0, 1, 2, 3, 4})
|
|
spendingTx.AddTxIn(spendingTxIn)
|
|
spendingTxOut1 := btcwire.NewTxOut(1e7, []byte{5, 6, 7, 8, 9})
|
|
spendingTxOut2 := btcwire.NewTxOut(9e7, []byte{10, 11, 12, 13, 14})
|
|
spendingTx.AddTxOut(spendingTxOut1)
|
|
spendingTx.AddTxOut(spendingTxOut2)
|
|
TstSpendingTx := btcutil.NewTx(spendingTx)
|
|
var _ = TstSpendingTx
|
|
|
|
tests := []struct {
|
|
name string
|
|
f func(*Store) (*Store, error)
|
|
bal, unc btcutil.Amount
|
|
unspents map[btcwire.OutPoint]struct{}
|
|
unmined map[btcwire.ShaHash]struct{}
|
|
}{
|
|
{
|
|
name: "new store",
|
|
f: func(_ *Store) (*Store, error) {
|
|
return New("/tmp/tx.bin"), nil
|
|
},
|
|
bal: 0,
|
|
unc: 0,
|
|
unspents: map[btcwire.OutPoint]struct{}{},
|
|
unmined: map[btcwire.ShaHash]struct{}{},
|
|
},
|
|
{
|
|
name: "txout insert",
|
|
f: func(s *Store) (*Store, error) {
|
|
r, err := s.InsertTx(TstRecvTx, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
_, err = r.AddCredit(0, false)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
// Verify that we can create the JSON output without any
|
|
// errors.
|
|
_, err = r.ToJSON("", 100, &btcnet.MainNetParams)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return s, nil
|
|
},
|
|
bal: 0,
|
|
unc: btcutil.Amount(TstRecvTx.MsgTx().TxOut[0].Value),
|
|
unspents: map[btcwire.OutPoint]struct{}{
|
|
*btcwire.NewOutPoint(TstRecvTx.Sha(), 0): {},
|
|
},
|
|
unmined: map[btcwire.ShaHash]struct{}{},
|
|
},
|
|
{
|
|
name: "insert duplicate unconfirmed",
|
|
f: func(s *Store) (*Store, error) {
|
|
r, err := s.InsertTx(TstRecvTx, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
_, err = r.AddCredit(0, false)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
_, err = r.ToJSON("", 100, &btcnet.MainNetParams)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return s, nil
|
|
},
|
|
bal: 0,
|
|
unc: btcutil.Amount(TstRecvTx.MsgTx().TxOut[0].Value),
|
|
unspents: map[btcwire.OutPoint]struct{}{
|
|
*btcwire.NewOutPoint(TstRecvTx.Sha(), 0): {},
|
|
},
|
|
unmined: map[btcwire.ShaHash]struct{}{},
|
|
},
|
|
{
|
|
name: "confirmed txout insert",
|
|
f: func(s *Store) (*Store, error) {
|
|
TstRecvTx.SetIndex(TstRecvIndex)
|
|
r, err := s.InsertTx(TstRecvTx, TstRecvTxBlockDetails)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
_, err = r.AddCredit(0, false)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
_, err = r.ToJSON("", 100, &btcnet.MainNetParams)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return s, nil
|
|
},
|
|
bal: btcutil.Amount(TstRecvTx.MsgTx().TxOut[0].Value),
|
|
unc: 0,
|
|
unspents: map[btcwire.OutPoint]struct{}{
|
|
*btcwire.NewOutPoint(TstRecvTx.Sha(), 0): {},
|
|
},
|
|
unmined: map[btcwire.ShaHash]struct{}{},
|
|
},
|
|
{
|
|
name: "insert duplicate confirmed",
|
|
f: func(s *Store) (*Store, error) {
|
|
TstRecvTx.SetIndex(TstRecvIndex)
|
|
r, err := s.InsertTx(TstRecvTx, TstRecvTxBlockDetails)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
_, err = r.AddCredit(0, false)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
_, err = r.ToJSON("", 100, &btcnet.MainNetParams)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return s, nil
|
|
},
|
|
bal: btcutil.Amount(TstRecvTx.MsgTx().TxOut[0].Value),
|
|
unc: 0,
|
|
unspents: map[btcwire.OutPoint]struct{}{
|
|
*btcwire.NewOutPoint(TstRecvTx.Sha(), 0): {},
|
|
},
|
|
unmined: map[btcwire.ShaHash]struct{}{},
|
|
},
|
|
{
|
|
name: "rollback confirmed credit",
|
|
f: func(s *Store) (*Store, error) {
|
|
err := s.Rollback(TstRecvTxBlockDetails.Height)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return s, nil
|
|
},
|
|
bal: 0,
|
|
unc: btcutil.Amount(TstRecvTx.MsgTx().TxOut[0].Value),
|
|
unspents: map[btcwire.OutPoint]struct{}{
|
|
*btcwire.NewOutPoint(TstRecvTx.Sha(), 0): {},
|
|
},
|
|
unmined: map[btcwire.ShaHash]struct{}{},
|
|
},
|
|
{
|
|
name: "insert confirmed double spend",
|
|
f: func(s *Store) (*Store, error) {
|
|
TstDoubleSpendTx.SetIndex(TstRecvIndex)
|
|
r, err := s.InsertTx(TstDoubleSpendTx, TstRecvTxBlockDetails)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
_, err = r.AddCredit(0, false)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
_, err = r.ToJSON("", 100, &btcnet.MainNetParams)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return s, nil
|
|
},
|
|
bal: btcutil.Amount(TstDoubleSpendTx.MsgTx().TxOut[0].Value),
|
|
unc: 0,
|
|
unspents: map[btcwire.OutPoint]struct{}{
|
|
*btcwire.NewOutPoint(TstDoubleSpendTx.Sha(), 0): {},
|
|
},
|
|
unmined: map[btcwire.ShaHash]struct{}{},
|
|
},
|
|
{
|
|
name: "insert unconfirmed debit",
|
|
f: func(s *Store) (*Store, error) {
|
|
_, err := s.InsertTx(TstDoubleSpendTx, TstRecvTxBlockDetails)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
r, err := s.InsertTx(TstSpendingTx, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
_, err = r.AddDebits()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
_, err = r.ToJSON("", 100, &btcnet.MainNetParams)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return s, nil
|
|
},
|
|
bal: 0,
|
|
unc: 0,
|
|
unspents: map[btcwire.OutPoint]struct{}{},
|
|
unmined: map[btcwire.ShaHash]struct{}{
|
|
*TstSpendingTx.Sha(): {},
|
|
},
|
|
},
|
|
{
|
|
name: "insert unconfirmed debit again",
|
|
f: func(s *Store) (*Store, error) {
|
|
_, err := s.InsertTx(TstDoubleSpendTx, TstRecvTxBlockDetails)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
r, err := s.InsertTx(TstSpendingTx, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
_, err = r.AddDebits()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
_, err = r.ToJSON("", 100, &btcnet.MainNetParams)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return s, nil
|
|
},
|
|
bal: 0,
|
|
unc: 0,
|
|
unspents: map[btcwire.OutPoint]struct{}{},
|
|
unmined: map[btcwire.ShaHash]struct{}{
|
|
*TstSpendingTx.Sha(): {},
|
|
},
|
|
},
|
|
{
|
|
name: "insert change (index 0)",
|
|
f: func(s *Store) (*Store, error) {
|
|
r, err := s.InsertTx(TstSpendingTx, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
_, err = r.AddCredit(0, true)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
_, err = r.ToJSON("", 100, &btcnet.MainNetParams)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return s, nil
|
|
},
|
|
bal: 0,
|
|
unc: btcutil.Amount(TstSpendingTx.MsgTx().TxOut[0].Value),
|
|
unspents: map[btcwire.OutPoint]struct{}{
|
|
*btcwire.NewOutPoint(TstSpendingTx.Sha(), 0): {},
|
|
},
|
|
unmined: map[btcwire.ShaHash]struct{}{
|
|
*TstSpendingTx.Sha(): {},
|
|
},
|
|
},
|
|
{
|
|
name: "insert output back to this own wallet (index 1)",
|
|
f: func(s *Store) (*Store, error) {
|
|
r, err := s.InsertTx(TstSpendingTx, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
_, err = r.AddCredit(1, true)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
_, err = r.ToJSON("", 100, &btcnet.MainNetParams)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return s, nil
|
|
},
|
|
bal: 0,
|
|
unc: btcutil.Amount(TstSpendingTx.MsgTx().TxOut[0].Value + TstSpendingTx.MsgTx().TxOut[1].Value),
|
|
unspents: map[btcwire.OutPoint]struct{}{
|
|
*btcwire.NewOutPoint(TstSpendingTx.Sha(), 0): {},
|
|
*btcwire.NewOutPoint(TstSpendingTx.Sha(), 1): {},
|
|
},
|
|
unmined: map[btcwire.ShaHash]struct{}{
|
|
*TstSpendingTx.Sha(): {},
|
|
},
|
|
},
|
|
{
|
|
name: "confirm signed tx",
|
|
f: func(s *Store) (*Store, error) {
|
|
TstSpendingTx.SetIndex(TstSignedTxIndex)
|
|
r, err := s.InsertTx(TstSpendingTx, TstSignedTxBlockDetails)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
_, err = r.ToJSON("", 100, &btcnet.MainNetParams)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return s, nil
|
|
},
|
|
bal: btcutil.Amount(TstSpendingTx.MsgTx().TxOut[0].Value + TstSpendingTx.MsgTx().TxOut[1].Value),
|
|
unc: 0,
|
|
unspents: map[btcwire.OutPoint]struct{}{
|
|
*btcwire.NewOutPoint(TstSpendingTx.Sha(), 0): {},
|
|
*btcwire.NewOutPoint(TstSpendingTx.Sha(), 1): {},
|
|
},
|
|
unmined: map[btcwire.ShaHash]struct{}{},
|
|
},
|
|
{
|
|
name: "rollback after spending tx",
|
|
f: func(s *Store) (*Store, error) {
|
|
err := s.Rollback(TstSignedTxBlockDetails.Height + 1)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return s, nil
|
|
},
|
|
bal: btcutil.Amount(TstSpendingTx.MsgTx().TxOut[0].Value + TstSpendingTx.MsgTx().TxOut[1].Value),
|
|
unc: 0,
|
|
unspents: map[btcwire.OutPoint]struct{}{
|
|
*btcwire.NewOutPoint(TstSpendingTx.Sha(), 0): {},
|
|
*btcwire.NewOutPoint(TstSpendingTx.Sha(), 1): {},
|
|
},
|
|
unmined: map[btcwire.ShaHash]struct{}{},
|
|
},
|
|
{
|
|
name: "rollback spending tx block",
|
|
f: func(s *Store) (*Store, error) {
|
|
err := s.Rollback(TstSignedTxBlockDetails.Height)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return s, nil
|
|
},
|
|
bal: 0,
|
|
unc: btcutil.Amount(TstSpendingTx.MsgTx().TxOut[0].Value + TstSpendingTx.MsgTx().TxOut[1].Value),
|
|
unspents: map[btcwire.OutPoint]struct{}{
|
|
*btcwire.NewOutPoint(TstSpendingTx.Sha(), 0): {},
|
|
*btcwire.NewOutPoint(TstSpendingTx.Sha(), 1): {},
|
|
},
|
|
unmined: map[btcwire.ShaHash]struct{}{
|
|
*TstSpendingTx.Sha(): {},
|
|
},
|
|
},
|
|
{
|
|
name: "rollback double spend tx block",
|
|
f: func(s *Store) (*Store, error) {
|
|
err := s.Rollback(TstRecvTxBlockDetails.Height)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return s, nil
|
|
},
|
|
bal: 0,
|
|
unc: btcutil.Amount(TstSpendingTx.MsgTx().TxOut[0].Value + TstSpendingTx.MsgTx().TxOut[1].Value),
|
|
unspents: map[btcwire.OutPoint]struct{}{
|
|
*btcwire.NewOutPoint(TstSpendingTx.Sha(), 0): {},
|
|
*btcwire.NewOutPoint(TstSpendingTx.Sha(), 1): {},
|
|
},
|
|
unmined: map[btcwire.ShaHash]struct{}{
|
|
*TstSpendingTx.Sha(): {},
|
|
},
|
|
},
|
|
{
|
|
name: "insert original recv txout",
|
|
f: func(s *Store) (*Store, error) {
|
|
TstRecvTx.SetIndex(TstRecvIndex)
|
|
r, err := s.InsertTx(TstRecvTx, TstRecvTxBlockDetails)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
_, err = r.AddCredit(0, false)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
_, err = r.ToJSON("", 100, &btcnet.MainNetParams)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return s, nil
|
|
},
|
|
bal: btcutil.Amount(TstRecvTx.MsgTx().TxOut[0].Value),
|
|
unc: 0,
|
|
unspents: map[btcwire.OutPoint]struct{}{
|
|
*btcwire.NewOutPoint(TstRecvTx.Sha(), 0): {},
|
|
},
|
|
unmined: map[btcwire.ShaHash]struct{}{},
|
|
},
|
|
}
|
|
|
|
var s *Store
|
|
for _, test := range tests {
|
|
tmpStore, err := test.f(s)
|
|
if err != nil {
|
|
t.Fatalf("%s: got error: %v", test.name, err)
|
|
}
|
|
s = tmpStore
|
|
bal, err := s.Balance(1, TstRecvCurrentHeight)
|
|
if err != nil {
|
|
t.Fatalf("%s: Confirmed Balance() failed: %v", test.name, err)
|
|
}
|
|
if bal != test.bal {
|
|
t.Fatalf("%s: balance mismatch: expected: %d, got: %d", test.name, test.bal, bal)
|
|
}
|
|
unc, err := s.Balance(0, TstRecvCurrentHeight)
|
|
if err != nil {
|
|
t.Fatalf("%s: Unconfirmed Balance() failed: %v", test.name, err)
|
|
}
|
|
unc -= bal
|
|
if unc != test.unc {
|
|
t.Errorf("%s: unconfirmed balance mismatch: expected %d, got %d", test.name, test.unc, unc)
|
|
}
|
|
|
|
// Check that unspent outputs match expected.
|
|
unspent, err := s.UnspentOutputs()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
for _, r := range unspent {
|
|
if r.Spent() {
|
|
t.Errorf("%s: unspent record marked as spent", test.name)
|
|
}
|
|
|
|
op := *r.OutPoint()
|
|
if _, ok := test.unspents[op]; !ok {
|
|
t.Errorf("%s: unexpected unspent output: %v", test.name, op)
|
|
}
|
|
delete(test.unspents, op)
|
|
}
|
|
if len(test.unspents) != 0 {
|
|
t.Errorf("%s: missing expected unspent output(s)", test.name)
|
|
}
|
|
|
|
// Check that unmined sent txs match expected.
|
|
for _, tx := range s.UnminedDebitTxs() {
|
|
if _, ok := test.unmined[*tx.Sha()]; !ok {
|
|
t.Fatalf("%s: unexpected unmined signed tx: %v", test.name, *tx.Sha())
|
|
}
|
|
delete(test.unmined, *tx.Sha())
|
|
}
|
|
if len(test.unmined) != 0 {
|
|
t.Errorf("%s: missing expected unmined signed tx(s)", test.name)
|
|
}
|
|
|
|
// Pass a re-serialized version of the store to each next test.
|
|
buf := new(bytes.Buffer)
|
|
nWritten, err := s.WriteTo(buf)
|
|
if err != nil {
|
|
t.Fatalf("%v: serialization failed: %v (wrote %v bytes)", test.name, err, nWritten)
|
|
}
|
|
if nWritten != int64(buf.Len()) {
|
|
t.Errorf("%v: wrote %v bytes but buffer has %v", test.name, nWritten, buf.Len())
|
|
}
|
|
nRead, err := s.ReadFrom(buf)
|
|
if err != nil {
|
|
t.Fatalf("%v: deserialization failed: %v (read %v bytes after writing %v)",
|
|
test.name, err, nRead, nWritten)
|
|
}
|
|
if nWritten != nRead {
|
|
t.Errorf("%v: number of bytes written (%v) does not match those read (%v)",
|
|
test.name, nWritten, nRead)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestFindingSpentCredits(t *testing.T) {
|
|
s := New("/tmp/tx.bin")
|
|
|
|
// Insert transaction and credit which will be spent.
|
|
r, err := s.InsertTx(TstRecvTx, TstRecvTxBlockDetails)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
_, err = r.AddCredit(0, false)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Insert confirmed transaction which spends the above credit.
|
|
TstSpendingTx.SetIndex(TstSignedTxIndex)
|
|
r2, err := s.InsertTx(TstSpendingTx, TstSignedTxBlockDetails)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
_, err = r2.AddCredit(0, false)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
_, err = r2.AddDebits()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
bal, err := s.Balance(1, TstSignedTxBlockDetails.Height)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if bal != btcutil.Amount(TstSpendingTx.MsgTx().TxOut[0].Value) {
|
|
t.Fatal("bad balance")
|
|
}
|
|
unspents, err := s.UnspentOutputs()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
op := btcwire.NewOutPoint(TstSpendingTx.Sha(), 0)
|
|
if *unspents[0].OutPoint() != *op {
|
|
t.Fatal("unspent outpoint doesn't match expected")
|
|
}
|
|
if len(unspents) > 1 {
|
|
t.Fatal("has more than one unspent credit")
|
|
}
|
|
}
|