47ca1ca6e5
StartWithdrawal now persists the WithdrawalStatus before returning, and also returns a previously saved one in subsequent calls with the same parameters.
104 lines
3.2 KiB
Go
104 lines
3.2 KiB
Go
/*
|
|
* Copyright (c) 2014 The btcsuite developers
|
|
*
|
|
* 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 votingpool
|
|
|
|
import (
|
|
"github.com/btcsuite/btcd/wire"
|
|
"github.com/btcsuite/btcutil/hdkeychain"
|
|
"github.com/btcsuite/btcwallet/waddrmgr"
|
|
"github.com/btcsuite/btcwallet/walletdb"
|
|
)
|
|
|
|
var TstLastErr = lastErr
|
|
|
|
const TstEligibleInputMinConfirmations = eligibleInputMinConfirmations
|
|
|
|
// TstPutSeries transparently wraps the voting pool putSeries method.
|
|
func (vp *Pool) TstPutSeries(version, seriesID, reqSigs uint32, inRawPubKeys []string) error {
|
|
return vp.putSeries(version, seriesID, reqSigs, inRawPubKeys)
|
|
}
|
|
|
|
var TstBranchOrder = branchOrder
|
|
|
|
// TstExistsSeries checks whether a series is stored in the database.
|
|
func (vp *Pool) TstExistsSeries(seriesID uint32) (bool, error) {
|
|
var exists bool
|
|
err := vp.namespace.View(
|
|
func(tx walletdb.Tx) error {
|
|
poolBucket := tx.RootBucket().Bucket(vp.ID)
|
|
if poolBucket == nil {
|
|
return nil
|
|
}
|
|
bucket := poolBucket.Bucket(seriesBucketName)
|
|
if bucket == nil {
|
|
return nil
|
|
}
|
|
exists = bucket.Get(uint32ToBytes(seriesID)) != nil
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
return exists, nil
|
|
}
|
|
|
|
// TstNamespace exposes the Pool's namespace as it's needed in some tests.
|
|
func (vp *Pool) TstNamespace() walletdb.Namespace {
|
|
return vp.namespace
|
|
}
|
|
|
|
// TstGetRawPublicKeys gets a series public keys in string format.
|
|
func (s *SeriesData) TstGetRawPublicKeys() []string {
|
|
rawKeys := make([]string, len(s.publicKeys))
|
|
for i, key := range s.publicKeys {
|
|
rawKeys[i] = key.String()
|
|
}
|
|
return rawKeys
|
|
}
|
|
|
|
// TstGetRawPrivateKeys gets a series private keys in string format.
|
|
func (s *SeriesData) TstGetRawPrivateKeys() []string {
|
|
rawKeys := make([]string, len(s.privateKeys))
|
|
for i, key := range s.privateKeys {
|
|
if key != nil {
|
|
rawKeys[i] = key.String()
|
|
}
|
|
}
|
|
return rawKeys
|
|
}
|
|
|
|
// TstGetReqSigs expose the series reqSigs attribute.
|
|
func (s *SeriesData) TstGetReqSigs() uint32 {
|
|
return s.reqSigs
|
|
}
|
|
|
|
// TstEmptySeriesLookup empties the voting pool seriesLookup attribute.
|
|
func (vp *Pool) TstEmptySeriesLookup() {
|
|
vp.seriesLookup = make(map[uint32]*SeriesData)
|
|
}
|
|
|
|
// TstDecryptExtendedKey expose the decryptExtendedKey method.
|
|
func (vp *Pool) TstDecryptExtendedKey(keyType waddrmgr.CryptoKeyType, encrypted []byte) (*hdkeychain.ExtendedKey, error) {
|
|
return vp.decryptExtendedKey(keyType, encrypted)
|
|
}
|
|
|
|
// TstGetMsgTx returns a copy of the withdrawal transaction with the given
|
|
// ntxid.
|
|
func (s *WithdrawalStatus) TstGetMsgTx(ntxid Ntxid) *wire.MsgTx {
|
|
return s.transactions[ntxid].MsgTx.Copy()
|
|
}
|